def test_value_multi(self): op = Assignments() op1 = Set(first='foo') op2 = Set(last='bar') op.add(op1, op2) self.assertEquals(op.cql, '{}, {}'.format(op1.cql, op2.cql)) self.assertEquals(op.values, op1.values + op2.values)
def get_query(self, assignment, condition, using=None): query = 'UPDATE {}'.format(self.column_family) if using: query = '{} {}'.format(query, using.cql) assignments = Assignments() assignments.add(assignment) where = Where(condition) return '{} SET {} WHERE {}'.format(query, assignments.cql, where.cql)
def test_value_single(self): op = Assignments() kwargs = {'name': 'bar'} op.add(Set(**kwargs)) self.assertEquals(op.cql, '{}=%s'.format(*kwargs.keys())) self.assertEquals(op.values, kwargs.values())
def test_invalid_instance_value(self): op = Assignments() op.add(Set(name='foo')) op.add(eq('name', 'bar')) self.assertRaises(ValidationError, op.validate)
def test_none_value(self): op = Assignments() op.add(Set(name='foo')) op.add(None) self.assertRaises(ValidationError, op.validate)
def test_no_values(self): op = Assignments() self.assertRaises(ValidationError, op.validate)
def __init__(self, column_family, keyspace=None): Statement.__init__(self, column_family, keyspace) self.assignments = Assignments() self.conditions = None self.options = None