Forgot the socket error

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()