def testBeginTransaction(self): request = datastore.BeginTransactionRequest() payload = request.SerializeToString() response = datastore.BeginTransactionResponse() self.expectRequest( 'https://example.com/datastore/v1/projects/foo:beginTransaction', method='POST', body=payload, headers=self.makeExpectedHeaders(payload)).AndReturn( (TestResponse(status=200, reason='Found'), response.SerializeToString())) self.mox.ReplayAll() resp = self.conn.begin_transaction(request) self.assertEqual(response, resp) self.mox.VerifyAll()
def testBeginTransaction(self): request = datastore.BeginTransactionRequest() payload = request.SerializeToString() response = datastore.BeginTransactionResponse() self.expectRequest( 'https://datastore.com/datastore/v1beta1/datasets/foo/' 'beginTransaction', method='POST', body=payload, headers={'Content-Type': 'application/x-protobuf', 'Content-Length': str(len(payload))}).AndReturn( (TestResponse(status=200, reason='Found'), response.SerializeToString())) self.mox.ReplayAll() resp = self.conn.begin_transaction(request) self.assertEqual(response, resp) self.mox.VerifyAll()
def testBeginTransaction(self): request = datastore.BeginTransactionRequest() payload = request.SerializeToString() proto_response = datastore.BeginTransactionResponse() response = httplib2.Response({ 'status': 200, 'content-type': 'application/x-protobuf', }) self.expectRequest( 'https://example.com/datastore/v1/projects/foo:beginTransaction', method='POST', body=payload, headers=self.makeExpectedHeaders(payload)).AndReturn( (response, proto_response.SerializeToString())) self.mox.ReplayAll() resp = self.conn.begin_transaction(request) self.assertEqual(proto_response, resp) self.mox.VerifyAll()