def test_array_ref(self):
qhi = QueryHelper(Invoice)
self.assertEqual({"items[0].product-id": 10},
(qhi.items[0].product_id == 10))
qhl = QueryHelper(ListOfInts)
self.assertEqual({"py/state.values[0]": {
"$lt": 10
}}, (qhl.values[0] < 10))
def test_project(self):
qhi = QueryHelper(Invoice)
self.assertEqual(
{
"datetime": True,
"number": True,
"items": False
},
qhi.project(qhi.datetime, qhi.number, (qhi.items, False)),
)
def test_basic_operators(self):
qhi = QueryHelper(Invoice)
self.assertEqual({"items.product-id": 10},
(qhi.items.product_id == 10))
self.assertEqual({"items.product-id": {
"$ne": 10
}}, (qhi.items.product_id != 10))
self.assertEqual({"items.product-id": {
"$lt": 10
}}, (qhi.items.product_id < 10))
self.assertEqual({"items.product-id": {
"$lte": 10
}}, (qhi.items.product_id <= 10))
self.assertEqual({"items.product-id": {
"$gt": 10
}}, (qhi.items.product_id > 10))
self.assertEqual({"items.product-id": {
"$gte": 10
}}, (qhi.items.product_id >= 10))
self.assertEqual({"items.product-id": {
"$in": [1, 2]
}}, qhi.items.product_id.in_([1, 2]))
self.assertEqual({"items.product-id": {
"$nin": [1, 2]
}}, qhi.items.product_id.nin([1, 2]))
self.assertEqual({"items.product-id": {
"$exists": True
}}, (qhi.items.product_id.exists()))
def test_logic_operators(self):
qhi = QueryHelper(Invoice)
self.assertEqual(
{"$and": [{
"items.product-id": 10
}, {
"items.quantity": 5
}]},
qhi.and_(qhi.items.product_id == 10, qhi.items.quantity == 5),
)
self.assertEqual(
{
"items.product-id": 10,
"items.quantity": 5
},
qhi(qhi.items.product_id == 10, qhi.items.quantity == 5),
)
self.assertEqual(
{"$or": [{
"items.product-id": 10
}, {
"items.quantity": 5
}]},
qhi.or_(qhi.items.product_id == 10, qhi.items.quantity == 5),
)
self.assertEqual(
{"$nor": [{
"items.product-id": 10
}, {
"items.quantity": 5
}]},
qhi.nor(qhi.items.product_id == 10, qhi.items.quantity == 5),
)
self.assertEqual(
{"$not": {
"items.product-id": 10
}},
qhi.not_(qhi.items.product_id == 10),
)
self.assertEqual(
{
"$not": {
"$and": [{
"items.product-id": 10
}, {
"items.quantity": 5
}]
}
},
qhi.not_(
qhi.and_(qhi.items.product_id == 10, qhi.items.quantity == 5)),
)
def test_set(self):
qhi = QueryHelper(Invoice)
self.assertEqual({"$set": {"number": 1}}, qhi.set(qhi.number(1)))
self.assertEqual({"$setOrInsert": {
"number": 1
}}, qhi.set_or_insert(qhi.number(1)))
def test_sort(self):
qhi = QueryHelper(Invoice)
self.assertEqual([("datetime", -1), ("number", 1)],
qhi.sort(qhi.datetime.desc(), qhi.number.asc()))
def test_obj_type_expr(self):
qhi = QueryHelper(Invoice)
self.assertEqual({"py/object": "test.test_query_helper.Invoice"},
qhi.type_expr())
def test_json_pickle_format(self):
qhi = QueryHelper(JsonPickleInvoice)
self.assertEqual({"py/state.items.product-id": 10},
(qhi.items.product_id == 10))