From d1a4fdecfc9537b3cdd50da3c7c523cbb8d2df2b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Konstantin=20Gru=CC=88ndger?= Date: Sat, 21 Nov 2015 23:17:15 +0100 Subject: [PATCH] Forgot the socket error --- tests/gateway/test_manage.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/tests/gateway/test_manage.py b/tests/gateway/test_manage.py index 178673a..b3a4bb0 100644 --- a/tests/gateway/test_manage.py +++ b/tests/gateway/test_manage.py @@ -1,6 +1,8 @@ import unittest import unittest.mock as mock +import socket + from ogn.gateway.manage import run @@ -23,13 +25,14 @@ class GatewayTest(unittest.TestCase): @mock.patch('ogn.gateway.manage.ognGateway') def test_BrokenPipeError(self, mock_gateway): instance = mock_gateway.return_value - instance.run.side_effect = [BrokenPipeError(), BrokenPipeError(), KeyboardInterrupt()] + instance.run.side_effect = [BrokenPipeError(), socket.error(), KeyboardInterrupt()] run("user_2") instance.connect_db.assert_called_once_with() - self.assertTrue(instance.run.call_count, 3) - self.assertTrue(instance.disconnect.call_count, 2) # not called if socket crashed + self.assertEqual(instance.connect.call_count, 3) + self.assertEqual(instance.run.call_count, 3) + self.assertEqual(instance.disconnect.call_count, 2) # not called if socket crashed if __name__ == '__main__': unittest.main()