def test_internet_is_ok_after_first_check(self):
# Given
connection_handler = ConnectionHandler(*fake_params)
# When
connection_handler.update()
# Then
self.assertTrue(connection_handler.internet_ok)
def test_have_internet_again(self):
# Given
connection_handler = ConnectionHandler(*fake_params)
connection_handler.internet_ok = False
# When
connection_handler.update()
# Then
self.assertTrue(connection_handler.internet_ok)
def __init__(self, parent=None):
super(SystemTray, self).__init__(parent)
self.tray_icon_menu = QMenu(self)
self.tray_icon = QSystemTrayIcon(self)
self.tray_icon.setContextMenu(self.tray_icon_menu)
self.icon_management = IconManagement(self.tray_icon)
self.connection_handler = ConnectionHandler(FREQUENCY_CHECK_MS,
TIME_OUT_CALL_S, self)
self.connection_handler.value_changed.connect(self.internet_connection)
self.connection_handler.start()
def test_when_no_internet(self):
# Given
connection_handler = ConnectionHandler(*fake_params)
connection_handler.check_internet_call = MagicMock(
side_effect=Exception('HTTP Error'))
# When
connection_handler.update()
# Then
self.assertFalse(connection_handler.internet_ok)
def test_internet_is_None_by_default(self):
# Given
connection_handler = ConnectionHandler(*fake_params)
# Then
self.assertIsNone(connection_handler.internet_ok)
