From: Mike Bayer <mike_mp@zzzcomputing.com>
Date: Sun, 29 Jul 2007 20:30:32 +0000 (+0000)
Subject: - added Session constructor which turns autoflush/transactional on
X-Git-Tag: rel_0_4beta1~147
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=cf633e297502696b09f18919de541091560f9351;p=thirdparty%2Fsqlalchemy%2Fsqlalchemy.git

- added Session constructor which turns autoflush/transactional on
- Session is used by unitofwork unit test now as well as session.py tests
- fixes to table/schema reflection broken last night
- doc updates
- other unittest fixes
---

diff --git a/doc/build/content/tutorial.txt b/doc/build/content/tutorial.txt
index d2077043df..702cf859a0 100644
--- a/doc/build/content/tutorial.txt
+++ b/doc/build/content/tutorial.txt
@@ -43,6 +43,14 @@ Note that the SQLite library download is not required with Windows, as the Windo
 
 Getting Started {@name=gettingstarted}
 --------------------------
+### Checking the Version
+ 
+**Note:  This tutorial is oriented towards version 0.4 of SQLAlchemy. ** Check the version of SQLAlchemy you have installed via:
+
+     {python}
+     >>> import sqlalchemy
+     >>> sqlalchemy.__version__ # doctest: +SKIP
+     0.4.0
 
 ### Imports
 
diff --git a/doc/build/lib/docstring.py b/doc/build/lib/docstring.py
index f0aebe92ba..5d33667c75 100644
--- a/doc/build/lib/docstring.py
+++ b/doc/build/lib/docstring.py
@@ -34,7 +34,7 @@ class ObjectDoc(AbstractDoc):
                     for x in objects 
                     if getattr(obj,x,None) is not None and 
                         (isinstance(getattr(obj,x), types.FunctionType))
-                        and not getattr(obj,x).__name__[0] == '_'
+                        and not self._is_private_name(getattr(obj,x).__name__)
                     ]
                 if sort:
                     functions.sort(lambda a, b: cmp(a.__name__, b.__name__))
@@ -43,7 +43,7 @@ class ObjectDoc(AbstractDoc):
                     if getattr(obj,x,None) is not None and 
                         (isinstance(getattr(obj,x), types.TypeType) 
                         or isinstance(getattr(obj,x), types.ClassType))
-                        and (self.include_all_classes or not getattr(obj,x).__name__[0] == '_')
+                        and (self.include_all_classes or not self._is_private_name(getattr(obj,x).__name__))
                     ]
                 classes = list(set(classes))
                 if sort:
@@ -53,11 +53,11 @@ class ObjectDoc(AbstractDoc):
                 functions = (
                     [getattr(obj, x).im_func for x in obj.__dict__.keys() if isinstance(getattr(obj,x), types.MethodType) 
                     and 
-                    (getattr(obj, x).__name__ == '__init__' or not getattr(obj,x).__name__[0] == '_')
+                    (getattr(obj, x).__name__ == '__init__' or not self._is_private_name(getattr(obj,x).__name__))
                     ] + 
                     [(x, getattr(obj, x)) for x in obj.__dict__.keys() if _is_property(getattr(obj,x)) 
                     and 
-                    not x[0] == '_'
+                    not self._is_private_name(x)
                     ]
                  )
                 functions.sort(lambda a, b: cmp(getattr(a, '__name__', None) or a[0], getattr(b, '__name__', None) or b[0] ))
@@ -100,7 +100,15 @@ class ObjectDoc(AbstractDoc):
         self.classes = []
         for class_ in classes:
             self.classes.append(ObjectDoc(class_))
-            
+    
+    def _is_private_name(self, name):
+        if re.match(r'^__.*__$', name):
+            return False
+        elif name.startswith('_'):
+            return True
+        else:
+            return False
+
     def _get_inherits(self):
         for item in self._inherits:
             if item[0] in self.allobjects:
diff --git a/lib/sqlalchemy/orm/__init__.py b/lib/sqlalchemy/orm/__init__.py
index 380d295770..42964545d6 100644
--- a/lib/sqlalchemy/orm/__init__.py
+++ b/lib/sqlalchemy/orm/__init__.py
@@ -18,18 +18,46 @@ from sqlalchemy.orm import mapper as mapperlib
 from sqlalchemy.orm import collections, strategies
 from sqlalchemy.orm.query import Query
 from sqlalchemy.orm.util import polymorphic_union
-from sqlalchemy.orm.session import Session as create_session
+from sqlalchemy.orm.session import Session as _Session
 from sqlalchemy.orm.session import object_session, attribute_manager
 
 __all__ = ['relation', 'column_property', 'composite', 'backref', 'eagerload',
            'eagerload_all', 'lazyload', 'noload', 'deferred', 'defer', 'undefer',
            'undefer_group', 'extension', 'mapper', 'clear_mappers',
            'compile_mappers', 'class_mapper', 'object_mapper', 'dynamic_loader',
-           'MapperExtension', 'Query', 'polymorphic_union', 'create_session',
+           'MapperExtension', 'Query', 'polymorphic_union', 'create_session','Session',
            'synonym', 'contains_alias', 'contains_eager', 'EXT_PASS',
            'object_session', 'PropComparator'
            ]
 
