fixed update clause

This commit is contained in:
Jon Haddad
2014-06-18 15:07:58 -07:00
parent d66ec78756
commit 52c736b57d
3 changed files with 4 additions and 4 deletions

View File

@@ -8,7 +8,7 @@ class BaseClauseTests(TestCase):
ss = BaseClause('a', 'b')
assert ss.get_context_size() == 1
ctx = []
ctx = {}
ss.set_context_id(10)
ss.update_context(ctx)
assert ctx == {'10': 'b'}

View File

@@ -31,7 +31,7 @@ class DeleteStatementTests(TestCase):
def test_where_clause_rendering(self):
ds = DeleteStatement('table', None)
ds.add_where_clause(WhereClause('a', EqualsOperator(), 'b'))
self.assertEqual(unicode(ds), 'DELETE FROM table WHERE "a" = ?', unicode(ds))
self.assertEqual(unicode(ds), 'DELETE FROM table WHERE "a" = %(0)s', unicode(ds))
def test_context_update(self):
ds = DeleteStatement('table', None)
@@ -39,7 +39,7 @@ class DeleteStatementTests(TestCase):
ds.add_where_clause(WhereClause('a', EqualsOperator(), 'b'))
ds.update_context_id(7)
self.assertEqual(unicode(ds), 'DELETE "d"[?] FROM table WHERE "a" = ?')
self.assertEqual(unicode(ds), 'DELETE "d"[%(8)s] FROM table WHERE "a" = %(7)s')
self.assertEqual(ds.get_context(), {'7': 'b', '8': 3})
def test_context(self):

View File

@@ -31,7 +31,7 @@ class UpdateStatementTests(TestCase):
us.add_assignment_clause(AssignmentClause('c', 'd'))
us.add_where_clause(WhereClause('a', EqualsOperator(), 'x'))
us.update_context_id(3)
self.assertEqual(unicode(us), 'UPDATE table SET "a" = ?, "c" = ? WHERE "a" = ?')
self.assertEqual(unicode(us), 'UPDATE table SET "a" = %(4)s, "c" = %(5)s WHERE "a" = %(3)s')
self.assertEqual(us.get_context(), {'4': 'b', '5': 'd', '3': 'x'})
def test_additional_rendering(self):