Пример #1
0
    def test_mock_method_calls(self):
        mock = Mock()
        mock()
        mock.method().attribute.method()
        mock.attribute.method()

        # mock_calls: inclusive, recursive; split by '()', good for callable mocks.
        self.assertEqual(mock.mock_calls, [
            call(),
            call.method(),
            call.method().attribute.method(),
            call.attribute.method()])

        # method_calls: exclusive, recursive; terminated by '()', good for non-callable mocks.
        self.assertEqual(mock.method_calls, [call.method(), call.attribute.method()])
Пример #2
0
    def test_any_mock_calls_comparison_order(self):
        mock = Mock()
        d = datetime.now()

        class Foo(object):
            def __eq__(self, other):
                return False

            def __ne__(self, other):
                return True

        for d in datetime.now(), Foo():
            mock.reset_mock()

            mock(d, foo=d, bar=d)
            mock.method(d, zinga=d, alpha=d)
            mock().method(a1=d, z99=d)

            expected = [
                call(ANY, foo=ANY, bar=ANY),
                call.method(ANY, zinga=ANY, alpha=ANY),
                call(),
                call().method(a1=ANY, z99=ANY)
            ]
            self.assertEqual(expected, mock.mock_calls)
            self.assertEqual(mock.mock_calls, expected)
Пример #3
0
    def test_corba_exception(self):
        self.corba_object.method.side_effect = CORBA.TRANSIENT
        with self.assertRaises(CORBA.TRANSIENT):
            self.corba_client.method()

        self.assertEqual(self.corba_object.mock_calls, [call.method()])
        self.assertLogs(['DEBUG', 'ERROR'],
                        ('method()', 'method failed with CORBA.TRANSIENT'))
Пример #4
0
    def test_user_exception(self):
        self.corba_object.method.side_effect = CustomError
        with self.assertRaises(CustomError):
            self.corba_client.method()

        self.assertEqual(self.corba_object.mock_calls, [call.method()])
        self.assertLogs(['DEBUG', 'DEBUG'],
                        ('method()', 'method failed with .*CustomError'))
Пример #5
0
    def test_mock_method_calls(self):
        mock = Mock()
        mock()
        mock.method().attribute.method()
        mock.attribute.method()

        # mock_calls: inclusive, recursive; split by '()', good for callable mocks.
        self.assertEqual(mock.mock_calls, [
            call(),
            call.method(),
            call.method().attribute.method(),
            call.attribute.method()
        ])

        # method_calls: exclusive, recursive; terminated by '()', good for non-callable mocks.
        self.assertEqual(
            mock.method_calls,
            [call.method(), call.attribute.method()])
Пример #6
0
    def test_internal_server_error(self):
        self.corba_object.method.side_effect = InternalServerError
        with self.assertRaises(InternalServerError):
            self.corba_client.method()

        self.assertEqual(self.corba_object.mock_calls, [call.method()])
        self.assertLogs(
            ['DEBUG', 'ERROR'],
            ('method()', 'method failed with .*InternalServerError'))
Пример #7
0
 def test_args_encoded_py2(self):
     # Test strings in arguments are encoded before corba is called
     self.corba_client.method('ěščřž', 'ýáíé')
     self.assertEqual(
         self.corba_object.mock_calls,
         [call.method('ěščřž'.encode('utf-8'), 'ýáíé'.encode('utf-8'))])
     self.assertIsInstance(self.corba_object.mock_calls[0][1][0],
                           six.binary_type)
     self.assertIsInstance(self.corba_object.mock_calls[0][1][1],
                           six.binary_type)
Пример #8
0
    def test_any_mock_calls_comparison_order(self):
        mock = Mock()
        class Foo(object):
            def __eq__(self, other): pass
            def __ne__(self, other): pass

        for d in datetime.now(), Foo():
            mock.reset_mock()

            mock(d, foo=d, bar=d)
            mock.method(d, zinga=d, alpha=d)
            mock().method(a1=d, z99=d)

            expected = [
                call(ANY, foo=ANY, bar=ANY),
                call.method(ANY, zinga=ANY, alpha=ANY),
                call(), call().method(a1=ANY, z99=ANY)
            ]
            self.assertEqual(expected, mock.mock_calls)
            self.assertEqual(mock.mock_calls, expected)
Пример #9
0
 def test_args_encoded(self):
     # Test strings in arguments are encoded before corba is called
     self.corba_client.method('ěščřž', 'ýáíé')
     self.assertEqual(self.corba_object.mock_calls,
                      [call.method('ěščřž', 'ýáíé')])
Пример #10
0
 def test_return_value(self):
     self.assertEqual(self.corba_client.method(), sentinel.result)
     self.assertEqual(self.corba_object.mock_calls, [call.method()])
Пример #11
0
 def test_call_args(self):
     self.corba_client.method(sentinel.arg, sentinel.other_arg)
     self.assertEqual(self.corba_object.mock_calls,
                      [call.method(sentinel.arg, sentinel.other_arg)])
Пример #12
0
    def test_call(self):
        self.corba_client.method()
        self.assertEqual(self.corba_object.mock_calls, [call.method()])

        self.assertLogs(['DEBUG', 'DEBUG'], ('method()', 'method returned'))
Пример #13
0
def testDownloadHttpd():
    downloadHttpd()
    expected = [call.method("curl -o httpd-2.4.2.tar.gz -L %s"%HTTPD_URL),
                call.method("tar xvzf httpd-2.4.2.tar.gz -C %s"%DOWNLOAD_DIR)]
    assert mock.mock_calls == expected