コード例 #1
0
 def test_get_file_event_client_calls_get_stored_value_with_expected_key(
         self, mock_session, session_factory, user_context,
         key_value_store_client):
     factory = MicroserviceClientFactory(TEST_ROOT_URL, mock_session,
                                         session_factory, user_context,
                                         key_value_store_client)
     factory.get_file_event_client()
     key_value_store_client.get_stored_value.assert_called_once_with(
         "FORENSIC_SEARCH-API_URL")
コード例 #2
0
 def test_get_file_event_client_calls_creates_client_with_expected_url(
         self, mock_session, session_factory, user_context,
         key_value_store_client):
     key_value_store_client.get_stored_value.return_value.text = FILE_EVENTS_URL
     factory = MicroserviceClientFactory(TEST_ROOT_URL, mock_session,
                                         session_factory, user_context,
                                         key_value_store_client)
     factory.get_file_event_client()
     session_factory.create_jwt_session.assert_called_once_with(
         FILE_EVENTS_URL, mock_session)
コード例 #3
0
 def test_get_file_event_client(self, mock_session, session_factory,
                                user_context, user_client):
     factory = MicroserviceClientFactory(TEST_ROOT_URL, mock_session,
                                         session_factory, user_context,
                                         user_client)
     client = factory.get_file_event_client()
     assert type(client) == file_event.FileEventClient
コード例 #4
0
    def test_get_file_event_client_returns_same_intance_on_multiple_calls(
            self, mock_session, session_factory, user_context):
        factory = MicroserviceClientFactory(TEST_ROOT_URL, mock_session,
                                            session_factory, user_context)
        client1 = factory.get_file_event_client()
        client2 = factory.get_file_event_client()

        assert client1 is client2
コード例 #5
0
    def test_get_file_event_client_when_no_sts_base_url_raises(
            self, mocker, mock_session, session_factory, user_context,
            user_client):
        def get(uri, *args, **kwargs):
            if uri == "/api/ServerEnv":
                mock_response = mocker.MagicMock(spec=Response)
                mock_response.text = "{}"
                return mock_response

        mock_session.get.side_effect = get
        factory = MicroserviceClientFactory(TEST_ROOT_URL, mock_session,
                                            session_factory, user_context,
                                            user_client)
        with pytest.raises(Py42FeatureUnavailableError) as err:
            factory.get_file_event_client()

        assert err.value.response