New unit test for added function with varargs parameter.

Reviewed by Hugo Lima <hugo.lima@openbossa.org>

2 files changed, 12 insertions, 0 deletions

diff --git a/tests/samplebinding/sample_test.py b/tests/samplebinding/sample_test.py
index 35110a0c9..785efaac3 100755
--- a/tests/samplebinding/sample_test.py
+++ b/tests/samplebinding/sample_test.py
@@ -63,6 +63,12 @@ class ModuleTest(unittest.TestCase):
         self.assertEqual(sample.multiplyString(str1, 3), str1 * 3)
         self.assertEqual(sample.multiplyString(str1, 0), str1 * 0)
 
+    def testAddedFunctionWithVarargs(self):
+        '''Calls function that receives varargs added to module from type system description.'''
+        self.assertEqual(sample.countVarargs(1), 0)
+        self.assertEqual(sample.countVarargs(1, 2), 1)
+        self.assertEqual(sample.countVarargs(1, 2, 3, 'a', 'b', 4, (5, 6)), 6)
+
 if __name__ == '__main__':
     unittest.main()
 
diff --git a/tests/samplebinding/typesystem_sample.xml b/tests/samplebinding/typesystem_sample.xml
index 9dc3d1798..631f5a9ee 100644
--- a/tests/samplebinding/typesystem_sample.xml
+++ b/tests/samplebinding/typesystem_sample.xml
@@ -828,6 +828,12 @@
         </inject-code>
     </add-function>
 
+    <add-function signature="countVarargs(int, ...)" return-type="int">
+        <inject-code class="target" position="beginning">
+            %PYARG_0 = %CONVERTTOPYTHON[int](PyTuple_GET_SIZE(%PYARG_2));
+        </inject-code>
+    </add-function>
+
     <rejection class="ListUser" function-name="createList()"/>
     <rejection class="ListUser" function-name="callCreateList()"/>
     <rejection class="ListUser" function-name="createComplexList(Complex, Complex)"/>