+def create_session(bind=None, **kwargs):
+    """create a new version 0.3-style [sqlalchemy.orm.session#Session].
+    
+    The session by default does not begin a transaction, and requires that
+    flush() be called explicitly in order to persist results to the database.
+    """
+    
+    kwargs.setdefault('autoflush', False)
+    kwargs.setdefault('transactional', False)
+    return _Session(bind=bind, **kwargs)
+    
+class Session(_Session):
+    """front-end for a [sqlalchemy.orm.session#Session].  By default, 
+    produces an autoflushing, transactional session."""
+    
+    def __init__(self, bind=None, **kwargs):
+        """create a new transactional [sqlalchemy.orm.session#Session].
+    
+        The session starts a new transaction for each database accessed.  To
+        commit the transaction, use the commit() method.  SQL is issued for
+        write operations (i.e. flushes) automatically in most cases, before each query
+        and during commit.
+        """    
+    
+        kwargs.setdefault('autoflush', True)
+        kwargs.setdefault('transactional', True)
+        super(Session, self).__init__(bind=bind, **kwargs)
+    
 def relation(argument, secondary=None, **kwargs):
     """Provide a relationship of a primary Mapper to a secondary Mapper.
 
diff --git a/lib/sqlalchemy/orm/attributes.py b/lib/sqlalchemy/orm/attributes.py
index f783a381fb..d50fb5a815 100644
--- a/lib/sqlalchemy/orm/attributes.py
+++ b/lib/sqlalchemy/orm/attributes.py
@@ -565,7 +565,8 @@ class CommittedState(object):
                     for item in self.data[attr.key]:
                         collection.append_without_event(item)
             else:
-                del obj.__dict__[attr.key]
+                if attr.key in obj.__dict__:
+                    del obj.__dict__[attr.key]
 
     def __repr__(self):
         return "CommittedState: %s" % repr(self.data)
diff --git a/lib/sqlalchemy/orm/query.py b/lib/sqlalchemy/orm/query.py
index 9040655b28..6d4ae34096 100644
--- a/lib/sqlalchemy/orm/query.py
+++ b/lib/sqlalchemy/orm/query.py
@@ -43,6 +43,7 @@ class Query(object):
         self._from_obj = [self.table]
         self._populate_existing = False
         self._version_check = False
+        self._autoflush = True
         
     def _clone(self):
         q = Query.__new__(Query)
@@ -126,6 +127,11 @@ class Query(object):
         return Query(target, **kwargs).filter(criterion)
     query_from_parent = classmethod(query_from_parent)
     
+    def autoflush(self, setting):
+        q = self._clone()
+        q._autoflush = setting
+        return q
+        
     def populate_existing(self):
         """return a Query that will refresh all instances loaded.
         
@@ -621,8 +627,8 @@ class Query(object):
     def __iter__(self):
         statement = self.compile()
         statement.use_labels = True
-        if self.session.autoflush:
-            self.session.flush()
+        if self._autoflush and not self._populate_existing:
+            self.session._autoflush()
         return self._execute_and_instances(statement)
     
     def _execute_and_instances(self, statement):
diff --git a/lib/sqlalchemy/orm/session.py b/lib/sqlalchemy/orm/session.py
index 68c538034c..87604dde4c 100644
--- a/lib/sqlalchemy/orm/session.py
+++ b/lib/sqlalchemy/orm/session.py
@@ -111,6 +111,7 @@ class SessionTransaction(object):
             return
         for t in util.Set(self.__connections.values()):
             if t[2]:
+                # closing the connection will also issue a rollback()
                 t[0].close()
         self.session.transaction = None
 
@@ -133,7 +134,7 @@ class Session(object):
     of Sessions, see the ``sqlalchemy.ext.sessioncontext`` module.
     """
 
-    def __init__(self, bind=None, autoflush=False, transactional=False, twophase=False, echo_uow=False, weak_identity_map=False):
+    def __init__(self, bind=None, autoflush=True, transactional=False, twophase=False, echo_uow=False, weak_identity_map=False):
         self.uow = unitofwork.UnitOfWork(weak_identity_map=weak_identity_map)
 
         self.bind = bind
@@ -143,7 +144,7 @@ class Session(object):
         self.transaction = None
         self.hash_key = id(self)
         self.autoflush = autoflush
-        self.transactional = transactional or autoflush
+        self.transactional = transactional
         self.twophase = twophase
         self._query_cls = query.Query
         self._mapper_flush_opts = {}
@@ -174,15 +175,20 @@ class Session(object):
     
     def rollback(self):
         if self.transaction is None:
-            raise exceptions.InvalidRequestError("No transaction is begun.")
+            pass
         else:
             self.transaction = self.transaction.rollback()
+        attribute_manager.rollback(*self.identity_map.values())
         if self.transaction is None and self.transactional:
             self.begin()
             
     def commit(self):
         if self.transaction is None:
-            raise exceptions.InvalidRequestError("No transaction is begun.")
+            if self.transactional:
+                self.begin()
+                self.transaction = self.transaction.commit()
+            else:
+                raise exceptions.InvalidRequestError("No transaction is begun.")
         else:
             self.transaction = self.transaction.commit()
         if self.transaction is None and self.transactional:
@@ -230,12 +236,25 @@ class Session(object):
         return self.connection(mapper, close_with_result=True).scalar(clause, params or {}, **kwargs)
 
     def close(self):
-        """Close this Session."""
+        """Close this Session.  
+        
+        This clears all items and ends any transaction in progress.
+        """
 
         self.clear()
         if self.transaction is not None:
             self.transaction.close()
-
+        if self.transactional:
+            # note this doesnt use any connection resources
+            self.begin()
+    
+    def close_all(cls):
+        """Close *all* sessions in memory."""
+        
+        for sess in _sessions.values():
+            sess.close()
+    close_all = classmethod(close_all)
+    
     def clear(self):
         """Remove all object instances from this ``Session``.
 
@@ -345,6 +364,10 @@ class Session(object):
 
     sql = property(_sql)
 
+    def _autoflush(self):
+        if self.autoflush and (self.transaction is None or self.transaction.autoflush):
+            self.flush()
+
     def flush(self, objects=None):
         """Flush all the object modifications present in this session
         to the database.
diff --git a/lib/sqlalchemy/orm/shard.py b/lib/sqlalchemy/orm/shard.py
index cc13f8c1fe..26d03372f3 100644
--- a/lib/sqlalchemy/orm/shard.py
+++ b/lib/sqlalchemy/orm/shard.py
@@ -1,6 +1,8 @@
 from sqlalchemy.orm.session import Session
 from sqlalchemy.orm import Query
 
+__all__ = ['ShardedSession', 'ShardedQuery']
+
 class ShardedSession(Session):
     def __init__(self, shard_chooser, id_chooser, query_chooser, **kwargs):
         """construct a ShardedSession.
diff --git a/lib/sqlalchemy/orm/strategies.py b/lib/sqlalchemy/orm/strategies.py
index 54a4c65247..8a0cc60357 100644
--- a/lib/sqlalchemy/orm/strategies.py
+++ b/lib/sqlalchemy/orm/strategies.py
@@ -323,7 +323,7 @@ class LazyLoader(AbstractRelationLoader):
 
             # if we have a simple straight-primary key load, use mapper.get()
             # to possibly save a DB round trip
-            q = session.query(self.mapper)
+            q = session.query(self.mapper).autoflush(False)
             if self.use_get:
                 params = {}
                 for col, bind in self.lazybinds.iteritems():
diff --git a/lib/sqlalchemy/schema.py b/lib/sqlalchemy/schema.py
index 737a992f6d..21822e3a46 100644
--- a/lib/sqlalchemy/schema.py
+++ b/lib/sqlalchemy/schema.py
@@ -728,7 +728,7 @@ class ForeignKey(SchemaItem):
                     schema = None
                 else:
                     (schema,tname,colname) = m.group(1,2,3)
