import pytest from labelaudit import presentation @pytest.fixture def vapi_audits_data(vapi_audit_data): return { 'reports': [vapi_audit_data], 'total_reports': 1 } @pytest.fixture def vapi_audit_data(): return { 'auditStatus': 'complete', 'initiatedById': 123, 'reportLocation': 'over there', 'updatedTimestamp': '1901-07-23 01:23:45', 'vendorId': 456, 'youtubeAuditId': 123456, 'createdTimestamp': '1849-11-05 11:33:22' } def test_failure(): errors = {'foo': 'was bad', 'bar': 'was even worse'} data = presentation.failure(errors) assert data == {'errors': errors} def test_label_audit(vapi_audit_data): source = vapi_audit_data output = presentation.label_audit(source) assert output.get('audit_status') == source.get('auditStatus') assert output.get('youtube_audit_id') == source.get('youtubeAuditId') assert output.get('vendor_id') == source.get('vendorId') assert output.get('initiated_by_id') == source.get('initiatedById') assert output.get('report_location') == source.get('reportLocation') assert output.get('updated_timestamp') == source.get('updatedTimestamp') assert output.get('created_timestamp') == source.get('createdTimestamp') assert output.get('initiated_by_login') == source.get('initiatedByLogin') assert len(output) == 8 def test_label_audits_data(vapi_audits_data): output = presentation.label_audits(vapi_audits_data) audit = output.get('audits')[0] source = vapi_audits_data.get('reports')[0] assert audit.get('audit_status') == source.get('auditStatus') assert audit.get('youtube_audit_id') == source.get('youtubeAuditId') assert audit.get('vendor_id') == source.get('vendorId') assert audit.get('initiated_by_id') == source.get('initiatedById') assert audit.get('report_location') == source.get('reportLocation') assert audit.get('updated_timestamp') == source.get('updatedTimestamp') assert audit.get('created_timestamp') == source.get('createdTimestamp') assert output.get('initiated_by_login') == source.get('initiatedByLogin') assert len(audit) == 8 def test_label_audits_total(vapi_audits_data): output = presentation.label_audits(vapi_audits_data) assert output.get('total_records') == vapi_audits_data.get('total_reports')