def test_get_all_calls_expected_auditlogs_service(self, auditlog_service):
client = AuditLogsClient(auditlog_service)
for _ in client.get_all():
pass
auditlog_service.get_all.assert_called_once_with(
begin_time=None,
end_time=None,
event_types=None,
user_ids=None,
usernames=None,
user_ip_addresses=None,
affected_user_ids=None,
affected_usernames=None,
)
def test_get_all_passes_undefined_param_to_service(self, auditlog_service):
client = AuditLogsClient(auditlog_service)
for _ in client.get_all(customParam="abc"):
pass
auditlog_service.get_all.assert_called_once_with(
begin_time=None,
end_time=None,
event_types=None,
user_ids=None,
usernames=None,
user_ip_addresses=None,
affected_user_ids=None,
affected_usernames=None,
customParam="abc",
)
def test_get_page_calls_expected_auditlogs_service(self, auditlog_service):
client = AuditLogsClient(auditlog_service)
client.get_page()
auditlog_service.get_page.assert_called_once_with(
page_num=1,
page_size=None,
begin_time=None,
end_time=None,
event_types=None,
user_ids=None,
usernames=None,
user_ip_addresses=None,
affected_user_ids=None,
affected_usernames=None,
)
def _init_clients(services, connection):
# clients are imported within function to prevent circular imports when a client
# imports anything from py42.sdk.queries
from py42.clients import Clients
from py42.clients._archiveaccess.accessorfactory import ArchiveAccessorFactory
from py42.clients.alertrules import AlertRulesClient
from py42.clients.alerts import AlertsClient
from py42.clients.archive import ArchiveClient
from py42.clients.auditlogs import AuditLogsClient
from py42.clients.authority import AuthorityClient
from py42.clients.cases import CasesClient
from py42.clients.detectionlists import DetectionListsClient
from py42.clients.loginconfig import LoginConfigurationClient
from py42.clients.securitydata import SecurityDataClient
from py42.clients.trustedactivities import TrustedActivitiesClient
from py42.services.storage._service_factory import ConnectionManager
from py42.services.storage._service_factory import StorageServiceFactory
authority = AuthorityClient(
administration=services.administration,
archive=services.archive,
devices=services.devices,
legalhold=services.legalhold,
orgs=services.orgs,
users=services.users,
)
detectionlists = DetectionListsClient(services.userprofile,
services.departingemployee,
services.highriskemployee)
storage_service_factory = StorageServiceFactory(connection,
services.devices,
ConnectionManager())
alertrules = AlertRulesClient(services.alerts, services.alertrules)
securitydata = SecurityDataClient(
services.fileevents,
services.preservationdata,
services.savedsearch,
storage_service_factory,
)
alerts = AlertsClient(services.alerts, alertrules)
archive_accessor_factory = ArchiveAccessorFactory(services.archive,
storage_service_factory)
archive = ArchiveClient(archive_accessor_factory, services.archive)
auditlogs = AuditLogsClient(services.auditlogs)
loginconfig = LoginConfigurationClient(connection)
trustedactivities = TrustedActivitiesClient(services.trustedactivities)
clients = Clients(
authority=authority,
detectionlists=detectionlists,
alerts=alerts,
securitydata=securitydata,
archive=archive,
auditlogs=auditlogs,
cases=CasesClient(services.cases, services.casesfileevents),
loginconfig=loginconfig,
trustedactivities=trustedactivities,
)
return clients
def test_get_page_passes_undefined_param_to_auditlogs_service(
self, auditlog_service
):
client = AuditLogsClient(auditlog_service)
client.get_page(customParam="abc")
auditlog_service.get_page.assert_called_once_with(
page_num=1,
page_size=None,
begin_time=None,
end_time=None,
event_types=None,
user_ids=None,
usernames=None,
user_ip_addresses=None,
affected_user_ids=None,
affected_usernames=None,
customParam="abc",
)
def _init_clients(services, connection):
authority = AuthorityClient(
administration=services.administration,
archive=services.archive,
devices=services.devices,
legalhold=services.legalhold,
orgs=services.orgs,
securitydata=services.securitydata,
users=services.users,
)
detectionlists = DetectionListsClient(services.userprofile,
services.departingemployee,
services.highriskemployee)
storage_service_factory = StorageServiceFactory(connection,
services.devices,
ConnectionManager())
alertrules = AlertRulesClient(services.alerts, services.alertrules)
securitydata = SecurityDataClient(
services.securitydata,
services.fileevents,
services.preservationdata,
services.savedsearch,
storage_service_factory,
)
alerts = AlertsClient(services.alerts, alertrules)
archive_accessor_mgr = ArchiveAccessorManager(services.archive,
storage_service_factory)
archive = ArchiveClient(archive_accessor_mgr, services.archive)
auditlogs = AuditLogsClient(services.auditlogs)
clients = Clients(
authority=authority,
detectionlists=detectionlists,
alerts=alerts,
securitydata=securitydata,
archive=archive,
auditlogs=auditlogs,
)
return clients