"""Test for subaccount model."""

import pytest

from carveouts.exceptions import SubaccountInvalid, VendorInvalid
from carveouts.models import subaccount


def test_for_valid_subaccount(db_fixture: None) -> None:
    """Test to check valid subaccount id."""
    result = subaccount.get_subaccount(1)

    assert result == (1, 7123)


def test_for_invalid_subaccount(db_fixture: None) -> None:
    """Test to check invalid subaccount id."""
    with pytest.raises(SubaccountInvalid) as e:
        subaccount.get_subaccount(123)
    assert str(e.value) == "400 Bad Request: Subaccount ID 123 is invalid"


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