def test_constructs_single_query_no_deps(self):
code = SqlCode("select company_name, company_id from companies", "companies")
repo = Repository()
repo.add_queries(code.queries)
qg = QueryGenerator(repo)
output = qg.generate_query("companies")
self.assertEqual(output.lower(), "select company_name, company_id from companies")
def test_repo_retrieves_dependencies_in_order(self):
testQuery = SqlCode("with companies as (select company_name, company_id) select company_name from companies", "test_query")
repo = Repository()
repo.add_queries(testQuery.queries)
queries = repo.retrieve_query_with_dependencies("test_query")
self.assertEqual(len(queries), 2)
self.assertTrue(testQuery.queries[0] == queries[0])
self.assertTrue(testQuery.queries[1] == queries[1])
def test_repo_retrieves_dependencies_in_order(self):
testQuery = SqlCode(
"with companies as (select company_name, company_id) select company_name from companies",
"test_query")
repo = Repository()
repo.add_queries(testQuery.queries)
queries = repo.retrieve_query_with_dependencies("test_query")
self.assertEqual(len(queries), 2)
self.assertTrue(testQuery.queries[0] == queries[0])
self.assertTrue(testQuery.queries[1] == queries[1])
def test_repo_keeps_dependencies(self):
testQuery = SqlCode(
"with companies as (select company_name, company_id) select company_name from companies",
"test_query")
repo = Repository()
repo.add_queries(testQuery.queries)
self.assertEqual(len(repo.retrieve_all_queries()), 2)
self.assertEqual(
repo.retrieve_query("companies").name,
repo.retrieve_query("test_query").dependencies[0])
def test_repo_keeps_dependencies(self):
testQuery = SqlCode("with companies as (select company_name, company_id) select company_name from companies", "test_query")
repo = Repository()
repo.add_queries(testQuery.queries)
self.assertEqual(len(repo.retrieve_all_queries()), 2)
self.assertEqual(repo.retrieve_query("companies").name, repo.retrieve_query("test_query").dependencies[0])
