def test_works_with_simple_query(self): query = "SELECT 1" self.assertEquals([], extract_query_ids(query))
def test_finds_queries_in_joins(self): query = "SELECT * FROM query_123 JOIN query_4566" self.assertEquals([123, 4566], extract_query_ids(query))
def test_finds_queries_with_whitespace_characters(self): query = "SELECT * FROM query_123 a JOIN\tquery_4566 b ON a.id=b.parent_id JOIN\r\nquery_78 c ON b.id=c.parent_id" self.assertEquals([123, 4566, 78], extract_query_ids(query))
def test_finds_queries_to_load(self): query = "SELECT * FROM query_123" self.assertEquals([123], extract_query_ids(query))
def test_finds_queries_with_whitespace_characters(self): query = "SELECT * FROM query_123 a JOIN\tquery_4566 b ON a.id=b.parent_id JOIN\r\nquery_78 c ON b.id=c.parent_id" self.assertEqual([123, 4566, 78], extract_query_ids(query))
def test_finds_queries_in_joins(self): query = "SELECT * FROM query_123 JOIN query_4566" self.assertEqual([123, 4566], extract_query_ids(query))
def test_finds_queries_to_load(self): query = "SELECT * FROM query_123" self.assertEqual([123], extract_query_ids(query))
def test_works_with_simple_query(self): query = "SELECT 1" self.assertEqual([], extract_query_ids(query))