def test_decorator_only(self):
tbl = Table("foo", [])
with self.assertRaises(BuildError):
Query(tbl).limit(5)
with self.assertRaises(BuildError):
Query(tbl).insert().where()
def test_decorator_start(self):
tbl = Table("foo", [])
with self.assertRaises(BuildError):
Query(tbl).insert().select()
with self.assertRaises(BuildError):
Query(tbl).insert().values().select().where()
def test_create(self):
query = Query(one).create()
self.assertEqual(query.table, one)
self.assertEqual(query._action, QueryAction.create)
self.assertFalse(query._if_not_exists)
query = Query(two).create(if_not_exists=True)
self.assertEqual(query.table, two)
self.assertEqual(query._action, QueryAction.create)
self.assertTrue(query._if_not_exists)
def test_prepare(self):
engine = MockEngine()
query = Query(one).select()
pquery = engine.prepare(query)
self.assertEqual(pquery.table, one)
self.assertEqual(pquery.sql, "select")
self.assertEqual(pquery.parameters, [])
with self.assertRaises(NotImplementedError):
query = Query(one).insert().values([1, 2])
engine.prepare(query)
def test_insert(self):
query = Query(one).insert().values((1, 2), (3, 4))
self.assertEqual(query.table, one)
self.assertEqual(query._action, QueryAction.insert)
self.assertEqual(query._columns, one._columns)
self.assertEqual(query._rows, [(1, 2), (3, 4)])
query = Query(one).insert(one.b).values((1, ), (3, )).values((2, ),
(4, ))
self.assertEqual(query.table, one)
self.assertEqual(query._action, QueryAction.insert)
self.assertEqual(query._columns, [one.b])
self.assertEqual(query._rows, [(1, ), (3, ), (2, ), (4, )])
def test_update(self):
query = (Query(one).update(one.a == 5,
b="hello").where(one.b != "hello").limit(5))
self.assertEqual(query.table, one)
self.assertEqual(query._action, QueryAction.update)
self.assertEqual(query._updates, {one.a: 5, one.b: "hello"})
self.assertEqual(len(query._where), 1)
self.assertEqual(query._where[0].clauses, (one.b != "hello", ))
self.assertEqual(query._limit, 5)
with self.assertRaises(BuildError):
Query(one).update()
with self.assertRaises(BuildError):
Query(one).update(one.a < 5)
def test_delete(self):
query = Query(one).delete().where(one.a == 20).limit(1)
self.assertEqual(query.table, one)
self.assertEqual(query._action, QueryAction.delete)
self.assertEqual(len(query._where), 1)
self.assertEqual(query._where[0].clauses, (one.a == 20, ))
self.assertEqual(query._limit, 1)
def test_select_group_by(self):
query = Query(one).select().groupby(one.a)
self.assertEqual(query.table, one)
self.assertEqual(query._columns, one._columns)
self.assertEqual(query._groupby, [one.a])
self.assertEqual(query._having, [])
with self.assertRaises(BuildError):
query.groupby(one.b)
with self.assertRaises(BuildError):
query.having()
query.having(one.b == 3)
self.assertEqual(len(query._having), 1)
self.assertEqual(query._having[0].clauses, (one.b == 3, ))
with self.assertRaises(BuildError):
Query(one).select().groupby()
with self.assertRaises(BuildError):
Query(one).select().having()
def test_factory(self):
@table
class Foo:
a: int
b: str
query = Query(Foo).select()
self.assertEqual(query.factory(), Foo._source)
query = Query(Foo).select(Foo.a)
factory = query.factory()
self.assertNotEqual(factory, Foo._source)
self.assertEqual(factory(1), factory(1))
self.assertEqual(factory(1).a, 1)
with self.assertRaises(AttributeError):
_value = factory(1).b
query = Query(one).select()
factory = query.factory()
self.assertEqual(factory(1, 2), factory(1, 2))
self.assertEqual(factory(1, 2).a, 1)
self.assertEqual(factory(1, 2).b, 2)
def test_select(self):
query = Query(one).select(one.a).where(one.b > 5, two.f < 10).limit(7)
self.assertEqual(query.table, one)
self.assertEqual(query._action, QueryAction.select)
self.assertEqual(query._selector, Select.all)
self.assertEqual(query._columns, [one.a])
self.assertEqual(len(query._where), 1)
self.assertEqual(query._where[0].clauses, (one.b > 5, two.f < 10))
self.assertEqual(query._limit, 7)
self.assertEqual(query._offset, None)
query.offset(25)
self.assertEqual(query._offset, 25)
query.distinct()
self.assertEqual(query._selector, Select.distinct)
with self.assertRaises(BuildError):
Query(one).select().where()
def test_select_joins(self):
query = Query(one).select().join(two, Join.left).on(one.a == two.e)
self.assertEqual(query.table, one)
self.assertEqual(query._columns, one._columns)
self.assertEqual(query._joins,
[TableJoin(two, Join.left, [one.a == two.e], [])])
with self.assertRaises(BuildError):
query.using(two.f)
query = query.join(tre).using(tre.e, tre.f)
self.assertEqual(
query._joins,
[
TableJoin(two, Join.left, [one.a == two.e], []),
TableJoin(tre, Join.inner, [], [tre.e, tre.f]),
],
)
with self.assertRaises(BuildError):
query.on(one.a == two.f)
def query(self):
"""
Create a query handle.
"""
return Query(self.base_url, self.name)
def test_delete_everything(self):
query = Query(one).delete()
self.assertFalse(query._everything)
query.everything()
self.assertTrue(query._everything)