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