From a7c250561ff8acd10c96d523fa472972e3105da8 Mon Sep 17 00:00:00 2001 From: Blake Eggleston Date: Fri, 25 Oct 2013 14:37:15 -0700 Subject: [PATCH] adding test around update statement ttls --- cqlengine/models.py | 1 + cqlengine/tests/statements/test_update_statement.py | 7 +++++++ 2 files changed, 8 insertions(+) diff --git a/cqlengine/models.py b/cqlengine/models.py index 21f78c26..b6da73c8 100644 --- a/cqlengine/models.py +++ b/cqlengine/models.py @@ -69,6 +69,7 @@ class QuerySetDescriptor(object): """ raise NotImplementedError + class TTLDescriptor(object): """ returns a query set descriptor diff --git a/cqlengine/tests/statements/test_update_statement.py b/cqlengine/tests/statements/test_update_statement.py index 1dabbfe3..6c633852 100644 --- a/cqlengine/tests/statements/test_update_statement.py +++ b/cqlengine/tests/statements/test_update_statement.py @@ -24,3 +24,10 @@ class UpdateStatementTests(TestCase): us.add_assignment_clause(AssignmentClause('c', 'd')) us.add_where_clause(WhereClause('a', EqualsOperator(), 'x')) self.assertEqual(us.get_context(), {0: 'b', 1: 'd', 2: 'x'}) + + def test_additional_rendering(self): + us = UpdateStatement('table', ttl=60) + us.add_assignment_clause(AssignmentClause('a', 'b')) + us.add_where_clause(WhereClause('a', EqualsOperator(), 'x')) + self.assertIn('USING TTL 60', unicode(us)) +