mirror
/
ogn-python
kopia lustrzana https://github.com/glidernet/ogn-python
1
0
Forkuj 0
Kod Zgłoszenia Projekty Wydania Wiki Aktywność

Forgot the socket error

pull/7/head
Konstantin Gründger 2015-11-21 23:17:15 +01:00
rodzic 92d2fa084d
commit d1a4fdecfc
1 zmienionych plików z 6 dodań i 3 usunięć
Pokaż statystyki Ściągnij plik aktualizacji Ściągnij plik porównania Expand all files Collapse all files

9
tests/gateway/test_manage.py
Wyświetl plik

@ -1,6 +1,8 @@
import unittest import unittest
import unittest.mock as mock import unittest.mock as mock
import socket
from ogn.gateway.manage import run from ogn.gateway.manage import run
@ -23,13 +25,14 @@ class GatewayTest(unittest.TestCase):
@mock.patch('ogn.gateway.manage.ognGateway') @mock.patch('ogn.gateway.manage.ognGateway')
def test_BrokenPipeError(self, mock_gateway): def test_BrokenPipeError(self, mock_gateway):
instance = mock_gateway.return_value instance = mock_gateway.return_value
instance.run.side_effect = [BrokenPipeError(), BrokenPipeError(), KeyboardInterrupt()] instance.run.side_effect = [BrokenPipeError(), socket.error(), KeyboardInterrupt()]
run("user_2") run("user_2")
instance.connect_db.assert_called_once_with() instance.connect_db.assert_called_once_with()
self.assertTrue(instance.run.call_count, 3) self.assertEqual(instance.connect.call_count, 3)
self.assertTrue(instance.disconnect.call_count, 2) # not called if socket crashed self.assertEqual(instance.run.call_count, 3)
self.assertEqual(instance.disconnect.call_count, 2) # not called if socket crashed
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()