"""Report fixtures for testing.""" from collections import namedtuple from datetime import datetime from collaborator.models.rds.report import Report from collaborator.models.rds.report_run import ReportRun report_payload = { "report_id": None, "report_run_id": 1, "collaborator_id": 1, "exclude_transaction_types": None, "filename": "test file name", "file_location": "s3://location/666", "amount": 100.00, "currency": "USD", "status": "REQUESTED", "requested_datetime": None, "generated_datetime": None, "deleted_by": None, "deleted_datetime": None, "transaction_id": 444, "collaborator_dp_enabled": False, } report_to_dict = { "id": None, "report_run_id": 1, "collaborator_id": 1, "exclude_transaction_types": None, "filename": "test file name", "file_location": "s3://location/666", "amount": 100.00, "currency": "USD", "status": "REQUESTED", "requested_datetime": None, "generated_datetime": None, "deleted_by": None, "deleted_datetime": None, "transaction_id": 444, "collaborator_dp_enabled": False, } created_report_run = ReportRun( report_run_id=6, name="My great report run", uuid="969cf84b-14c7-48bb-940a-a5a2fa1ee101", period_ids="4,5,6", period_name="Q2 2005", file_format="xls", requested_datetime=datetime.fromisoformat("2019-10-18T15:50:12+00:00"), source="LEGACY", requestor_identity_uuid="a9ead030-209d-48c1-929e-5f4da58e5f2e", notification_email="myguy@sonymusic-pde.com", number_format="us", trigger_type="MANUAL", ) created_reports = [ Report( amount=None, collaborator_id=1, currency="USD", exclude_transaction_types=None, file_location=None, filename="Mygreatreportrun_Test_Q22005_20191018", generated_datetime=None, report_id=9, report_run_id=6, requested_datetime=datetime.fromisoformat("2019-10-18T15:50:12+00:00"), status="REQUESTED", transaction_id=None, deleted_by=None, deleted_datetime=None, collaborator_dp_enabled=False, ), Report( amount=None, collaborator_id=2, currency="USD", exclude_transaction_types=None, file_location=None, filename="Mygreatreportrun_Person_Q22005_20191018", generated_datetime=None, report_id=10, report_run_id=6, requested_datetime=datetime.fromisoformat("2019-10-18T15:50:12+00:00"), status="REQUESTED", transaction_id=None, deleted_by=None, deleted_datetime=None, collaborator_dp_enabled=False, ), ] ReportRunsResult = namedtuple( "ReportRunsResult", [ "uuid", "name", "currency", "period_ids", "period_name", "requested_datetime", "trigger_type", "amount", "total_count", "error_count", "requested_count", "generated_count", "transaction_count", ], ) report_runs_results = [ ReportRunsResult( uuid="1234-abcd", name="Test Run 2", currency=None, period_ids="1,2,3", period_name="Sometime", requested_datetime="2019-10-09T08:07:06", trigger_type="MANUAL", amount=None, total_count=1, error_count=0, requested_count=1, generated_count=0, transaction_count=0, ), ReportRunsResult( uuid="abcd-1234", name="Test Run", currency="NOK", period_ids="1,2,3", period_name="Sometime", requested_datetime="2019-09-08T07:06:05", trigger_type="AUTO", amount=5555.0, total_count=2, error_count=0, requested_count=0, generated_count=2, transaction_count=1, ), ]