def test_get_all_user_security_events_when_multiple_plans_with_cursors_returned_calls_security_client_expected_number_of_times( self, mocker, security_client_two_plans_one_node, storage_client_factory, microservice_client_factory, ): mock_storage_client = mocker.MagicMock(spec=StorageClient) mock_storage_security_client = mocker.MagicMock( spec=StorageSecurityClient) mock_storage_client.securitydata = mock_storage_security_client response1 = mocker.MagicMock(spec=Py42Response) response1.text = '{"cursor": "1:1"}' response2 = mocker.MagicMock(spec=Py42Response) response2.text = "{}" mock_storage_security_client.get_plan_security_events.side_effect = [ response1, response2, response1, response2, ] storage_client_factory.from_plan_info.return_value = mock_storage_client security_module = SecurityModule(security_client_two_plans_one_node, storage_client_factory, microservice_client_factory) for _, _ in security_module.get_all_user_security_events("foo"): pass assert mock_storage_security_client.get_plan_security_events.call_count == 4
def test_get_all_user_security_events_calls_security_client_with_expected_params( self, mocker, security_client_one_location, storage_client_factory, microservice_client_factory, ): mock_storage_client = mocker.MagicMock(spec=StorageClient) mock_storage_security_client = mocker.MagicMock( spec=StorageSecurityClient) mock_storage_client.securitydata = mock_storage_security_client response = mocker.MagicMock(spec=Py42Response) response.text = "{}" mock_storage_security_client.get_plan_security_events.return_value = response storage_client_factory.from_plan_info.return_value = mock_storage_client security_module = SecurityModule(security_client_one_location, storage_client_factory, microservice_client_factory) for _, _ in security_module.get_all_user_security_events("foo"): pass mock_storage_security_client.get_plan_security_events.assert_called_once_with( "111111111111111111", cursor=None, event_types=None, include_files=True, max_timestamp=None, min_timestamp=None, )