from labelaudit.models import label_audit

import pytest


@pytest.fixture()
def vapi_data():
    return {
        'youtubeAuditId': 123,
        'reportLocation': 'over-there',
        'auditStatus': 'in_progress',
        'vendorId': '7123',
        'initiatedById': 456,
        'updatedTimestamp': '2015-06-29 13:33:11'
    }


def test_from_vapi(vapi_data):
    audit = label_audit.from_vapi(vapi_data)
    assert audit.youtube_audit_id == vapi_data.get('youtubeAuditId')
    assert audit.report_location == vapi_data.get('reportLocation')
    assert audit.audit_status == vapi_data.get('auditStatus')
    assert audit.vendor_id == int(vapi_data.get('vendorId'))
    assert audit.initiated_by_id == vapi_data.get('initiatedById')
    assert audit.updated_timestamp == vapi_data.get('updatedTimestamp')