def test_get_departing_employee_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_departing_employee_client()
assert type(client) == DepartingEmployeeClient
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
def test_get_alerts_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_alerts_client()
assert type(client) == alerts.AlertClient
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
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")
def test_get_departing_employee_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_departing_employee_client()
key_value_store_client.get_stored_value.assert_called_once_with(
"employeecasemanagement-API_URL")
def test_get_departing_employee_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)
user_client = UserClient(mock_session)
client1 = factory.get_departing_employee_client(user_client)
client2 = factory.get_departing_employee_client(user_client)
assert client1 is client2
def test_get_departing_employee_client(self, mock_session, session_factory,
user_context):
factory = MicroserviceClientFactory(TEST_ROOT_URL, mock_session,
session_factory, user_context)
user_client = UserClient(mock_session)
client = factory.get_departing_employee_client(user_client)
assert type(
client
) == detectionlists.departing_employee.DepartingEmployeeClient
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)
def test_get_departing_employee_client_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 = DEPARTING_EMPLOYEE_URL
factory = MicroserviceClientFactory(TEST_ROOT_URL, mock_session,
session_factory, user_context,
key_value_store_client)
factory.get_departing_employee_client()
session_factory.create_jwt_session.assert_called_once_with(
DEPARTING_EMPLOYEE_URL, mock_session)
def test_create_storage_preservation_client(self, mock_session,
user_context, user_client,
session_factory):
factory = MicroserviceClientFactory(TEST_ROOT_URL, mock_session,
session_factory, user_context,
user_client)
factory.create_storage_preservation_client("https://host.com")
session_factory.create_jwt_session.assert_called_once_with(
"https://host.com", mock_session)
session_factory.create_anonymous_session.assert_called_once_with(
"https://host.com")
def test_get_alert_rules_client_when_no_sts_base_url_raises_Py42FeatureUnavailableError(
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_alert_rules_client()
assert err.value.response
def test_get_alerts_client_calls_get_stored_value_with_expected_key(
self,
mock_session,
session_factory,
user_context,
user_client,
key_value_store_client,
):
factory = MicroserviceClientFactory(
TEST_ROOT_URL,
mock_session,
session_factory,
user_context,
user_client,
key_value_store_client,
)
factory.get_alerts_client()
key_value_store_client.get_stored_value.assert_called_once_with(
"AlertService-API_URL")
def __init__(self, host_address, session_factory, root_session):
self._set_v3_session(host_address, session_factory, root_session)
# authority clients
authority_client_factory = AuthorityClientFactory(self.session)
self.administration_client = (
authority_client_factory.create_administration_client()
)
self.user_client = authority_client_factory.create_user_client()
self.device_client = authority_client_factory.create_device_client()
self.org_client = authority_client_factory.create_org_client()
self.legal_hold_client = authority_client_factory.create_legal_hold_client()
self.archive_client = authority_client_factory.create_archive_client()
self.security_client = authority_client_factory.create_security_client()
self.user_context = UserContext(self.administration_client)
archive_locator_factory = StorageTokenProviderFactory(
self.session, self.security_client, self.device_client
)
self.storage_client_factory = _get_storage_client_factory(
session_factory, archive_locator_factory
)
archive_accessor_manager = ArchiveAccessorManager(
self.archive_client, self.storage_client_factory
)
microservice_client_factory = MicroserviceClientFactory(
host_address,
root_session,
session_factory,
self.user_context,
self.user_client,
)
# modules (feature sets that combine info from multiple clients)
self.archive_module = archive_module.ArchiveModule(
archive_accessor_manager, self.archive_client
)
self.security_module = sec_module.SecurityModule(
self.security_client,
self.storage_client_factory,
microservice_client_factory,
)
self.detection_lists_module = detectionlists.DetectionListsModule(
microservice_client_factory
)
self.alerts_module = alerts.AlertsModule(microservice_client_factory)