def test_connect_scheduler_with_bypass_leader_redirect(self, mock_transport): mock_transport.return_value.open.side_effect = [TTransport.TTransportException, True] mock_time = mock.create_autospec(spec=time, instance=True) auth = mock_auth() user_agent = 'Some-User-Agent' client = scheduler_client.SchedulerClient( auth, user_agent, verbose=True, bypass_leader_redirect=True) uri = 'https://scheduler.example.com:1337' client._connect_scheduler(uri, mock_time) mock_transport.assert_called_once_with( uri, auth=auth.auth(), user_agent=user_agent, session_factory=mock.ANY) _, _, kwargs = mock_transport.mock_calls[0] session = kwargs['session_factory']() assert session.headers[BYPASS_LEADER_REDIRECT_HEADER_NAME] == 'true'
def test_connect_scheduler(self, mock_client): mock_client.return_value.open.side_effect = [TTransport.TTransportException, True] mock_time = mock.create_autospec(spec=time, instance=True) client = scheduler_client.SchedulerClient(mock_auth(), 'Some-User-Agent', verbose=True) client._connect_scheduler('https://scheduler.example.com:1337', mock_time) assert mock_client.return_value.open.has_calls(mock.call(), mock.call()) mock_time.sleep.assert_called_once_with( scheduler_client.SchedulerClient.RETRY_TIMEOUT.as_(Time.SECONDS))
def test_connect_scheduler_with_user_agent(self, mock_transport): mock_transport.return_value.open.side_effect = [TTransport.TTransportException, True] mock_time = mock.create_autospec(spec=time, instance=True) auth = mock_auth() user_agent = 'Some-User-Agent' client = scheduler_client.SchedulerClient(auth, user_agent, verbose=True) uri = 'https://scheduler.example.com:1337' client._connect_scheduler(uri, mock_time) mock_transport.assert_called_once_with(uri, auth=auth.auth(), user_agent=user_agent)
def test__internal_connect(self): client = scheduler_client.SchedulerClient(mock_auth(), 'Some-User-Agent', verbose=True) self.assertIsNone(client._connect())