def test_nor(self):
"""Query.nor"""
q1 = query.Query([('a', 'aye')])
q2 = query.Query([('b', 'bee')])
c1 = q1.nor(q2).criteria
c2 = q1._op('$nor', q2).criteria
self.assertEqual(c1, c2, "failed to combind queries")
def test_and(self):
"""Query.__and__"""
q1 = query.Query([('a', 'aye')])
q2 = query.Query([('b', 'bee')])
c1 = (q1 & q2).criteria
c2 = q1._op('$and', q2).criteria
self.assertEqual(c1, c2, "failed to combind queries")
def test_negate(self):
"""Query.negate"""
c1 = OrderedDict([('a', 'aye')])
c2 = OrderedDict([('$not', c1)])
self.assertEqual(query.Query(c1).negate().criteria, c2, "negated query is incorrect")
self.assertEqual(query.Query(c2).negate().criteria, c1, "negated query is incorrect")
c1 = OrderedDict([])
self.assertEqual(query.Query(c1).negate().criteria, c1, "negated query is incorrect")
def test_eq(self):
"""Query.__eq__"""
c1 = {'name': 'nope'}
c2 = {'name': 'yep'}
self.assertTrue(query.Query(c1).__eq__(query.Query(c1)))
self.assertTrue(query.Query(None).__eq__(query.Query(None)))
self.assertTrue(query.Query({}).__eq__(query.Query({})))
self.assertFalse(query.Query(c1).__eq__(query.Query(c2)))
self.assertFalse(query.Query(c1).__eq__(object()))
self.assertFalse(query.Query(c1).__eq__(None))
def test_copy(self):
"""Query.copy"""
criteria = {'a': 'aye'}
q1 = query.Query(criteria)
q2 = q1.copy()
q2.criteria['b'] = 'bee'
self.assertNotIn('b', q1.criteria, "query does not copy criteria")
def test_encode(self):
"""Query.encode"""
class Doc(Document):
index = Field(int)
name = Field(str)
# test regular query
raw = {'index': "12", 'name': time(12, 37)}
want = {'index': 12, 'name': '12:37:00'}
self.assertEqual(query.Query(raw).encode(Doc), want, "encoded query is incorrect")
# test scalar operator
raw = {'index': {'$gt': "5"}, 'name': 'the best'}
want = {'index': {'$gt': 5}, 'name': 'the best'}
self.assertEqual(query.Query(raw).encode(Doc), want, "encoded query is incorrect")
# test list operator
raw = {'index': {'$in': [1, "12", 15]}}
want = {'index': {'$in': [1, 12, 15]}}
self.assertEqual(query.Query(raw).encode(Doc), want, "encoded query is incorrect")
# test lists
raw = {'$or': [{'index': "12"}, {'index': 15}]}
want = {'$or': [{'index': 12}, {'index': 15}]}
self.assertEqual(query.Query(raw).encode(Doc), want, "encoded query is incorrect")
# test dicts
raw = {'$not': {'index': "12"}}
want = {'$not': {'index': 12}}
self.assertEqual(query.Query(raw).encode(Doc), want, "encoded query is incorrect")
# test raw query
raw = {'$not': {'index': "12"}}
want = raw.copy()
self.assertEqual(
query.Query(raw).encode(Doc, raw=True), want, "encoded query is incorrect")
# test something we know we can't encode
raw = {'not-a-field': 'seems like a legitimate value'}
want = raw.copy()
self.assertEqual(query.Query(raw).encode(Doc), want, "encoded query is incorrect")
# test something that isn't a dict at all
raw = datetime.now()
self.assertRaises(TypeError, query.Query, raw)
def test(op, c1, c2, want):
have = query.Query(c1)._op(op, query.Query(c2))
self.assertEqual(have.criteria, want, "failed to combine queries")