Add a test when deleting an entity

hadrien · Jun 5, 2014 · 340c297 · 340c297
1 parent 889abe6
commit 340c297
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/pyramid_caching/tests/unittests/test_sqlalchemy_extension.py b/pyramid_caching/tests/unittests/test_sqlalchemy_extension.py
@@ -31,6 +31,10 @@ def setUp(self):
         self.config = Configurator(settings={
             'caching.enabled': True,
             })
+
+        Session.add(User(name='hadrien', address='down the hill'))
+        Session.commit()
+
         register_sqla_session_caching_hook(self.config, Session)
         self.config.registry.registerAdapter(DummyIdentityInspector(),
                                              required=[User],
@@ -67,6 +71,13 @@ def test_modify_entity(self):
         Session.commit()
         self.assertEqual(self.key_versioner.incr_keys, ['users', 'users:joe'])
 
+    def test_delete_entity(self):
+        user = Session.query(User).filter_by(name='hadrien').first()
+        Session.delete(user)
+        Session.commit()
+        self.assertEqual(self.key_versioner.incr_keys,
+                         ['users', 'users:hadrien'])
+
 
 class User(object):
     __tablename__ = 'users'