-                if tname not in parenttable.metadata:
+                if _get_table_key(tname, schema) not in parenttable.metadata:
                     raise exceptions.InvalidRequestError("Could not find table '%s' with which to generate a foreign key" % tname)
                 table = Table(tname, parenttable.metadata, mustexist=True, schema=schema)
                 try:
diff --git a/test/orm/fixtures.py b/test/orm/fixtures.py
index 73449eb96c..89df774645 100644
--- a/test/orm/fixtures.py
+++ b/test/orm/fixtures.py
@@ -28,6 +28,11 @@ class Base(object):
                     continue
                 value = getattr(self, attr)
                 if hasattr(value, '__iter__') and not isinstance(value, basestring):
+                    try:
+                        # catch AttributeError so that lazy loaders trigger
+                        otherattr = getattr(other, attr)
+                    except AttributeError:
+                        return False
                     if len(value) != len(getattr(other, attr)):
                        return False
                     for (us, them) in zip(value, getattr(other, attr)):
diff --git a/test/orm/lazy_relations.py b/test/orm/lazy_relations.py
index 6baba83941..4fe3b354ac 100644
--- a/test/orm/lazy_relations.py
+++ b/test/orm/lazy_relations.py
@@ -46,6 +46,7 @@ class LazyTest(QueryTest):
             u = q.filter(users.c.id == 7).first()
             sess.expunge(u)
             assert User(id=7, addresses=[Address(id=1, email_address='jack@bean.com')]) == u
+            assert False
         except exceptions.InvalidRequestError, err:
             assert "not bound to a Session, and no contextual session" in str(err)
 
diff --git a/test/orm/query.py b/test/orm/query.py
index e02c5a6432..1205221187 100644
--- a/test/orm/query.py
+++ b/test/orm/query.py
@@ -80,7 +80,7 @@ class GetTest(QueryTest):
         assert u is not u2
         
         u2.name = 'some name'
-        a = Address(name='some other name')
+        a = Address(email_address='some other name')
         u2.addresses.append(a)
         assert u2 in s.dirty
         assert a in u2.addresses
diff --git a/test/orm/session.py b/test/orm/session.py
index 4332796737..925f614972 100644
--- a/test/orm/session.py
+++ b/test/orm/session.py
@@ -97,7 +97,7 @@ class SessionTest(AssertMixin):
         conn1 = testbase.db.connect()
         conn2 = testbase.db.connect()
         
-        sess = create_session(autoflush=True, bind=conn1)
+        sess = Session(bind=conn1)
         u = User()
         u.user_name='ed'
         sess.save(u)
@@ -115,7 +115,7 @@ class SessionTest(AssertMixin):
         mapper(User, users)
 
         try:
-            sess = create_session(autoflush=True)
+            sess = Session()
             u = User()
             u.user_name='ed'
             sess.save(u)
@@ -136,20 +136,42 @@ class SessionTest(AssertMixin):
         conn1 = testbase.db.connect()
         conn2 = testbase.db.connect()
         
-        sess = create_session(autoflush=True, bind=conn1)
+        sess = Session(bind=conn1)
         u = User()
         u.user_name='ed'
         sess.save(u)
         sess.commit()
         assert conn1.execute("select count(1) from users").scalar() == 1
         assert testbase.db.connect().execute("select count(1) from users").scalar() == 1
+
+    def test_autoflush_rollback(self):
+        tables.data()
+        mapper(Address, addresses)
+        mapper(User, users, properties={
+            'addresses':relation(Address)
+        })
+        
+        sess = Session()
+        u = sess.query(User).get(8)
+        newad = Address()
+        newad.email_address == 'something new'
+        u.addresses.append(newad)
+        u.user_name = 'some new name'
+        assert u.user_name == 'some new name'
+        assert len(u.addresses) == 4
+        assert newad in u.addresses
+        sess.rollback()
+        assert u.user_name == 'ed'
+        assert len(u.addresses) == 3
+        assert newad not in u.addresses
+        
         
     def test_external_joined_transaction(self):
         class User(object):pass
         mapper(User, users)
         conn = testbase.db.connect()
         trans = conn.begin()
-        sess = create_session(bind=conn)
+        sess = Session(conn)
         sess.begin() 
         u = User()
         sess.save(u)
@@ -165,7 +187,7 @@ class SessionTest(AssertMixin):
         try:
             conn = testbase.db.connect()
             trans = conn.begin()
-            sess = create_session(bind=conn)
+            sess = Session(conn)
             u1 = User()
             sess.save(u1)
             sess.flush()
@@ -192,7 +214,7 @@ class SessionTest(AssertMixin):
         mapper(Address, addresses)
         
         engine2 = create_engine(testbase.db.url)
-        sess = create_session(twophase=True)
+        sess = Session(transactional=False, autoflush=False, twophase=True)
         sess.bind_mapper(User, testbase.db)
         sess.bind_mapper(Address, engine2)
         sess.begin()
@@ -211,8 +233,7 @@ class SessionTest(AssertMixin):
     def test_joined_transaction(self):
         class User(object):pass
         mapper(User, users)
-        sess = create_session()
-        sess.begin()
+        sess = Session()
         sess.begin()  
         u = User()
         sess.save(u)
diff --git a/test/orm/unitofwork.py b/test/orm/unitofwork.py
index a8ce797f24..b67a6faf3b 100644
--- a/test/orm/unitofwork.py
+++ b/test/orm/unitofwork.py
@@ -15,14 +15,15 @@ from testlib import tables
 class UnitOfWorkTest(AssertMixin):
     def setUpAll(self):
         global ctx, assign_mapper
-        ctx = SessionContext(create_session)
+        ctx = SessionContext(Session)
         def assign_mapper(*args, **kwargs):
             return assignmapper.assign_mapper(ctx, *args, **kwargs)
         global_extensions.append(ctx.mapper_extension)
     def tearDownAll(self):
         global_extensions.remove(ctx.mapper_extension)
     def tearDown(self):
-        ctx.current.clear()
+        Session.close_all()
+        ctx.current.close()
         clear_mappers()
 
 class HistoryTest(UnitOfWorkTest):
@@ -37,7 +38,7 @@ class HistoryTest(UnitOfWorkTest):
         UnitOfWorkTest.tearDownAll(self)
         
     def testbackref(self):
-        s = create_session()
+        s = Session()
         class User(object):pass
         class Address(object):pass
         am = mapper(Address, addresses)
