def test_heartbeat_return_true_if_token_server_is_working(self): with mock.patch('syncto.heartbeat.requests.get') as mocked_request: mocked_request.return_value.raise_for_status.return_value = None request = mock.MagicMock() request.registry.settings = { 'token_server_url': 'https://example.com', 'token_server_heartbeat_timeout_seconds': 5 } self.assertTrue(ping_sync_cluster(request)) mocked_request.assert_called_with( 'https://example.com/__heartbeat__', timeout=5)
def test_heartbeat_return_false_if_token_server_outofservice(self): with mock.patch('syncto.heartbeat.requests.get') as mocked_request: mocked_request.return_value.raise_for_status.side_effect = ( HTTPError()) request = mock.MagicMock() request.registry.settings = { 'token_server_url': 'https://example.com', 'token_server_heartbeat_timeout_seconds': 5 } self.assertFalse(ping_sync_cluster(request)) mocked_request.assert_called_with( 'https://example.com/__heartbeat__', timeout=5)