def test_afterCompletion_ACTIVE(self): txn = FakeTransaction() txn.status = TransactionStatus.ACTIVE fake_session = FakeRabbitSession() sync = RabbitSessionTransactionSync(fake_session) sync.afterCompletion(txn) self.assertEqual(["reset"], fake_session.log)
def test_afterCompletion_COMMITTED(self): txn = FakeTransaction() txn.status = TransactionStatus.COMMITTED fake_session = FakeRabbitSession() sync = RabbitSessionTransactionSync(fake_session) sync.afterCompletion(txn) self.assertEqual(["finish"], fake_session.log)