import pytest

from app.models.organization import Organization
from tests import organization_json


@pytest.mark.parametrize("purchase_order_number,expected_result", [
    [None, None],
    ["PO1234", [None, None, None, "PO1234"]]
])
def test_organization_billing_details(purchase_order_number, expected_result):
    organization = Organization(organization_json(purchase_order_number=purchase_order_number))
    assert organization.billing_details == expected_result