"""Test for Vendor Model.""" from oto import status as response_code # from backend.constants import error from backend.constants import vendor_field as vf from backend.models.vendor import VendorPersister from tests.testutils import db @db.test_schema def test_get_vendor_details(client, test_vendor): """Test getting track is successful.""" print(test_vendor) response = VendorPersister.get_by_vendor_id(test_vendor[vf.VENDOR_ID]) assert response.status == response_code.OK print(response.message) vendor = response.message for field in test_vendor: assert vendor[field] == test_vendor[field] @db.test_schema def test_get_by_invalid_tuid(client): """Test getting track by invalid tuid.""" response = VendorPersister.get_by_vendor_id(0) assert response.status == response_code.NOT_FOUND