[Checkins] SVN: z3c.sqlalchemy/trunk/src/z3c/sqlalchemy/tests/testSQLAlchemy.py test for registerSQLAlchemyWrapper()

[Andreas Jung]

Log message for revision 73714:
  test for registerSQLAlchemyWrapper()
  

Changed:
  U   z3c.sqlalchemy/trunk/src/z3c/sqlalchemy/tests/testSQLAlchemy.py

-=-
Modified: z3c.sqlalchemy/trunk/src/z3c/sqlalchemy/tests/testSQLAlchemy.py
===================================================================
--- z3c.sqlalchemy/trunk/src/z3c/sqlalchemy/tests/testSQLAlchemy.py	2007-03-27 10:52:54 UTC (rev 73713)
+++ z3c.sqlalchemy/trunk/src/z3c/sqlalchemy/tests/testSQLAlchemy.py	2007-03-27 11:14:13 UTC (rev 73714)
@@ -17,10 +17,12 @@
 import sqlalchemy
 
 from zope.interface.verify import verifyClass
+from zope.component import getUtility
+
 from z3c.sqlalchemy.interfaces import ISQLAlchemyWrapper
 from z3c.sqlalchemy.postgres import PythonPostgresWrapper,  ZopePostgresWrapper
 from z3c.sqlalchemy.base import BaseWrapper
-from z3c.sqlalchemy import createSQLAlchemyWrapper, Model
+from z3c.sqlalchemy import createSQLAlchemyWrapper, Model, registerSQLAlchemyWrapper
 
 
 class WrapperTests(unittest.TestCase):
@@ -51,15 +53,19 @@
                     """)""")
         db.close()
 
+
     def testIFaceBaseWrapper (self):
         verifyClass(ISQLAlchemyWrapper , BaseWrapper)
 
+
     def testIFacePythonPostgres(self):
         verifyClass(ISQLAlchemyWrapper , PythonPostgresWrapper)
 
+
     def testIFaceZopePostgres(self):
         verifyClass(ISQLAlchemyWrapper , ZopePostgresWrapper)
 
+
     def testSimplePopulation(self):
         db = createSQLAlchemyWrapper('sqlite:///test')
         # obtain mapper for table 'user'
@@ -77,6 +83,7 @@
         rows = session.query(User).select()
         self.assertEqual(len(rows), 2)
 
+
     def testMapperWithCustomModel(self):
 
         class myUser(object): 
@@ -89,6 +96,7 @@
         User = db.getMapper('user')
         self.assertEqual(User, myUser)
 
+
     def testGetMappers(self):
 
         db = createSQLAlchemyWrapper('sqlite:///test')
@@ -101,6 +109,7 @@
         M = Model()
         self.assertRaises(ValueError, M.add, 'user', relations=('foo', 'bar'), autodetect_relations=True)
 
+
     def testModelNonExistingTables(self):
         M = Model()
         M.add('non_existing_table')
@@ -111,6 +120,13 @@
             pass
 
 
+    def testWrapperRegistration(self):
+        wrapper = createSQLAlchemyWrapper('sqlite:///test')
+        registerSQLAlchemyWrapper(wrapper, 'test.wrapper')
+        wrapper2 = getUtility(ISQLAlchemyWrapper, 'test.wrapper')
+        self.assertEqual(wrapper, wrapper2)
+
+
 def test_suite():
     from unittest import TestSuite, makeSuite
     suite = TestSuite()