def test_get_page_posts_data_to_expected_url( self, mock_connection, user_context, mock_get_all_cases_response, mock_detection_list_user_client, ): client = DepartingEmployeeService( mock_connection, user_context, mock_detection_list_user_client ) client.get_page( filter_type="OPEN", sort_key="CREATED_AT", sort_direction="DESC", page_num=1, page_size=100, ) mock_connection.post.return_value = mock_get_all_cases_response first_call = mock_connection.post.call_args_list[0] posted_data = first_call[1]["json"] assert ( posted_data["tenantId"] == TENANT_ID_FROM_RESPONSE and posted_data["pgSize"] == 100 and posted_data["pgNum"] == 1 and posted_data["filterType"] == "OPEN" and posted_data["srtKey"] == "CREATED_AT" and posted_data["srtDirection"] == "DESC" ) assert ( mock_connection.post.call_args[0][0] == "/svc/api/v2/departingemployee/search" ) assert mock_connection.post.call_count == 1
def test_get_all_posts_expected_data_with_non_default_values( self, user_context, mock_connection, mock_detection_list_user_client ): client = DepartingEmployeeService( mock_connection, user_context, mock_detection_list_user_client ) for _ in client.get_all( filter_type="NEW_FILTER", sort_direction="DESC", sort_key="DISPLAY_NAME", page_size=200, ): break posted_data = mock_connection.post.call_args[1]["json"] assert mock_connection.post.call_count == 1 assert ( mock_connection.post.call_args[0][0] == "/svc/api/v2/departingemployee/search" ) assert ( posted_data["tenantId"] == user_context.get_current_tenant_id() and posted_data["filterType"] == "NEW_FILTER" and posted_data["pgNum"] == 1 and posted_data["pgSize"] == 200 and posted_data["srtKey"] == "DISPLAY_NAME" and posted_data["srtDirection"] == "DESC" )
def test_get_all_posts_expected_data_to_expected_url( self, mock_connection, user_context, mock_get_all_response, mock_detection_list_user_client, ): client = DepartingEmployeeService( mock_connection, user_context, mock_detection_list_user_client ) mock_connection.post.return_value = mock_get_all_response for _ in client.get_all(): break first_call = mock_connection.post.call_args_list[0] posted_data = first_call[1]["json"] assert ( posted_data["tenantId"] == TENANT_ID_FROM_RESPONSE and posted_data["pgSize"] == 100 and posted_data["pgNum"] == 1 and posted_data["filterType"] == "OPEN" and posted_data["srtKey"] == "CREATED_AT" and posted_data["srtDirection"] == "DESC" ) assert mock_connection.post.call_args[0][0] == "v2/departingemployee/search" assert mock_connection.post.call_count == 1
def test_add_posts_expected_data_and_to_expected_url( self, mock_connection, user_context, mock_get_all_cases_response, mock_detection_list_user_client, departing_date, ): client = DepartingEmployeeService( mock_connection, user_context, mock_detection_list_user_client ) user_context.get_current_tenant_id.return_value = _TENANT_ID_PARAM # Return value should have been set based on the arguments passed # in add, here however as we are mocking it, it doesn't matter. Can be refactored mock_connection.post.return_value = mock_get_all_cases_response client.add(_USER_ID, departing_date) # Have to convert the request data to a dict because # older versions of Python don't have deterministic order. posted_data = mock_connection.post.call_args[1]["json"] assert ( posted_data["userId"] == _USER_ID and posted_data["tenantId"] == _TENANT_ID_PARAM and posted_data["departureDate"] == "2022-12-20" ) assert ( mock_connection.post.call_args[0][0] == "/svc/api/v2/departingemployee/add" ) assert mock_connection.post.call_count == 2
def test_update_posts_to_expected_url(self, mock_connection, user_context, mock_detection_list_user_client): client = DepartingEmployeeService(mock_connection, user_context, mock_detection_list_user_client) client.update_departure_date(_USER_ID, "2022-12-20") assert mock_connection.post.call_args[0][ 0] == "v2/departingemployee/update"
def test_set_alerts_enabled_posts_to_expected_url( self, mock_connection, user_context, mock_get_all_response_empty, mock_detection_list_user_client, ): client = DepartingEmployeeService(mock_connection, user_context, mock_detection_list_user_client) mock_connection.post.return_value = mock_get_all_response_empty client.set_alerts_enabled() assert (mock_connection.post.call_args[0][0] == "v2/departingemployee/setalertstate")
def test_remove_raises_error_when_user_id_does_not_exist( self, user_context, mock_post_not_found_session, mock_detection_list_user_client, ): departing_employee_client = DepartingEmployeeService( mock_post_not_found_session, user_context, mock_detection_list_user_client) user_id = "942897397520289999" with pytest.raises(Py42UserNotOnListError) as err: departing_employee_client.remove(user_id) assert "User with ID '{}' is not currently on the departing-employee list.".format( user_id) in str(err.value)
def test_get_posts_to_expected_url( self, mock_connection, user_context, mock_get_all_cases_response_empty, mock_detection_list_user_client, ): client = DepartingEmployeeService( mock_connection, user_context, mock_detection_list_user_client ) mock_connection.post.return_value = mock_get_all_cases_response_empty client.get("999") assert ( mock_connection.post.call_args[0][0] == "/svc/api/v2/departingemployee/get" )
def test_set_alerts_enabled_posts_expected_data( self, mock_connection, user_context, mock_get_all_response_empty, mock_detection_list_user_client, ): client = DepartingEmployeeService(mock_connection, user_context, mock_detection_list_user_client) mock_connection.post.return_value = mock_get_all_response_empty client.set_alerts_enabled() posted_data = mock_connection.post.call_args[1]["json"] assert (posted_data["tenantId"] == TENANT_ID_FROM_RESPONSE and posted_data["alertsEnabled"] is True)
def test_add_when_user_already_on_list_raises_user_already_added_error( self, mocker, mock_connection, user_context, mock_detection_list_user_client): def side_effect(url, json): if "add" in url: raise create_mock_error(Py42BadRequestError, mocker, "User already on list") mock_connection.post.side_effect = side_effect client = DepartingEmployeeService(mock_connection, user_context, mock_detection_list_user_client) with pytest.raises(Py42UserAlreadyAddedError) as err: client.add("user_id") expected = "User with ID user_id is already on the departing-employee list." assert str(err.value) == expected
def _init_services(main_connection, main_auth): alert_rules_key = u"FedObserver-API_URL" alerts_key = u"AlertService-API_URL" file_events_key = u"FORENSIC_SEARCH-API_URL" preservation_data_key = u"PRESERVATION-DATA-SERVICE_API-URL" employee_case_mgmt_key = u"employeecasemanagement-API_URL" kv_prefix = u"simple-key-value-store" audit_logs_key = u"AUDIT-LOG_API-URL" kv_connection = Connection.from_microservice_prefix( main_connection, kv_prefix) kv_service = KeyValueStoreService(kv_connection) alert_rules_conn = Connection.from_microservice_key(kv_service, alert_rules_key, auth=main_auth) alerts_conn = Connection.from_microservice_key(kv_service, alerts_key, auth=main_auth) file_events_conn = Connection.from_microservice_key(kv_service, file_events_key, auth=main_auth) pds_conn = Connection.from_microservice_key(kv_service, preservation_data_key, auth=main_auth) ecm_conn = Connection.from_microservice_key(kv_service, employee_case_mgmt_key, auth=main_auth) audit_logs_conn = Connection.from_microservice_key(kv_service, audit_logs_key, auth=main_auth) user_svc = UserService(main_connection) administration_svc = AdministrationService(main_connection) file_event_svc = FileEventService(file_events_conn) user_ctx = UserContext(administration_svc) user_profile_svc = DetectionListUserService(ecm_conn, user_ctx, user_svc) services = Services( administration=administration_svc, archive=ArchiveService(main_connection), devices=DeviceService(main_connection), legalhold=LegalHoldService(main_connection), orgs=OrgService(main_connection), securitydata=SecurityDataService(main_connection), users=UserService(main_connection), alertrules=AlertRulesService(alert_rules_conn, user_ctx, user_profile_svc), alerts=AlertService(alerts_conn, user_ctx), fileevents=file_event_svc, savedsearch=SavedSearchService(file_events_conn, file_event_svc), preservationdata=PreservationDataService(pds_conn), departingemployee=DepartingEmployeeService(ecm_conn, user_ctx, user_profile_svc), highriskemployee=HighRiskEmployeeService(ecm_conn, user_ctx, user_profile_svc), userprofile=user_profile_svc, auditlogs=AuditLogsService(audit_logs_conn), ) return services, user_ctx
def test_get_posts_expected_data_to_expected_url( self, mock_connection, user_context, mock_get_all_response_empty, mock_detection_list_user_client, ): client = DepartingEmployeeService(mock_connection, user_context, mock_detection_list_user_client) mock_connection.post.return_value = mock_get_all_response_empty client.get("999") posted_data = mock_connection.post.call_args[1]["json"] assert mock_connection.post.call_args[0][ 0] == "v2/departingemployee/get" assert (posted_data["tenantId"] == TENANT_ID_FROM_RESPONSE and posted_data["userId"] == "999")
def test_update_posts_expected_data( self, mock_connection, user_context, mock_get_all_response, mock_detection_list_user_client, ): client = DepartingEmployeeService(mock_connection, user_context, mock_detection_list_user_client) mock_connection.post.return_value = mock_get_all_response client.update_departure_date(_USER_ID, "2020-12-20") # Have to convert the request data to a dict because # older versions of Python don't have deterministic order. posted_data = mock_connection.post.call_args[1]["json"] assert (posted_data["userId"] == _USER_ID and posted_data["tenantId"] == TENANT_ID_FROM_RESPONSE and posted_data["departureDate"] == "2020-12-20")
def test_add_when_user_already_on_list_raises_user_already_added_error( self, mocker, mock_connection, user_context, mock_detection_list_user_client): def side_effect(url, json): if "add" in url: base_err = mocker.MagicMock(spec=HTTPError) base_err.response = mocker.MagicMock(spec=Response) base_err.response.text = "User already on list" raise Py42BadRequestError(base_err) mock_connection.post.side_effect = side_effect client = DepartingEmployeeService(mock_connection, user_context, mock_detection_list_user_client) with pytest.raises(Py42UserAlreadyAddedError) as err: client.add("user_id") expected = "User with ID user_id is already on the departing-employee list." assert str(err.value) == expected
def test_remove_posts_expected_data_and_to_expected_url( self, mock_connection, user_context, mock_get_all_response_empty, mock_detection_list_user_client, ): client = DepartingEmployeeService(mock_connection, user_context, mock_detection_list_user_client) mock_connection.post.return_value = mock_get_all_response_empty client.remove("999") # Have to convert the request data to a dict because # older versions of Python don't have deterministic order. posted_data = mock_connection.post.call_args[1]["json"] assert (posted_data["userId"] == "999" and posted_data["tenantId"] == TENANT_ID_FROM_RESPONSE) assert mock_connection.post.call_args[0][ 0] == "v2/departingemployee/remove"
def _init_services(main_connection, main_auth): # services are imported within function to prevent circular imports when a service # imports anything from py42.sdk.queries from py42.services import Services from py42.services._keyvaluestore import KeyValueStoreService from py42.services.administration import AdministrationService from py42.services.alertrules import AlertRulesService from py42.services.alerts import AlertService from py42.services.archive import ArchiveService from py42.services.auditlogs import AuditLogsService from py42.services.cases import CasesService from py42.services.casesfileevents import CasesFileEventsService from py42.services.detectionlists.departing_employee import DepartingEmployeeService from py42.services.detectionlists.high_risk_employee import HighRiskEmployeeService from py42.services.detectionlists.user_profile import DetectionListUserService from py42.services.devices import DeviceService from py42.services.fileevent import FileEventService from py42.services.legalhold import LegalHoldService from py42.services.orgs import OrgService from py42.services.preservationdata import PreservationDataService from py42.services.savedsearch import SavedSearchService from py42.services.trustedactivities import TrustedActivitiesService from py42.services.users import UserService alert_rules_key = "FedObserver-API_URL" alerts_key = "AlertService-API_URL" file_events_key = "FORENSIC_SEARCH-API_URL" preservation_data_key = "PRESERVATION-DATA-SERVICE_API-URL" employee_case_mgmt_key = "employeecasemanagementV2-API_URL" kv_prefix = "simple-key-value-store" audit_logs_key = "AUDIT-LOG_API-URL" cases_key = "CASES_API-URL" trusted_activities_key = "TRUSTED-DOMAINS_API-URL" kv_connection = Connection.from_microservice_prefix( main_connection, kv_prefix) kv_service = KeyValueStoreService(kv_connection) alert_rules_conn = Connection.from_microservice_key(kv_service, alert_rules_key, auth=main_auth) alerts_conn = Connection.from_microservice_key(kv_service, alerts_key, auth=main_auth) file_events_conn = Connection.from_microservice_key(kv_service, file_events_key, auth=main_auth) pds_conn = Connection.from_microservice_key(kv_service, preservation_data_key, auth=main_auth) ecm_conn = Connection.from_microservice_key(kv_service, employee_case_mgmt_key, auth=main_auth) audit_logs_conn = Connection.from_microservice_key(kv_service, audit_logs_key, auth=main_auth) user_svc = UserService(main_connection) administration_svc = AdministrationService(main_connection) file_event_svc = FileEventService(file_events_conn) user_ctx = UserContext(administration_svc) user_profile_svc = DetectionListUserService(ecm_conn, user_ctx, user_svc) cases_conn = Connection.from_microservice_key(kv_service, cases_key, auth=main_auth) trusted_activities_conn = Connection.from_microservice_key( kv_service, trusted_activities_key, auth=main_auth) services = Services( administration=administration_svc, archive=ArchiveService(main_connection), devices=DeviceService(main_connection), legalhold=LegalHoldService(main_connection), orgs=OrgService(main_connection), users=UserService(main_connection), alertrules=AlertRulesService(alert_rules_conn, user_ctx, user_profile_svc), alerts=AlertService(alerts_conn, user_ctx), fileevents=file_event_svc, savedsearch=SavedSearchService(file_events_conn, file_event_svc), preservationdata=PreservationDataService(pds_conn), departingemployee=DepartingEmployeeService(ecm_conn, user_ctx, user_profile_svc), highriskemployee=HighRiskEmployeeService(ecm_conn, user_ctx, user_profile_svc), userprofile=user_profile_svc, auditlogs=AuditLogsService(audit_logs_conn), cases=CasesService(cases_conn), casesfileevents=CasesFileEventsService(cases_conn), trustedactivities=TrustedActivitiesService(trusted_activities_conn), ) return services, user_ctx