def test_multiple_mock(self):
     from affinitic.testing.tests import test_basetestcase
     self.mock(test_basetestcase, 'message', return_value=0)
     self.assertEqual(0, test_basetestcase.message())
     self.mock(test_basetestcase, 'message', return_value=1)
     self.assertEqual(1, test_basetestcase.message())
     self.assertEqual(2, len(self._mocks))
     self.unmock(test_basetestcase.message)
     self.assertEqual(1, len(self._mocks))
     self.assertEqual('FOO', test_basetestcase.message())
 def test_mock_multiple_possibilities(self):
     from affinitic.testing.tests import test_basetestcase
     self.mock(test_basetestcase, 'message', mock=mock.Mock(return_value=0),
               return_value=1)
     self.assertEqual(0, test_basetestcase.message())