"""Test for VendorAgreement model."""

import pytest

from collaborator.models.rds.vendor_agreement import VendorAgreement


@pytest.mark.parametrize(
    "fields, expected",
    [
        (
            {
                "vendor_agreement_id": 1,
                "vendor_id": 123,
                "date_accepted": "2000-12-11T10:09:08",
                "user_id": "81b5e5b0-e7d8-4112-aad9-b9189d53dcb2",
            },
            {
                "id": 1,
                "vendor_id": 123,
                "date_accepted": "2000-12-11T10:09:08",
                "user_id": "81b5e5b0-e7d8-4112-aad9-b9189d53dcb2",
            },
        ),
    ],
)
def test_to_dict(fields, expected):
    """Test getting dictionary of model fields."""
    item = VendorAgreement(**fields)

    assert item.to_dict() == expected