"""Test for Vendor model."""

import pytest

from carveouts.exceptions import VendorInvalid
from carveouts.models import vendor


def test_for_valid_vendor(db_fixture: None) -> None:
    """Test to check valid vendor id."""
    result = vendor.get_vendor_id(7123)

    assert result == 7123


def test_for_invalid_vendor(db_fixture: None) -> None:
    """Test to check invalid vendor id."""
    with pytest.raises(VendorInvalid) as e:
        vendor.get_vendor_id(2)
    assert str(e.value) == "400 Bad Request: Vendor ID 2 is invalid"