about summary refs log tree commit diff stats
path: root/test/ir/ir2C.py
diff options
context:
space:
mode:
authorAjax <commial@gmail.com>2015-03-23 19:40:44 +0100
committerAjax <commial@gmail.com>2015-03-24 19:08:38 +0100
commit40dcf88bfda7c8fc8492f408b08d58292af6bafa (patch)
treee3018442a0dcb5dc708d52f85df54b7e58f766e6 /test/ir/ir2C.py
parentaf43bf539554790edc10429ad906c487593d49d2 (diff)
downloadmiasm-40dcf88bfda7c8fc8492f408b08d58292af6bafa.tar.gz
miasm-40dcf88bfda7c8fc8492f408b08d58292af6bafa.zip
Translator: Update uses
Diffstat (limited to '')
-rw-r--r--test/ir/ir2C.py15
1 files changed, 8 insertions, 7 deletions
diff --git a/test/ir/ir2C.py b/test/ir/ir2C.py
index 11b9b10e..48518c50 100644
--- a/test/ir/ir2C.py
+++ b/test/ir/ir2C.py
@@ -7,16 +7,17 @@ import unittest
 class TestIrIr2C(unittest.TestCase):
 
     def translationTest(self, expr, expected):
-        from miasm2.ir.translators.C import TranslatorC
+        from miasm2.ir.translators import Translator
 
-        self.assertEqual(TranslatorC.from_expr(expr), expected)
+        translator = Translator.to_language("C")
+        self.assertEqual(translator.from_expr(expr), expected)
 
     def test_ExprOp_toC(self):
         from miasm2.expression.expression import ExprInt32, ExprOp
-        from miasm2.ir.translators.C import TranslatorC
+        from miasm2.ir.translators.C import Translator
 
         args = [ExprInt32(i) for i in xrange(9)]
-
+        translator = Translator.to_language("C")
 
         # Unary operators
         self.translationTest(
@@ -26,7 +27,7 @@ class TestIrIr2C(unittest.TestCase):
         self.translationTest(
             ExprOp('hex2bcd', *args[:1]), r'hex2bcd_32(0x0)')
         self.translationTest(ExprOp('fabs',    *args[:1]), r'fabs(0x0)')
-        self.assertRaises(NotImplementedError, TranslatorC.from_expr,
+        self.assertRaises(NotImplementedError, translator.from_expr,
                           ExprOp('X', *args[:1]))
 
         # Binary operators
@@ -50,7 +51,7 @@ class TestIrIr2C(unittest.TestCase):
             ExprOp('imod',    *args[:2]), r'imod32(vmcpu, 0x0, 0x1)')
         self.translationTest(
             ExprOp('bcdadd',  *args[:2]), r'bcdadd_32(0x0, 0x1)')
-        self.assertRaises(NotImplementedError, TranslatorC.from_expr,
+        self.assertRaises(NotImplementedError, translator.from_expr,
                           ExprOp('X', *args[:2]))
 
         # Ternary operators
@@ -60,7 +61,7 @@ class TestIrIr2C(unittest.TestCase):
         # Other cases
         self.translationTest(
             ExprOp('+',       *args[:3]), r'(((0x0&0xffffffff)+(0x1&0xffffffff)+(0x2&0xffffffff))&0xffffffff)')
-        self.assertRaises(NotImplementedError, TranslatorC.from_expr,
+        self.assertRaises(NotImplementedError, translator.from_expr,
                           ExprOp('X', *args[:3]))
 
 if __name__ == '__main__':