@@ -51,17 +52,16 @@ class HistoryTest(UnitOfWorkTest):
         #print repr(a.__class__._attribute_manager.get_history(a, 'user').added_items())
         #print repr(u.addresses.added_items())
         self.assert_(u.addresses == [a])
-        s.flush()
+        s.commit()
 
-        s.clear()
+        s.close()
         u = s.query(m).select()[0]
         print u.addresses[0].user
-
             
 class VersioningTest(UnitOfWorkTest):
     def setUpAll(self):
         UnitOfWorkTest.setUpAll(self)
-        ctx.current.clear()
+        ctx.current.close()
         global version_table
         version_table = Table('version_test', MetaData(testbase.db),
         Column('id', Integer, Sequence('version_test_seq'), primary_key=True ),
@@ -70,33 +70,34 @@ class VersioningTest(UnitOfWorkTest):
         )
         version_table.create()
     def tearDownAll(self):
-        version_table.drop()
         UnitOfWorkTest.tearDownAll(self)
+        version_table.drop()
     def tearDown(self):
-        version_table.delete().execute()
         UnitOfWorkTest.tearDown(self)
+        version_table.delete().execute()
     
     def testbasic(self):
-        s = create_session()
+        s = Session()
         class Foo(object):pass
         assign_mapper(Foo, version_table, version_id_col=version_table.c.version_id)
         f1 =Foo(value='f1', _sa_session=s)
         f2 = Foo(value='f2', _sa_session=s)
-        s.flush()
+        s.commit()
         
         f1.value='f1rev2'
-        s.flush()
-        s2 = create_session()
+        s.commit()
+        s2 = Session()
         f1_s = s2.query(Foo).get(f1.id)
         f1_s.value='f1rev3'
-        s2.flush()
+        s2.commit()
 
         f1.value='f1rev3mine'
         success = False
         try:
             # a concurrent session has modified this, should throw
             # an exception
-            s.flush()
+            s.commit()
+            assert False
         except exceptions.ConcurrentModificationError, e:
             #print e
             success = True
@@ -105,18 +106,18 @@ class VersioningTest(UnitOfWorkTest):
         if testbase.db.dialect.supports_sane_rowcount():
             assert success
         
-        s.clear()
+        s.close()
         f1 = s.query(Foo).get(f1.id)
         f2 = s.query(Foo).get(f2.id)
         
         f1_s.value='f1rev4'
-        s2.flush()
-    
+        s2.commit()
+
         s.delete(f1)
         s.delete(f2)
         success = False
         try:
-            s.flush()
+            s.commit()
         except exceptions.ConcurrentModificationError, e:
             #print e
             success = True
@@ -125,15 +126,15 @@ class VersioningTest(UnitOfWorkTest):
 
     def testversioncheck(self):
         """test that query.with_lockmode performs a 'version check' on an already loaded instance"""
-        s1 = create_session()
+        s1 = Session()
         class Foo(object):pass
         assign_mapper(Foo, version_table, version_id_col=version_table.c.version_id)
         f1s1 =Foo(value='f1', _sa_session=s1)
-        s1.flush()
-        s2 = create_session()
+        s1.commit()
+        s2 = Session()
         f1s2 = s2.query(Foo).get(f1s1.id)
         f1s2.value='f1 new value'
-        s2.flush()
+        s2.commit()
         try:
             # load, version is wrong
             s1.query(Foo).with_lockmode('read').get(f1s1.id)
@@ -146,25 +147,24 @@ class VersioningTest(UnitOfWorkTest):
         s1.query(Foo).with_lockmode('read').get(f1s1.id)
         
         # assert brand new load is OK too
-        s1.clear()
+        s1.close()
         s1.query(Foo).with_lockmode('read').get(f1s1.id)
         
     def testnoversioncheck(self):
         """test that query.with_lockmode works OK when the mapper has no version id col"""
-        s1 = create_session()
+        s1 = Session()
         class Foo(object):pass
         assign_mapper(Foo, version_table)
         f1s1 =Foo(value='f1', _sa_session=s1)
         f1s1.version_id=0
-        s1.flush()
-        s2 = create_session()
+        s1.commit()
+        s2 = Session()
         f1s2 = s2.query(Foo).with_lockmode('read').get(f1s1.id)
         assert f1s2.id == f1s1.id
         assert f1s2.value == f1s1.value
         
 class UnicodeTest(UnitOfWorkTest):
     def setUpAll(self):
-        UnitOfWorkTest.setUpAll(self)
         global metadata, uni_table, uni_table2
         metadata = MetaData(testbase.db)
         uni_table = Table('uni_test', metadata,
@@ -174,13 +174,18 @@ class UnicodeTest(UnitOfWorkTest):
             Column('id',  Integer, Sequence("uni2_test_id_seq", optional=True), primary_key=True),
             Column('txt', Unicode(50), ForeignKey(uni_table.c.txt)))
         metadata.create_all()
+        UnitOfWorkTest.setUpAll(self)
+
     def tearDownAll(self):
-        metadata.drop_all()
         UnitOfWorkTest.tearDownAll(self)
+        metadata.drop_all()
+
     def tearDown(self):
+        UnitOfWorkTest.tearDown(self)
         clear_mappers()
         for t in metadata.table_iterator(reverse=True):
             t.delete().execute()
+            
     def testbasic(self):
         class Test(object):
             def __init__(self, id, txt):
@@ -191,7 +196,7 @@ class UnicodeTest(UnitOfWorkTest):
         txt = u"\u0160\u0110\u0106\u010c\u017d"
         t1 = Test(id=1, txt = txt)
         self.assert_(t1.txt == txt)
-        ctx.current.flush()
+        ctx.current.commit()
         self.assert_(t1.txt == txt)
     def testrelation(self):
         class Test(object):
@@ -208,8 +213,8 @@ class UnicodeTest(UnitOfWorkTest):
         t1 = Test(txt=txt)
         t1.t2s.append(Test2())
         t1.t2s.append(Test2())
-        ctx.current.flush()
-        ctx.current.clear()
+        ctx.current.commit()
+        ctx.current.close()
         t1 = ctx.current.query(Test).get_by(id=t1.id)
         assert len(t1.t2s) == 2
 
@@ -233,13 +238,13 @@ class MutableTypesTest(UnitOfWorkTest):
         mapper(Foo, table)
         f1 = Foo()
         f1.data = pickleable.Bar(4,5)
-        ctx.current.flush()
-        ctx.current.clear()
+        ctx.current.commit()
+        ctx.current.close()
         f2 = ctx.current.query(Foo).get_by(id=f1.id)
         assert f2.data == f1.data
         f2.data.y = 19
