Changeset 516 for elixir/trunk

Timestamp:

11/13/09 11:06:20 (3 years ago)

Author:

ged

Message:

added tests for custom collection, or custom session on base class (they serve
more as examples than real tests but that's already quite useful).

Location:

elixir/trunk/tests

Files:

• test_abstract.py (modified) (2 diffs)
• test_custombase.py (modified) (2 diffs)

elixir/trunk/tests/test_abstract.py

@@ -12 +12 @@
 
 def setup():
-    # this is an ugly hack because globally defined entities of other tests 
-    # (a.*, b.*, db1.* and db2.*) leak into this one because of nosetests 
-    # habit of importing all modules before running the first test. 
-    # test_abstract is usually the first test to be run, so it gets all the 
+    # this is an ugly hack because globally defined entities of other tests
+    # (a.*, b.*, db1.* and db2.*) leak into this one because of nosetests
+    # habit of importing all modules before running the first test.
+    # test_abstract is usually the first test to be run, so it gets all the
     # crap.
     cleanup_all()
@@ -196 +196 @@
         session.commit()
 
-

elixir/trunk/tests/test_custombase.py

@@ -4 +4 @@
 
 from elixir import *
+import elixir
 
 def setup():
@@ -182 +183 @@
         assert TestA._descriptor.identity == 'test_a'
 
+    def test_base_custom_collection(self):
+        global A, B
+
+        collection = elixir.collection.RelativeEntityCollection()
+
+        class Base(object):
+            __metaclass__ = EntityMeta
+            using_options_defaults(collection=collection)
+
+        class A(Base):
+            b = ManyToOne('B')
+
+        class B(Base):
+            a = OneToOne('A')
+
+        assert not elixir.entities
+        assert A.table is None
+        assert B.table is None
+
+        setup_entities(collection)
+
+        assert A.table is not None
+        assert B.table is not None
+
+        del A
+        del B
+
+    def test_base_custom_session(self):
+        from sqlalchemy.orm import sessionmaker
+
+        class Base(object):
+            __metaclass__ = EntityMeta
+            using_options_defaults(session=None)
+
+        class A(Base):
+            b = ManyToOne('B')
+
+        class B(Base):
+            a = OneToOne('A')
+        
+        setup_all(True)
+
+        a = A()
+
+        assert a not in elixir.session
+
+        session = sessionmaker()()
+        session.add(a)
+        assert a in session
+