예제 #1
0
def test_get_row_id_by_date_returns_negative_one_on_exception(connection):
    connection.execute.side_effect = OperationalError()

    holiday = Holiday(connection)
    actual = holiday.get_row_id_by_date(datetime.now())

    assert actual is -1
예제 #2
0
def test_get_row_id_by_date_returns_row_id(connection):
    expected = randint(1,999)
    connection.fetchall.return_value = [(expected,)]

    holiday = Holiday(connection)
    actual = holiday.get_row_id_by_date(datetime.now())

    connection.execute.assert_called_once()
    assert actual is expected