def test_query_max_results():
query = Query()
query.set_table("Movies")
query.set_max_results(2000)
cmd = query.to_sql_query()
assert cmd == "SELECT * FROM boxofficementat.Movies LIMIT 0, 2000"
def test_query_max_results_and_offset():
query = Query()
query.set_table("Movies")
query.set_results_offset(200)
query.set_max_results(300)
cmd = query.to_sql_query()
assert cmd == "SELECT * FROM boxofficementat.Movies LIMIT 200, 300"
def test_inner_join_query():
query = Query()
query.set_table("Movies")
query.set_return_columns(["Studios.Name"])
query.add_inner_join('Studio', 'Studios', 'Id')
query.add_aggregate_column(AggregateType.COUNT, 'Movies.Id', True)
query.set_order_by_columns(['COUNT'])
query.set_results_order(ResultsOrder.DESC)
query.set_results_offset(80)
query.set_max_results(20)
cmd = query.to_sql_query()
assert cmd == "SELECT Studios.Name,COUNT(DISTINCT Movies.Id) AS COUNT FROM boxofficementat.Movies " \
"INNER JOIN boxofficementat.Studios ON boxofficementat.Movies.Studio=boxofficementat.Studios.Id " \
"GROUP BY Studios.Name ORDER BY COUNT DESC LIMIT 80, 20"