예제 #1
0
def test_add_holiday_returns_negative_one_on_exception_thrown(connection):
    connection.commit.side_effect = OperationalError()

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

    assert actual is -1
예제 #2
0
def test_add_holiday_returns_row_id(connection):
    expected = randint(1, 999)
    connection.get_last_row_id.return_value = expected

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

    connection.execute.assert_called_once()
    connection.commit.assert_called_once()
    connection.get_last_row_id.assert_called_once()
    assert actual is expected