def test_tx_with_statement_already_commited(self):
self._active_transaction = False
def on_tx(*_):
if not self._active_transaction:
channel.rpc.on_frame(specification.Tx.SelectOk())
self._active_transaction = True
return
self._active_transaction = False
channel.rpc.on_frame(specification.Tx.CommitOk())
connection = FakeConnection(on_write=on_tx)
channel = Channel(0, connection, 0.01)
channel.set_state(Channel.OPEN)
tx = Tx(channel)
with tx:
tx.commit()
self.assertFalse(tx._tx_active)
self.assertFalse(tx._tx_active)
def test_tx_commit(self):
def on_tx_commit(*_):
channel.rpc.on_frame(specification.Tx.CommitOk())
connection = FakeConnection(on_write=on_tx_commit)
channel = Channel(0, connection, 0.01)
channel.set_state(Channel.OPEN)
tx = Tx(channel)
self.assertIsInstance(tx.commit(), dict)
self.assertFalse(tx._tx_active)
def test_tx_with_statement_when_raises(self):
def on_tx(_, frame):
if isinstance(frame, specification.Tx.Select):
channel.rpc.on_frame(specification.Tx.SelectOk())
return
channel.rpc.on_frame(specification.Tx.CommitOk())
connection = FakeConnection(on_write=on_tx)
channel = Channel(0, connection, 0.01)
channel.set_state(Channel.OPEN)
tx = Tx(channel)
try:
with tx:
tx.commit()
raise Exception('travis-ci')
except Exception:
self.assertEqual(self.get_last_log(),
'Leaving Transaction on exception: travis-ci')
self.assertFalse(tx._tx_active)