def test_where_clause(self):
qb = QueryBuilder('company')
qb.where("a")
self.assertEqual(qb.incomplete_filter_flag, True)
self.assertEqual(len(qb.filters), 1)
self.assertEqual(qb.filters[0], 'a')
def test_like_clause_raise_error_on_non_str(self):
qb = QueryBuilder('company')
qb.where("a")
self.assertRaisesRegexp(InvalidQueryError, r"value be of type string",
qb.like, True)
def _test_clause_chainable(self, clause, param, where=True):
qb = QueryBuilder('company')
if where:
qb.where("a")
ret = getattr(qb, clause)(param)
self.assertEqual(qb, ret)
def test_gte_clause(self):
qb = QueryBuilder('company')
qb.where("a").gte("5")
self._test_clause(qb, "a", ">=", "'5'")
def test_like_clause(self):
qb = QueryBuilder('company')
qb.where("a").like("5")
self._test_clause(qb, "a", "Like", "'5'")
def test_equal_clause(self):
qb = QueryBuilder('company')
qb.where("a").equals("b")
self._test_clause(qb, "a", '=', "'b'")
def test_like_clause(self):
qb = QueryBuilder("company")
qb.where("a").like("5")
self._test_clause(qb, "a", "Like", "'5'")
def test_equal_clause(self):
qb = QueryBuilder("company")
qb.where("a").equals("b")
self._test_clause(qb, "a", "=", "'b'")
def test_contains_clause_tuple(self):
qb = QueryBuilder("company")
values = (1, 2, 3)
qb.where("a").contains(values)
self._test_clause(qb, "a", "in", str(values))
def test_incomplete_where_clause_raises_error(self):
qb = QueryBuilder("company")
qb.where("a")
self.assertRaisesRegexp(InvalidQueryError, r"Incomplete Filter Clause", qb.build)
def test_equal_clause_bool(self):
qb = QueryBuilder("company")
qb.where("a").equals(True)
self._test_clause(qb, "a", "=", "true")
def test_where_clause(self):
qb = QueryBuilder("company")
qb.where("a")
self.assertEqual(qb.incomplete_filter_flag, True)
self.assertEqual(len(qb.filters), 1)
self.assertEqual(qb.filters[0], "a")
def _test_clause_chainable(self, clause, param, where=True):
qb = QueryBuilder("company")
if where:
qb.where("a")
ret = getattr(qb, clause)(param)
self.assertEqual(qb, ret)
def test_like_clause_raise_error_on_non_str(self):
qb = QueryBuilder("company")
qb.where("a")
self.assertRaisesRegexp(InvalidQueryError, r"value be of type string", qb.like, True)
def test_incomplete_where_clause_raises_error(self):
qb = QueryBuilder('company')
qb.where("a")
self.assertRaisesRegexp(InvalidQueryError,
r'Incomplete Filter Clause', qb.build)
def test_contains_clause_raises_exception_str_value(self):
qb = QueryBuilder("company")
values = "fgf"
qb = qb.where("a")
self.assertRaisesRegexp(InvalidQueryError, "list/tuple of values", qb.contains, values)
def test_equal_clause_bool(self):
qb = QueryBuilder('company')
qb.where("a").equals(True)
self._test_clause(qb, "a", '=', "true")
def test_contains_clause_raises_exception_str_value(self):
qb = QueryBuilder('company')
values = "fgf"
qb = qb.where('a')
self.assertRaisesRegexp(InvalidQueryError, "list/tuple of values",
qb.contains, values)
def test_contains_clause_tuple(self):
qb = QueryBuilder('company')
values = (1, 2, 3)
qb.where('a').contains(values)
self._test_clause(qb, "a", 'in', str(values))
def test_gte_clause(self):
qb = QueryBuilder("company")
qb.where("a").gte("5")
self._test_clause(qb, "a", ">=", "'5'")
