def test_rollback_transcation(self):
db_batch = flexmock()
db_batch.should_receive('valid_data_version').and_return(True)
zookeeper = flexmock()
zookeeper.should_receive("release_lock").and_return(True)
zookeeper.should_receive("notify_failed_transaction").and_return(True)
dd = DatastoreDistributed(db_batch, zookeeper)
commit_request = datastore_pb.Transaction()
commit_request.set_handle(123)
commit_request.set_app("aaa")
http_request = commit_request.Encode()
self.assertEquals(dd.rollback_transaction("app_id", http_request),
(api_base_pb.VoidProto().Encode(), 0, ""))
def test_rollback_transcation(self):
db_batch = flexmock()
db_batch.should_receive('valid_data_version').and_return(True)
zookeeper = flexmock()
zookeeper.should_receive("release_lock").and_return(True)
zookeeper.should_receive("notify_failed_transaction").and_return(True)
dd = DatastoreDistributed(db_batch, zookeeper)
commit_request = datastore_pb.Transaction()
commit_request.set_handle(123)
commit_request.set_app("aaa")
http_request = commit_request.Encode()
self.assertEquals(dd.rollback_transaction("app_id", http_request),
(api_base_pb.VoidProto().Encode(), 0, ""))
def test_rollback_transcation(self):
db_batch = flexmock()
db_batch.should_receive('valid_data_version_sync').and_return(True)
zk_client = flexmock()
zk_client.should_receive('add_listener')
zookeeper = flexmock(handle=zk_client)
zookeeper.should_receive("release_lock").and_return(True)
zookeeper.should_receive("notify_failed_transaction").and_return(True)
transaction_manager = flexmock()
dd = DatastoreDistributed(db_batch, transaction_manager, zookeeper)
commit_request = datastore_pb.Transaction()
commit_request.set_handle(123)
commit_request.set_app("aaa")
self.assertEquals(
dd.rollback_transaction("app_id", commit_request.handle()), None)
