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
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