Adds tests for argument removal and type modification for virtual methods.

Reviewed by Lauro Moura <lauro.neto@openbossa.org>

1 files changed, 21 insertions, 0 deletions

diff --git a/tests/samplebinding/injectcode_test.py b/tests/samplebinding/injectcode_test.py
index d328de16b..ffc8ff115 100755
--- a/tests/samplebinding/injectcode_test.py
+++ b/tests/samplebinding/injectcode_test.py
@@ -29,6 +29,13 @@
 import unittest
 from sample import InjectCode
 
+class MyInjectCode(InjectCode):
+    def __init__(self):
+        InjectCode.__init__(self)
+        self.multiplier = 2
+    def arrayMethod(self, values):
+        return self.multiplier * sum(values)
+
 class InjectCodeTest(unittest.TestCase):
 
     def testTypeNativeBeginning_TypeTargetBeginning(self):
@@ -68,5 +75,19 @@ class InjectCodeTest(unittest.TestCase):
         result = ic.arrayMethod(values)
         self.assertEqual(result, sum(values))
 
+    def testCallVirtualMethodWithArgumentRemovalAndArgumentTypeModification(self):
+        '''A virtual method has its first argument removed and the second modified.'''
+        ic = InjectCode()
+        values = (1, 2, 3, 4, 5)
+        result = ic.callArrayMethod(values)
+        self.assertEqual(result, sum(values))
+
+    def testCallReimplementedVirtualMethodWithArgumentRemovalAndArgumentTypeModification(self):
+        '''Calls a reimplemented virtual method that had its first argument removed and the second modified.'''
+        ic = MyInjectCode()
+        values = (1, 2, 3, 4, 5)
+        result = ic.callArrayMethod(values)
+        self.assertEqual(result, ic.multiplier * sum(values))
+
 if __name__ == '__main__':
     unittest.main()