def test_MongoClause_expr(self):
"Test MongoClause class generated expression"
fld = 'foo'
val = 10
for op, r_op in (('>=', '$lt'), ('<', '$gte'), ('=', '$ne')):
cn = vv.Constraint(fld, op, val)
obj = vv.MongoClause(cn)
expected = {fld: {r_op: val}}
self.failUnlessEqual(obj.expr, expected)
obj = vv.MongoClause(vv.Constraint(fld, '!=', val))
self.failUnlessEqual(obj.expr, {fld: val})
def test_MongoQuery_exists(self):
"Test MongoQuery class, with an exists clause"
q = vv.MongoQuery()
c = vv.MongoClause(vv.Constraint('foo', 'exists', True))
q.add_clause(c)
m = q.to_mongo()
self.failUnlessEqual(m, c.expr)
def test_MongoQuery_base(self):
"Test MongoQuery class"
q = vv.MongoQuery()
self.assertRaises(AttributeError, q.add_clause, None)
c = vv.MongoClause(vv.Constraint('foo', '>', 10))
q.add_clause(c)
m = q.to_mongo()
self.failUnlessEqual(m, c.expr)
def test_MongoClause(self):
"Test MongoClause class"
self.assertRaises(AssertionError, vv.MongoClause, None)
fld = vv.Field('foo')
op = vv.ConstraintOperator('>=')
cn = vv.Constraint(fld, op, 10)
obj = vv.MongoClause(cn)
self.assert_(obj.query_loc == vv.MongoClause.LOC_MAIN)
def test_Constraint_type(self):
"Test the Constraint class for types"
obj = vv.Constraint('foo', 'type', 'number')
self.failIf(obj.passes('123')[0])
self.failIf(obj.passes(True)[0])
self.failUnless(obj.passes(123)[0])
self.failUnless(obj.passes(1.23)[0])
obj = vv.Constraint('foo', 'type', 'string')
self.failUnless(obj.passes('123')[0])
self.failIf(obj.passes(True)[0])
self.failIf(obj.passes(123)[0])
self.failIf(obj.passes(1.23)[0])
obj = vv.Constraint('foo', 'type', 'bool')
self.failIf(obj.passes('123')[0])
self.failUnless(obj.passes(True)[0])
self.failIf(obj.passes(123)[0])
self.failIf(obj.passes(1.23)[0])
def test_Constraint(self):
"Test the Constraint class"
for f, o, v in ((None, None, None), (1, 2, 3), ('a', 'b', 'c')):
self.assertRaises(ValueError, vv.Constraint, f, o, v)
obj = vv.Constraint(vv.Field('foo'), vv.ConstraintOperator('>'), 10)
self.failIf(obj.passes(None)[0])
self.failIf(obj.passes('dude')[0])
self.failIf(obj.passes(10)[0])
self.failUnless(obj.passes(11)[0])
def test_MongoClause_type(self):
"Test MongoClause class for types"
fld, op = 'foo', 'type'
for rev, whereop in ((True, '!='), (False, '==')):
for val, type_js in (('int', 'number'), ('str', 'string'),
('bool', 'boolean'), ('number', 'number')):
obj = vv.MongoClause(vv.Constraint(fld, op, val), rev=rev)
self.failUnlessEqual(obj.query_loc, vv.MongoClause.LOC_WHERE)
expected = 'typeof this.{} {} "{}"'.format(
fld, whereop, type_js)
self.failUnlessEqual(obj.expr, expected)
def test_MongoClause_where(self):
"Test MongoClause class for where"
fld = vv.Field('foo')
for suffix, where in zip(('$', '>', ''), (True, True, True)):
sz = 'size' + suffix
op = vv.ConstraintOperator(sz)
cn = vv.Constraint(fld, op, 10)
obj = vv.MongoClause(cn)
if where:
self.failUnlessEqual(obj.query_loc, vv.MongoClause.LOC_WHERE)
else:
self.failIfEqual(obj.query_loc, vv.MongoClause.LOC_WHERE)
def test_MongoClause_size(self):
"Test MongoClause class for sizes"
fld = 'foo'
val = 10
# reversed
for op, r_op in (('size>', '<='), ('size<', '>=')):
obj = vv.MongoClause(vv.Constraint(fld, op, val))
expected = 'this.{}.length {} {}'.format(fld, r_op, val)
self.failUnlessEqual(obj.query_loc, vv.MongoClause.LOC_WHERE)
self.failUnlessEqual(obj.expr, expected)
# not reversed
for op, r_op in (('size>', '>'), ('size<', '<')):
obj = vv.MongoClause(vv.Constraint(fld, op, val), rev=False)
expected = 'this.{}.length {} {}'.format(fld, r_op, val)
self.failUnlessEqual(obj.query_loc, vv.MongoClause.LOC_WHERE)
self.failUnlessEqual(obj.expr, expected)
# variable
obj = vv.MongoClause(vv.Constraint('foo', 'size$', 'bar'))
self.failUnlessEqual(obj.query_loc, vv.MongoClause.LOC_WHERE)
self.failUnlessEqual(obj.expr, 'this.foo.length != this.bar')
# variable not reversed
obj = vv.MongoClause(vv.Constraint('foo', 'size$', 'bar'), rev=False)
self.failUnlessEqual(obj.query_loc, vv.MongoClause.LOC_WHERE)
self.failUnlessEqual(obj.expr, 'this.foo.length == this.bar')
# eq, reversed, in where
obj = vv.MongoClause(vv.Constraint('foo', 'size', 10))
self.failUnlessEqual(obj.query_loc, vv.MongoClause.LOC_WHERE)
self.failUnlessEqual(obj.expr, 'this.foo.length != 10')
# not reversed should be in main (but never used?)
obj = vv.MongoClause(vv.Constraint('foo', 'size', 10), rev=False)
self.failUnlessEqual(obj.query_loc, vv.MongoClause.LOC_MAIN)
self.failUnlessEqual(obj.expr, {'foo': {'$size': 10}})
def test_violations(self):
"Test error determination in CollectionValidator.why_bad"
obj = vv.Validator()
# main & where clause, both fail
q = vv.MongoQuery()
q.add_clause(vv.MongoClause(vv.Constraint('foo', 'size>', 2)))
q.add_clause(vv.MongoClause(vv.Constraint('bar', '>', 1)))
rec = {'foo': [0], 'bar': 0}
reasons = obj._get_violations(q, rec)
self.failUnlessEqual(len(reasons), 2)
for r in reasons:
if r.field == 'bar':
self.failUnless(r.op == '>' and r.got_value == 0
and r.expected_value == 1)
# all pass
q = vv.MongoQuery()
q.add_clause(vv.MongoClause(vv.Constraint('foo', 'size>', 2)))
q.add_clause(vv.MongoClause(vv.Constraint('bar', '>', 1)))
rec = {'foo': [0, 1, 2], 'bar': 9}
reasons = obj._get_violations(q, rec)
rtuples = [r.as_tuple() for r in reasons]
print('\n'.join(map(str, rtuples)))
self.failUnlessEqual(len(reasons), 0)
def test_MongoQuery_where(self):
"Test MongoQuery class, with where clauses"
q = vv.MongoQuery()
c1 = vv.MongoClause(vv.Constraint('foo', 'size', 10))
c2 = vv.MongoClause(vv.Constraint('bar', 'size', 10))
q.add_clause(c1)
q.add_clause(c2)
# no disjunction
m = q.to_mongo(False)
w = 'this.{}.length != 10 || this.{}.length != 10'
wheres = (w.format('foo', 'bar'), w.format('bar', 'foo'))
self.failUnless(m['$where'] in wheres)
# disjunction
m = q.to_mongo()
w = 'this.{}.length != 10 || this.{}.length != 10'
wheres = (w.format('foo', 'bar'), w.format('bar', 'foo'))
mwhere = m['$or'][0]['$where']
self.failUnless(mwhere in wheres)
# where and non-where together
q.add_clause(vv.MongoClause(vv.Constraint('foo', '<=', 10)))
m = q.to_mongo(False)
w = 'this.{}.length != 10 || this.{}.length != 10'
self.failUnless(m['$where'] in wheres and m['foo'] == {'$gt': 10})
def test_Constraint_init(self):
"Test the Constraint class init variations"
obj = vv.Constraint('foo', '>', 10)