-        ctx.current.flush()
-        ctx.current.clear()
+        ctx.current.commit()
+        ctx.current.close()
         f3 = ctx.current.query(Foo).get_by(id=f1.id)
         print f2.data, f3.data
         assert f3.data != f1.data
@@ -252,12 +257,12 @@ class MutableTypesTest(UnitOfWorkTest):
         f1 = Foo()
         f1.data = pickleable.Bar(4,5)
         f1.value = unicode('hi')
-        ctx.current.flush()
+        ctx.current.commit()
         def go():
-            ctx.current.flush()
+            ctx.current.commit()
         self.assert_sql_count(testbase.db, go, 0)
         f1.value = unicode('someothervalue')
-        self.assert_sql(testbase.db, lambda: ctx.current.flush(), [
+        self.assert_sql(testbase.db, lambda: ctx.current.commit(), [
             (
                 "UPDATE mutabletest SET value=:value WHERE mutabletest.id = :mutabletest_id",
                 {'mutabletest_id': f1.id, 'value': u'someothervalue'}
@@ -265,7 +270,7 @@ class MutableTypesTest(UnitOfWorkTest):
         ])
         f1.value = unicode('hi')
         f1.data.x = 9
-        self.assert_sql(testbase.db, lambda: ctx.current.flush(), [
+        self.assert_sql(testbase.db, lambda: ctx.current.commit(), [
             (
                 "UPDATE mutabletest SET data=:data, value=:value WHERE mutabletest.id = :mutabletest_id",
                 {'mutabletest_id': f1.id, 'value': u'hi', 'data':f1.data}
@@ -279,32 +284,32 @@ class MutableTypesTest(UnitOfWorkTest):
         mapper(Foo, table)
         f1 = Foo()
         f1.data = pickleable.BarWithoutCompare(4,5)
-        ctx.current.flush()
+        ctx.current.commit()
         
         def go():
-            ctx.current.flush()
+            ctx.current.commit()
         self.assert_sql_count(testbase.db, go, 0)
         
-        ctx.current.clear()
+        ctx.current.close()
 
         f2 = ctx.current.query(Foo).get_by(id=f1.id)
 
         def go():
-            ctx.current.flush()
+            ctx.current.commit()
         self.assert_sql_count(testbase.db, go, 0)
 
         f2.data.y = 19
         def go():
-            ctx.current.flush()
+            ctx.current.commit()
         self.assert_sql_count(testbase.db, go, 1)
         
-        ctx.current.clear()
+        ctx.current.close()
         f3 = ctx.current.query(Foo).get_by(id=f1.id)
         print f2.data, f3.data
         assert (f3.data.x, f3.data.y) == (4,19)
 
         def go():
-            ctx.current.flush()
+            ctx.current.commit()
         self.assert_sql_count(testbase.db, go, 0)
         
     def testunicode(self):
@@ -316,12 +321,12 @@ class MutableTypesTest(UnitOfWorkTest):
         mapper(Foo, table)
         f1 = Foo()
         f1.value = u'hi'
-        ctx.current.flush()
-        ctx.current.clear()
+        ctx.current.commit()
+        ctx.current.close()
         f1 = ctx.current.get(Foo, f1.id)
         f1.value = u'hi'
         def go():
-            ctx.current.flush()
+            ctx.current.commit()
         self.assert_sql_count(testbase.db, go, 0)
         
         
@@ -366,8 +371,8 @@ class PKTest(UnitOfWorkTest):
         e.name = 'entry1'
         e.value = 'this is entry 1'
         e.multi_rev = 2
-        ctx.current.flush()
-        ctx.current.clear()
+        ctx.current.commit()
+        ctx.current.close()
         e2 = Query(Entry).get((e.multi_id, 2))
         self.assert_(e is not e2 and e._instance_key == e2._instance_key)
         
@@ -380,7 +385,7 @@ class PKTest(UnitOfWorkTest):
         e.pk_col_1 = 'pk1'
         e.pk_col_2 = 'pk1_related'
         e.data = 'im the data'
-        ctx.current.flush()
+        ctx.current.commit()
         
     def testkeypks(self):
         import datetime
@@ -392,7 +397,7 @@ class PKTest(UnitOfWorkTest):
         e.secondary = 'pk2'
         e.assigned = datetime.date.today()
         e.data = 'some more data'
-        ctx.current.flush()
+        ctx.current.commit()
 
     def testpksimmutable(self):
         class Entry(object):
@@ -402,11 +407,11 @@ class PKTest(UnitOfWorkTest):
         e.multi_id=5
         e.multi_rev=5
         e.name='somename'
-        ctx.current.flush()
+        ctx.current.commit()
         e.multi_rev=6
         e.name = 'someothername'
         try:
-            ctx.current.flush()
+            ctx.current.commit()
             assert False
         except exceptions.FlushError, fe:
             assert str(fe) == "Can't change the identity of instance Entry@%s in session (existing identity: (%s, (5, 5), None); new identity: (%s, (5, 6), None))" % (hex(id(e)), repr(e.__class__), repr(e.__class__))
@@ -452,7 +457,7 @@ class ForeignPKTest(UnitOfWorkTest):
         ps = PersonSite()
         ps.site = 'asdf'
         p.sites.append(ps)
-        ctx.current.flush()
+        ctx.current.commit()
         assert people.count(people.c.person=='im the key').scalar() == peoplesites.count(peoplesites.c.person=='im the key').scalar() == 1
 
 class PassiveDeletesTest(UnitOfWorkTest):
@@ -499,12 +504,12 @@ class PassiveDeletesTest(UnitOfWorkTest):
         mc.children.append(MyOtherClass())
         mc.children.append(MyOtherClass())
         sess.save(mc)
-        sess.flush()
-        sess.clear()
+        sess.commit()
+        sess.close()
         assert myothertable.count().scalar() == 4
         mc = sess.query(MyClass).get(mc.id)
         sess.delete(mc)
-        sess.flush()
+        sess.commit()
         assert mytable.count().scalar() == 0
         assert myothertable.count().scalar() == 0
         
@@ -550,7 +555,7 @@ class DefaultTest(UnitOfWorkTest):
         h3 = Hoho(hoho=self.althohoval, counter=12)
         h4 = Hoho()
         h5 = Hoho(foober='im the new foober')
-        ctx.current.flush()
+        ctx.current.commit()
         
         self.assert_(h1.hoho==self.althohoval)
         self.assert_(h3.hoho==self.althohoval)
@@ -570,7 +575,7 @@ class DefaultTest(UnitOfWorkTest):
             self.assert_(h5.foober=='im the new foober')
         self.assert_sql_count(testbase.db, go, 0)
         
-        ctx.current.clear()
+        ctx.current.close()
         
         l = Query(Hoho).select()
         
@@ -591,7 +596,7 @@ class DefaultTest(UnitOfWorkTest):
         
         h1 = Hoho(hoho="15", counter="15")
         
-        ctx.current.flush()
+        ctx.current.commit()
         def go():
             self.assert_(h1.hoho=="15")
             self.assert_(h1.counter=="15")
@@ -602,10 +607,10 @@ class DefaultTest(UnitOfWorkTest):
         class Hoho(object):pass
         assign_mapper(Hoho, default_table)
         h1 = Hoho()
-        ctx.current.flush()
+        ctx.current.commit()
         self.assert_(h1.foober == 'im foober')
         h1.counter = 19
-        ctx.current.flush()
+        ctx.current.commit()
         self.assert_(h1.foober == 'im the update')
 
 class OneToManyTest(UnitOfWorkTest):
@@ -634,7 +639,7 @@ class OneToManyTest(UnitOfWorkTest):
         a2.email_address = 'lala@test.org'
         u.addresses.append(a2)
         print repr(u.addresses)
-        ctx.current.flush()
+        ctx.current.commit()
 
         usertable = users.select(users.c.user_id.in_(u.user_id)).execute().fetchall()
         self.assertEqual(usertable[0].values(), [u.user_id, 'one2manytester'])
@@ -647,7 +652,7 @@ class OneToManyTest(UnitOfWorkTest):
 
         a2.email_address = 'somethingnew@foo.com'
 
-        ctx.current.flush()
+        ctx.current.commit()
 
         addresstable = addresses.select(addresses.c.address_id == addressid).execute().fetchall()
         self.assertEqual(addresstable[0].values(), [addressid, userid, 'somethingnew@foo.com'])
@@ -675,7 +680,7 @@ class OneToManyTest(UnitOfWorkTest):
         a3 = Address()
         a3.email_address = 'emailaddress3'
 
-        ctx.current.flush()
+        ctx.current.commit()
 
         # modify user2 directly, append an address to user1.
         # upon commit, user2 should be updated, user1 should not
@@ -683,7 +688,7 @@ class OneToManyTest(UnitOfWorkTest):
         u2.user_name = 'user2modified'
         u1.addresses.append(a3)
         del u1.addresses[0]
-        self.assert_sql(testbase.db, lambda: ctx.current.flush(), 
+        self.assert_sql(testbase.db, lambda: ctx.current.commit(), 
                 [
                     (
                         "UPDATE users SET user_name=:user_name WHERE users.user_id = :users_user_id",
@@ -711,12 +716,12 @@ class OneToManyTest(UnitOfWorkTest):
         a = Address()
         a.email_address = 'address1'
         u1.addresses.append(a)
-        ctx.current.flush()
+        ctx.current.commit()
         del u1.addresses[0]
         u2.addresses.append(a)
         ctx.current.delete(u1)
-        ctx.current.flush()
-        ctx.current.clear()
+        ctx.current.commit()
+        ctx.current.close()
         u2 = ctx.current.get(User, u2.user_id)
         assert len(u2.addresses) == 1
 
@@ -731,11 +736,11 @@ class OneToManyTest(UnitOfWorkTest):
         a = Address()
         a.email_address = 'address1'
         u1.addresses.append(a)
-        ctx.current.flush()
+        ctx.current.commit()
         del u1.addresses[0]
         u2.addresses.append(a)
-        ctx.current.flush()
-        ctx.current.clear()
+        ctx.current.commit()
+        ctx.current.close()
         u2 = ctx.current.get(User, u2.user_id)
         assert len(u2.addresses) == 1
 
@@ -748,9 +753,9 @@ class OneToManyTest(UnitOfWorkTest):
         u.user_name = 'one2onetester'
         u.address = a
         u.address.email_address = 'myonlyaddress@foo.com'
-        ctx.current.flush()
+        ctx.current.commit()
         ctx.current.delete(u)
-        ctx.current.flush()
+        ctx.current.commit()
         self.assert_(a.address_id is not None and a.user_id is None and not ctx.current.identity_map.has_key(u._instance_key) and ctx.current.identity_map.has_key(a._instance_key))
 
     def testonetoone(self):
@@ -761,11 +766,11 @@ class OneToManyTest(UnitOfWorkTest):
         u.user_name = 'one2onetester'
         u.address = Address()
         u.address.email_address = 'myonlyaddress@foo.com'
-        ctx.current.flush()
+        ctx.current.commit()
         u.user_name = 'imnew'
-        ctx.current.flush()
+        ctx.current.commit()
         u.address.email_address = 'imnew@foo.com'
-        ctx.current.flush()
+        ctx.current.commit()
 
     def testbidirectional(self):
         m1 = mapper(User, users)
@@ -781,7 +786,7 @@ class OneToManyTest(UnitOfWorkTest):
         a = Address()
         a.email_address = 'testaddress'
         a.user = u
-        ctx.current.flush()
+        ctx.current.commit()
         print repr(u.addresses)
         x = False
         try:
@@ -794,7 +799,7 @@ class OneToManyTest(UnitOfWorkTest):
             self.assert_(False, "User addresses element should be scalar based")
 
         ctx.current.delete(u)
-        ctx.current.flush()
+        ctx.current.commit()
 
     def testdoublerelation(self):
         m2 = mapper(Address, addresses)
@@ -814,7 +819,7 @@ class OneToManyTest(UnitOfWorkTest):
 
         u.boston_addresses.append(a)
         u.newyork_addresses.append(b)
-        ctx.current.flush()
+        ctx.current.commit()
 
 class SaveTest(UnitOfWorkTest):
 
@@ -851,7 +856,7 @@ class SaveTest(UnitOfWorkTest):
         ctx.current.save(u)
         
         ctx.current.flush([u])
-        ctx.current.flush()
+        ctx.current.commit()
 
         # assert the first one retreives the same from the identity map
         nu = ctx.current.get(m, u.user_id)
@@ -859,7 +864,7 @@ class SaveTest(UnitOfWorkTest):
         self.assert_(u is nu)
         
         # clear out the identity map, so next get forces a SELECT
-        ctx.current.clear()
+        ctx.current.close()
 
         # check it again, identity should be different but ids the same
         nu = ctx.current.get(m, u.user_id)
@@ -869,10 +874,10 @@ class SaveTest(UnitOfWorkTest):
         ctx.current.update(u)
         u.user_name = 'modifiedname'
         assert u in ctx.current.dirty
-        ctx.current.flush()
+        ctx.current.commit()
 
         # select both
-        #ctx.current.clear()
+        #ctx.current.close()
         userlist = Query(m).select(users.c.user_id.in_(u.user_id, u2.user_id), order_by=[users.c.user_name])
         print repr(u.user_id), repr(userlist[0].user_id), repr(userlist[0].user_name)
         self.assert_(u.user_id == userlist[0].user_id and userlist[0].user_name == 'modifiedname')
@@ -890,12 +895,12 @@ class SaveTest(UnitOfWorkTest):
         u.addresses.append(Address())
         u.addresses.append(Address())
         u.addresses.append(Address())
-        ctx.current.flush()
-        ctx.current.clear()
+        ctx.current.commit()
+        ctx.current.close()
         ulist = ctx.current.query(m1).select()
         u1 = ulist[0]
         u1.user_name = 'newname'
-        ctx.current.flush()
+        ctx.current.commit()
         self.assert_(len(u1.addresses) == 4)
         
     def testinherits(self):
@@ -912,30 +917,30 @@ class SaveTest(UnitOfWorkTest):
                 )
         
         au = AddressUser()
-        ctx.current.flush()
-        ctx.current.clear()
+        ctx.current.commit()
+        ctx.current.close()
         l = ctx.current.query(AddressUser).selectone()
         self.assert_(l.user_id == au.user_id and l.address_id == au.address_id)
     
     def testdeferred(self):
-        """test that a deferred load within a flush() doesnt screw up the connection"""
+        """test that a deferred load within a commit() doesnt screw up the connection"""
         mapper(User, users, properties={
             'user_name':deferred(users.c.user_name)
         })
         u = User()
         u.user_id=42
-        ctx.current.flush()
+        ctx.current.commit()
     
     def test_dont_update_blanks(self):
         mapper(User, users)
         u = User()
         u.user_name = ""
-        ctx.current.flush()
-        ctx.current.clear()
+        ctx.current.commit()
+        ctx.current.close()
         u = ctx.current.query(User).get(u.user_id)
         u.user_name = ""
         def go():
-            ctx.current.flush()
+            ctx.current.commit()
         self.assert_sql_count(testbase.db, go, 0)
 
     def testmultitable(self):
@@ -953,10 +958,10 @@ class SaveTest(UnitOfWorkTest):
         u.user_name = 'multitester'
         u.email = 'multi@test.org'
 
-        ctx.current.flush()
+        ctx.current.commit()
         id = m.primary_key_from_instance(u)
 
-        ctx.current.clear()
+        ctx.current.close()
         
         u = ctx.current.get(User, id)
         assert u.user_name == 'multitester'
@@ -968,14 +973,14 @@ class SaveTest(UnitOfWorkTest):
 
         u.email = 'lala@hey.com'
         u.user_name = 'imnew'
-        ctx.current.flush()
+        ctx.current.commit()
 
         usertable = users.select(users.c.user_id.in_(u.foo_id)).execute().fetchall()
         self.assertEqual(usertable[0].values(), [u.foo_id, 'imnew'])
         addresstable = addresses.select(addresses.c.address_id.in_(u.address_id)).execute().fetchall()
         self.assertEqual(addresstable[0].values(), [u.address_id, u.foo_id, 'lala@hey.com'])
 
-        ctx.current.clear()
+        ctx.current.close()
         u = ctx.current.get(User, id)
         assert u.user_name == 'imnew'
     
@@ -989,11 +994,11 @@ class SaveTest(UnitOfWorkTest):
         u = User()
         u.addresses.append(Address())
         u.addresses.append(Address())
-        ctx.current.flush()
-        ctx.current.clear()
+        ctx.current.commit()
+        ctx.current.close()
         u = ctx.current.query(User).get(u.user_id)
         ctx.current.delete(u)
-        ctx.current.flush()
+        ctx.current.commit()
         assert users.count().scalar() == 0
         assert addresses.count().scalar() == 0
         
@@ -1010,7 +1015,7 @@ class SaveTest(UnitOfWorkTest):
         u1.username = 'user1'
         u2 = User()
         u2.username = 'user2'
-        ctx.current.flush()
+        ctx.current.commit()
         
         clear_mappers()
         
@@ -1020,7 +1025,7 @@ class SaveTest(UnitOfWorkTest):
         u2 = User()
         u2.username = 'user2'
         try:
-            ctx.current.flush()
+            ctx.current.commit()
             assert False
         except AssertionError:
             assert True
@@ -1057,11 +1062,11 @@ class ManyToOneTest(UnitOfWorkTest):
             a.user.user_name = elem['user_name']
             objects.append(a)
             
-        ctx.current.flush()
+        ctx.current.commit()
         objects[2].email_address = 'imnew@foo.bar'
         objects[3].user = User()
         objects[3].user.user_name = 'imnewlyadded'
-        self.assert_sql(testbase.db, lambda: ctx.current.flush(), [
+        self.assert_sql(testbase.db, lambda: ctx.current.commit(), [
                 (
                     "INSERT INTO users (user_name) VALUES (:user_name)",
                     {'user_name': 'imnewlyadded'}
@@ -1105,15 +1110,15 @@ class ManyToOneTest(UnitOfWorkTest):
         u1.user_name='user1'
         
         a1.user = u1
-        ctx.current.flush()
-        ctx.current.clear()
+        ctx.current.commit()
+        ctx.current.close()
         a1 = ctx.current.query(Address).get(a1.address_id)
         u1 = ctx.current.query(User).get(u1.user_id)
         assert a1.user is u1
 
         a1.user = None
-        ctx.current.flush()
-        ctx.current.clear()
+        ctx.current.commit()
+        ctx.current.close()
         a1 = ctx.current.query(Address).get(a1.address_id)
         u1 = ctx.current.query(User).get(u1.user_id)
         assert a1.user is None
@@ -1130,16 +1135,16 @@ class ManyToOneTest(UnitOfWorkTest):
         u1.user_name='user1'
 
         a1.user = u1
-        ctx.current.flush()
-        ctx.current.clear()
+        ctx.current.commit()
+        ctx.current.close()
         a1 = ctx.current.query(Address).get(a1.address_id)
         a2 = ctx.current.query(Address).get(a2.address_id)
         u1 = ctx.current.query(User).get(u1.user_id)
         assert a1.user is u1
         a1.user = None
         a2.user = u1
-        ctx.current.flush()
-        ctx.current.clear()
+        ctx.current.commit()
+        ctx.current.close()
         a1 = ctx.current.query(Address).get(a1.address_id)
         a2 = ctx.current.query(Address).get(a2.address_id)
         u1 = ctx.current.query(User).get(u1.user_id)
@@ -1158,16 +1163,16 @@ class ManyToOneTest(UnitOfWorkTest):
         u2.user_name='user2'
 
         a1.user = u1
-        ctx.current.flush()
-        ctx.current.clear()
+        ctx.current.commit()
+        ctx.current.close()
         a1 = ctx.current.query(Address).get(a1.address_id)
         u1 = ctx.current.query(User).get(u1.user_id)
         u2 = ctx.current.query(User).get(u2.user_id)
         assert a1.user is u1
         
         a1.user = u2
-        ctx.current.flush()
-        ctx.current.clear()
+        ctx.current.commit()
+        ctx.current.close()
         a1 = ctx.current.query(Address).get(a1.address_id)
         u1 = ctx.current.query(User).get(u1.user_id)
         u2 = ctx.current.query(User).get(u2.user_id)
@@ -1222,7 +1227,7 @@ class ManyToManyTest(UnitOfWorkTest):
                     k.name = kname
                 item.keywords.append(k)
 
-        ctx.current.flush()
+        ctx.current.commit()
         
         l = ctx.current.query(m).select(items.c.item_name.in_(*[e['item_name'] for e in data[1:]]), order_by=[items.c.item_name])
         self.assert_result(l, *data)
@@ -1231,7 +1236,7 @@ class ManyToManyTest(UnitOfWorkTest):
         k = Keyword()
         k.name = 'yellow'
         objects[5].keywords.append(k)
-        self.assert_sql(testbase.db, lambda:ctx.current.flush(), [
+        self.assert_sql(testbase.db, lambda:ctx.current.commit(), [
             {
                 "UPDATE items SET item_name=:item_name WHERE items.item_id = :items_item_id":
                 {'item_name': 'item4updated', 'items_item_id': objects[4].item_id}
@@ -1260,7 +1265,7 @@ class ManyToManyTest(UnitOfWorkTest):
         objects[2].keywords.append(k)
         dkid = objects[5].keywords[1].keyword_id
         del objects[5].keywords[1]
-        self.assert_sql(testbase.db, lambda:ctx.current.flush(), [
+        self.assert_sql(testbase.db, lambda:ctx.current.commit(), [
                 (
                     "DELETE FROM itemkeywords WHERE itemkeywords.item_id = :item_id AND itemkeywords.keyword_id = :keyword_id",
                     [{'item_id': objects[5].item_id, 'keyword_id': dkid}]
@@ -1272,7 +1277,7 @@ class ManyToManyTest(UnitOfWorkTest):
         ])
         
         ctx.current.delete(objects[3])
-        ctx.current.flush()
+        ctx.current.commit()
 
     def testmanytomanyremove(self):
         """tests that setting a list-based attribute to '[]' properly affects the history and allows
@@ -1288,11 +1293,11 @@ class ManyToManyTest(UnitOfWorkTest):
         k2 = Keyword()
         i.keywords.append(k1)
         i.keywords.append(k2)
-        ctx.current.flush()
+        ctx.current.commit()
         
         assert itemkeywords.count().scalar() == 2
         i.keywords = []
-        ctx.current.flush()
+        ctx.current.commit()
         assert itemkeywords.count().scalar() == 0
 
     def testscalar(self):
@@ -1305,9 +1310,9 @@ class ManyToManyTest(UnitOfWorkTest):
             ))
         
         i = Item()
-        ctx.current.flush()
+        ctx.current.commit()
         ctx.current.delete(i)
-        ctx.current.flush()
+        ctx.current.commit()
         
         
 
@@ -1332,14 +1337,14 @@ class ManyToManyTest(UnitOfWorkTest):
         item.keywords.append(k1)
         item.keywords.append(k2)
         item.keywords.append(k3)
-        ctx.current.flush()
+        ctx.current.commit()
         
         item.keywords = []
         item.keywords.append(k1)
         item.keywords.append(k2)
-        ctx.current.flush()
+        ctx.current.commit()
         
-        ctx.current.clear()
+        ctx.current.close()
         item = ctx.current.query(Item).get(item.item_id)
         print [k1, k2]
         print item.keywords
@@ -1404,8 +1409,8 @@ class ManyToManyTest(UnitOfWorkTest):
                 ik.keyword = k
                 item.keywords.append(ik)
 
-        ctx.current.flush()
-        ctx.current.clear()
+        ctx.current.commit()
+        ctx.current.close()
         l = Query(m).select(items.c.item_name.in_(*[e['item_name'] for e in data[1:]]), order_by=[items.c.item_name])
         self.assert_result(l, *data)
 
@@ -1430,10 +1435,10 @@ class ManyToManyTest(UnitOfWorkTest):
         k = KeywordUser()
         k.user_name = 'keyworduser'
         k.keyword_name = 'a keyword'
-        ctx.current.flush()
+        ctx.current.commit()
         
         id = (k.user_id, k.keyword_id)
-        ctx.current.clear()
+        ctx.current.close()
         k = ctx.current.query(KeywordUser).get(id)
         assert k.user_name == 'keyworduser'
         assert k.keyword_name == 'a keyword'
@@ -1442,7 +1447,7 @@ class ManyToManyTest(UnitOfWorkTest):
 class SaveTest2(UnitOfWorkTest):
 
     def setUp(self):
-        ctx.current.clear()
+        ctx.current.close()
         clear_mappers()
         global meta, users, addresses
         meta = MetaData(testbase.db)
@@ -1477,7 +1482,7 @@ class SaveTest2(UnitOfWorkTest):
             a.user = User()
             a.user.user_name = elem['user_name']
             objects.append(a)
-        self.assert_sql(testbase.db, lambda: ctx.current.flush(), [
+        self.assert_sql(testbase.db, lambda: ctx.current.commit(), [
                 (
                     "INSERT INTO users (user_name) VALUES (:user_name)",
                     {'user_name': 'thesub'}
@@ -1563,12 +1568,12 @@ class SaveTest3(UnitOfWorkTest):
         k2 = Keyword()
         i.keywords.append(k1)
         i.keywords.append(k2)
-        ctx.current.flush()
+        ctx.current.commit()
 
         assert t2.count().scalar() == 2
         i.keywords = []
         print i.keywords
-        ctx.current.flush()
+        ctx.current.commit()
         assert t2.count().scalar() == 0