class QueryBuilderTest(unittest.TestCase):
def setUp(self):
self.builder = QueryBuilder()
def test_select_column_from_table(self):
self.builder.select('account_name').from_table('account')
sql = self.builder.generate_sql()
self.assertEqual('SELECT account_name FROM account;', sql)
def test_select_star_from_table(self):
self.builder.select('*').from_table('account')
sql = self.builder.generate_sql()
self.assertEqual('SELECT * FROM account;', sql)
def test_column_defaults_to_star(self):
self.builder.from_table('account')
sql = self.builder.generate_sql()
self.assertEqual('SELECT * FROM account;', sql)
def test_column_can_take_list(self):
self.builder.select(['col1', 'col2']).from_table('account')
sql = self.builder.generate_sql()
self.assertEqual('SELECT col1, col2 FROM account;', sql)
def test_select_with_where_clause(self):
self.builder.select('account_name').from_table('account').where(
'account_id', 13977)
sql = self.builder.generate_sql()
self.assertEqual(
'SELECT account_name FROM account WHERE account_id=13977;', sql)
class QueryBuilderTest(unittest.TestCase):
def setUp(self):
self.builder = QueryBuilder()
def test_select_column_from_table(self):
self.builder.select("account_name").from_table("account")
sql = self.builder.generate_sql()
self.assertEqual("SELECT account_name FROM account;", sql)
def test_select_star_from_table(self):
self.builder.select("*").from_table("account")
sql = self.builder.generate_sql()
self.assertEqual("SELECT * FROM account;", sql)
def test_column_defaults_to_star(self):
self.builder.from_table("account")
sql = self.builder.generate_sql()
self.assertEqual("SELECT * FROM account;", sql)
def test_column_can_take_list(self):
self.builder.select(["col1", "col2"]).from_table("account")
sql = self.builder.generate_sql()
self.assertEqual("SELECT col1, col2 FROM account;", sql)
def test_select_with_where_clause(self):
self.builder.select("account_name").from_table("account").where("account_id", 13977)
sql = self.builder.generate_sql()
self.assertEqual("SELECT account_name FROM account WHERE account_id=13977;", sql)
