Commits

Andriy Kornatskyy committed 822855b

Ensure NullTPCSession can be enlist in active state.

Comments (0)

Files changed (2)

src/wheezy/core/db.py

     def enlist(self, session):
         """ Begins a TPC transaction with the given session.
         """
+        assert session
         assert self.status != SESSION_STATUS_IDLE
         self.enlised_sessions.append(session)
         session.__enter__()
         """ Ensure session is entered.
         """
         assert session
-        assert self.status == SESSION_STATUS_ENTERED
+        assert self.status != SESSION_STATUS_IDLE
         self.status = SESSION_STATUS_ACTIVE
 
     def commit(self):

src/wheezy/core/tests/test_db.py

         """
         self.session.__enter__()
         self.session.enlist('x')
+        self.session.enlist('y')
 
     def test_commit_raise_error(self):
         """ If session is not entered raise error.