def test_valid(self): assignment = Set(first='foo') condition = eq('last', 'bar') op = (Update( self.column_family).set(**assignment.kwargs).where(condition)) cql, args = op.statement() self.assertEquals(cql, self.get_query(assignment, condition)) self.assertEquals(args, assignment.values + condition.values)
def test_options(self): assignment = Set(first='foo') condition = eq('last', 'bar') using = Using(ttl=10800) op = (Update(self.column_family).using(**using.options).set( **assignment.kwargs).where(condition)) cql, args = op.statement() self.assertEquals(cql, self.get_query(assignment, condition, using)) self.assertEquals(args, using.values + assignment.values + condition.values)
def update(column_family, keyspace=None): return Update(column_family, keyspace)
def test_no_condition(self): op = (Update(self.column_family).set(name='foo')) self.assertRaises(ValidationError, op.statement)
def test_no_assignment(self): op = (Update(self.column_family).where(eq('name', 'foo'))) self.assertRaises(ValidationError, op.statement)