- [feature] Simplified the versioning example

a bit to use a declarative mixin as well
as an event listener, instead of a metaclass +
SessionExtension.  [ticket:2313]

1 files changed, 3 insertions, 5 deletions

diff --git a/examples/versioning/test_versioning.py b/examples/versioning/test_versioning.py
index d91f12037..389dba918 100644
--- a/examples/versioning/test_versioning.py
+++ b/examples/versioning/test_versioning.py
@@ -1,6 +1,6 @@
 from unittest import TestCase
 from sqlalchemy.ext.declarative import declarative_base
-from history_meta import VersionedMeta, VersionedListener
+from history_meta import Versioned, versioned_session
 from sqlalchemy import create_engine, Column, Integer, String, ForeignKey
 from sqlalchemy.orm import clear_mappers, sessionmaker, deferred, relationship
 from _lib import ComparableEntity, eq_
@@ -13,10 +13,8 @@ class TestVersioning(TestCase):
     def setUp(self):
         global Base, Session, Versioned
         Base = declarative_base(bind=engine)
-        class Versioned(object):
-            __metaclass__ = VersionedMeta
-            _decl_class_registry = Base._decl_class_registry
-        Session = sessionmaker(extension=VersionedListener())
+        Session = sessionmaker()
+        versioned_session(Session)
 
     def tearDown(self):
         clear_mappers()