def test_get_columns_specified(self):
qb = QueryBuilder('company')
columns = 'a, b'
qb.select(columns)
self.assertEqual(qb.get_columns(), columns)
def test_select_columns_tuple(self):
qb = QueryBuilder('company')
columns = ('a', 'b')
columns_str = ", ".join(columns)
qb.select(columns)
self.assertEqual(qb.columns, columns_str)
def test_select_columns_str(self):
qb = QueryBuilder('company')
columns = 'a,b'
qb.select(columns)
self.assertEqual(qb.columns, columns)
def test_select_chainable(self):
qb = QueryBuilder('company')
ret = qb.select("a, b")
self.assertEqual(qb, ret)
def test_get_columns_specified(self):
qb = QueryBuilder("company")
columns = "a, b"
qb.select(columns)
self.assertEqual(qb.get_columns(), columns)
def test_select_columns_str(self):
qb = QueryBuilder("company")
columns = "a,b"
qb.select(columns)
self.assertEqual(qb.columns, columns)
def test_select_columns_tuple(self):
qb = QueryBuilder("company")
columns = ("a", "b")
columns_str = ", ".join(columns)
qb.select(columns)
self.assertEqual(qb.columns, columns_str)
def test_select_chainable(self):
qb = QueryBuilder("company")
ret = qb.select("a, b")
self.assertEqual(qb, ret)
