def test_remove(self):
self.middleware.remove(None, None)
for db in databases.values():
db.remove.assert_called_once_with()
def test_rollback(self):
self.middleware.rollback(None, None)
for db in databases.values():
db.rollback.assert_called_once_with()
def test_flush(self):
self.middleware.flush(None, None)
for db in databases.values():
db.flush.assert_called_once_with()
def test_commit(self):
self.middleware.commit(None, None)
for db in databases.values():
db.commit.assert_called_once_with()