"""Tests for VendorContract model."""

import datetime

from contracts.models import vendor_advance


def test_get_advances(db_fixture, monkeypatch):
    """Test advances."""
    vendor_id = 1
    expected_response = [
        (2063, 300000.0, datetime.date(2016, 5, 2), 1.0),
        (2064, 1050000.0, datetime.date(2016, 6, 22), 1.0),
        (2065, 150000.0, datetime.date(2016, 6, 23), 1.0),
    ]

    response = vendor_advance.get_advances(vendor_id)

    assert response.message == expected_response