def testAllocateIds(self):
request = datastore.AllocateIdsRequest()
payload = request.SerializeToString()
response = datastore.AllocateIdsResponse()
self.expectRequest(
'https://datastore.com/datastore/v1beta1/datasets/foo/allocateIds',
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.allocate_ids(request)
self.assertEqual(response, resp)
self.mox.VerifyAll()
def testAllocateIds(self):
request = datastore.AllocateIdsRequest()
payload = request.SerializeToString()
response = datastore.AllocateIdsResponse()
self.expectRequest(
'https://example.com/datastore/v1/projects/foo:allocateIds',
method='POST',
body=payload,
headers=self.makeExpectedHeaders(payload)).AndReturn(
(TestResponse(status=200,
reason='Found'), response.SerializeToString()))
self.mox.ReplayAll()
resp = self.conn.allocate_ids(request)
self.assertEqual(response, resp)
self.mox.VerifyAll()
def testAllocateIds(self):
request = datastore.AllocateIdsRequest()
payload = request.SerializeToString()
proto_response = datastore.AllocateIdsResponse()
response = httplib2.Response({
'status': 200,
'content-type': 'application/x-protobuf',
})
self.expectRequest(
'https://example.com/datastore/v1/projects/foo:allocateIds',
method='POST',
body=payload,
headers=self.makeExpectedHeaders(payload)).AndReturn(
(response, proto_response.SerializeToString()))
self.mox.ReplayAll()
resp = self.conn.allocate_ids(request)
self.assertEqual(proto_response, resp)
self.mox.VerifyAll()