"""Tests for mech admin."""

import datetime
from unittest.mock import MagicMock

import pytest

from contracts import response
from contracts.logic import advances


@pytest.fixture
def fixture_advances():
    """Fixture for mechadmin vendor."""
    return response.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),
        ]
    )


def test_get_advances(monkeypatch, fixture_advances):
    """Test gettting advances."""
    vendor_id = 123
    monkeypatch.setattr(
        advances.vendor_advance,
        'get_advances',
        MagicMock(return_value=fixture_advances),
    )

    result = advances.get_advances(vendor_id)

    # Assertions
    assert advances.vendor_advance.get_advances.called
    assert result.message == fixture_advances.message