Updated necessary unit tests to handle with exceptions.

Reviewer: Marcelo Lira <marcelo.lira@openbossa.org>
          Luciano Wolf <luciano.wolf@openbossa.org>

2 files changed, 8 insertions, 2 deletions

diff --git a/tests/samplebinding/invalid_virtual_return_test.py b/tests/samplebinding/invalid_virtual_return_test.py
index 164271e8f..bf0a6e1f6 100644
--- a/tests/samplebinding/invalid_virtual_return_test.py
+++ b/tests/samplebinding/invalid_virtual_return_test.py
@@ -29,6 +29,9 @@
 import unittest
 from sample import ObjectModel, ObjectType, ObjectView
 
+import warnings
+warnings.simplefilter('error')
+
 
 class MyObject(ObjectType):
     pass
@@ -50,7 +53,7 @@ class ModelWrongReturnTest(unittest.TestCase):
     def testWrongTypeReturn(self):
         model = ListModelWrong()
         view = ObjectView(model)
-        self.assertRaises(TypeError, view.getRawModelData) # calls model.data()
+        self.assertRaises(RuntimeWarning, view.getRawModelData) # calls model.data()
 
 
 if __name__ == '__main__':
diff --git a/tests/samplebinding/virtualmethods_test.py b/tests/samplebinding/virtualmethods_test.py
index 0ddb7e447..73490eb84 100644
--- a/tests/samplebinding/virtualmethods_test.py
+++ b/tests/samplebinding/virtualmethods_test.py
@@ -31,6 +31,9 @@ import unittest
 
 from sample import *
 
+import warnings
+warnings.simplefilter('error')
+
 class ExtendedVirtualMethods(VirtualMethods):
     def __init__(self):
         VirtualMethods.__init__(self)
@@ -114,7 +117,7 @@ class VirtualMethodsTest(unittest.TestCase):
 class PrettyErrorMessageTest(unittest.TestCase):
     def testIt(self):
         obj = ExtendedVirtualMethods()
-        self.assertRaises(TypeError, obj.callStrListToStdList, StrList())
+        self.assertRaises(RuntimeWarning, obj.callStrListToStdList, StrList())
 
 if __name__ == '__main__':
     unittest.main()