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)
