def test_append_two_items(self):
builder = QueryBuilder()
builder.set_schema.add("SET search_path TO %s,public;", ["hg38"])
builder.query.add("select 1 from table where value = %s;", [5])
query, params = builder.get_query_and_params()
expected_sql = "SET search_path TO %s,public;\nselect 1 from table where value = %s;"
expected_params = ["hg38", 5]
self.assertEqual(expected_sql, query)
self.assertEqual(expected_params, params)
def test_append_two_items(self):
builder = QueryBuilder()
builder.set_schema.add("SET search_path TO %s,public;", ["hg38"])
builder.query.add("select 1 from table where value = %s;", [5])
query, params = builder.get_query_and_params()
expected_sql = "SET search_path TO %s,public;\nselect 1 from table where value = %s;"
expected_params = ["hg38", 5]
self.assertEqual(expected_sql, query)
self.assertEqual(expected_params, params)
def get_query_and_params(self):
builder = QueryBuilder()
builder.set_schema.add_part(set_search_path(self.schema))
if self.count:
builder.query.add_part(begin_count())
builder.query.add_parts(self.main_query_parts())
if self.count:
builder.query.add_part(end_count())
return builder.get_query_and_params()
def get_query_and_params(self):
builder = QueryBuilder()
builder.set_schema.add_part(set_search_path(self.schema))
if self.count:
builder.query.add_part(begin_count())
builder.query.add_parts(self.main_query_parts())
if self.count:
builder.query.add_part(end_count())
return builder.get_query_and_params()
def get_query_and_params(self):
builder = QueryBuilder()
builder.set_schema.add_part(set_search_path(self.schema))
builder.query.add_parts([
items_not_in_gene_list(self.custom_list_id,
self.custom_list_filter,
self.custom_gene_name_type)
])
return builder.get_query_and_params()
def test_all_items(self):
builder = QueryBuilder()
builder.set_schema.add_part(set_search_path("hg38"))
builder.with_clause.add("WITH regional_sales AS (SELECT amount AS total_sales FROM orders)", [])
builder.query.add("SELECT * from orders where total_sales > %s;", [200])
query, params = builder.get_query_and_params()
expected_sql = """SET search_path TO %s,public;
WITH regional_sales AS (SELECT amount AS total_sales FROM orders)
SELECT * from orders where total_sales > %s;"""
expected_params = ["hg38", 200]
self.assertEqual(expected_sql, query)
self.assertEqual(expected_params, params)
def test_all_items(self):
builder = QueryBuilder()
builder.set_schema.add_part(set_search_path("hg38"))
builder.with_clause.add(
"WITH regional_sales AS (SELECT amount AS total_sales FROM orders)",
[])
builder.query.add("SELECT * from orders where total_sales > %s;",
[200])
query, params = builder.get_query_and_params()
expected_sql = """SET search_path TO %s,public;
WITH regional_sales AS (SELECT amount AS total_sales FROM orders)
SELECT * from orders where total_sales > %s;"""
expected_params = ["hg38", 200]
self.assertEqual(expected_sql, query)
self.assertEqual(expected_params, params)
def test_starts_blank(self):
builder = QueryBuilder()
query, params = builder.get_query_and_params()
self.assertEqual("", query)
self.assertEqual([], params)
def test_simple_query(self):
builder = QueryBuilder()
builder.query.add("select 1 from table where value = %s;", [5])
query, params = builder.get_query_and_params()
self.assertEqual("select 1 from table where value = %s;", query)
self.assertEqual([5], params)
def test_starts_blank(self):
builder = QueryBuilder()
query, params = builder.get_query_and_params()
self.assertEqual("", query)
self.assertEqual([], params)
def test_simple_query(self):
builder = QueryBuilder()
builder.query.add("select 1 from table where value = %s;", [5])
query, params = builder.get_query_and_params()
self.assertEqual("select 1 from table where value = %s;", query)
self.assertEqual([5], params)