Exemplo n.º 1
0
 def test_assert_called_once_with(self):
     fail_code1 = """
            mock = Mock()
            mock.method(1, 2, 3, test='wow')
            mock.method.assert_called_once()
            """
     fail_code2 = """
            mock = Mock()
            mock.method(1, 2, 3, test='wow')
            mock.method.assertCalledOnceWith()
            """
     pass_code = """
            mock = Mock()
            mock.method(1, 2, 3, test='wow')
            mock.method.assert_called_once_with()
            """
     self.assertEqual(
         1,
         len(
             list(
                 checks.check_assert_called_once_with(
                     fail_code1, "neutron/tests/test_assert.py"))))
     self.assertEqual(
         1,
         len(
             list(
                 checks.check_assert_called_once_with(
                     fail_code2, "neutron/tests/test_assert.py"))))
     self.assertEqual(
         0,
         len(
             list(
                 checks.check_assert_called_once_with(
                     pass_code, "neutron/tests/test_assert.py"))))
Exemplo n.º 2
0
 def test_assert_called_once_with(self):
     fail_code1 = """
            mock = Mock()
            mock.method(1, 2, 3, test='wow')
            mock.method.assert_called_once()
            """
     fail_code2 = """
            mock = Mock()
            mock.method(1, 2, 3, test='wow')
            mock.method.assertCalledOnceWith()
            """
     pass_code = """
            mock = Mock()
            mock.method(1, 2, 3, test='wow')
            mock.method.assert_called_once_with()
            """
     self.assertEqual(
         1, len(list(checks.check_assert_called_once_with(fail_code1,
                                         "neutron/tests/test_assert.py"))))
     self.assertEqual(
         1, len(list(checks.check_assert_called_once_with(fail_code2,
                                         "neutron/tests/test_assert.py"))))
     self.assertEqual(
         0, len(list(checks.check_assert_called_once_with(pass_code,
                                         "neutron/tests/test_assert.py"))))