Updated interop tests to handle proto3 changes

pull/1561/head
murgatroid99 10 years ago
parent c73a2be603
commit 20afbb9140
  1. 32
      src/node/interop/interop_client.js
  2. 23
      src/node/interop/interop_server.js

@ -86,7 +86,7 @@ function emptyUnary(client, done) {
*/ */
function largeUnary(client, done) { function largeUnary(client, done) {
var arg = { var arg = {
response_type: testProto.PayloadType.COMPRESSABLE, response_type: 'COMPRESSABLE',
response_size: 314159, response_size: 314159,
payload: { payload: {
body: zeroBuffer(271828) body: zeroBuffer(271828)
@ -94,9 +94,8 @@ function largeUnary(client, done) {
}; };
var call = client.unaryCall(arg, function(err, resp) { var call = client.unaryCall(arg, function(err, resp) {
assert.ifError(err); assert.ifError(err);
assert.strictEqual(resp.payload.type, testProto.PayloadType.COMPRESSABLE); assert.strictEqual(resp.payload.type, 'COMPRESSABLE');
assert.strictEqual(resp.payload.body.limit - resp.payload.body.offset, assert.strictEqual(resp.payload.body.length, 314159);
314159);
}); });
call.on('status', function(status) { call.on('status', function(status) {
assert.strictEqual(status.code, grpc.status.OK); assert.strictEqual(status.code, grpc.status.OK);
@ -138,7 +137,7 @@ function clientStreaming(client, done) {
*/ */
function serverStreaming(client, done) { function serverStreaming(client, done) {
var arg = { var arg = {
response_type: testProto.PayloadType.COMPRESSABLE, response_type: 'COMPRESSABLE',
response_parameters: [ response_parameters: [
{size: 31415}, {size: 31415},
{size: 9}, {size: 9},
@ -150,8 +149,8 @@ function serverStreaming(client, done) {
var resp_index = 0; var resp_index = 0;
call.on('data', function(value) { call.on('data', function(value) {
assert(resp_index < 4); assert(resp_index < 4);
assert.strictEqual(value.payload.type, testProto.PayloadType.COMPRESSABLE); assert.strictEqual(value.payload.type, 'COMPRESSABLE');
assert.strictEqual(value.payload.body.limit - value.payload.body.offset, assert.strictEqual(value.payload.body.length,
arg.response_parameters[resp_index].size); arg.response_parameters[resp_index].size);
resp_index += 1; resp_index += 1;
}); });
@ -182,23 +181,21 @@ function pingPong(client, done) {
}); });
var index = 0; var index = 0;
call.write({ call.write({
response_type: testProto.PayloadType.COMPRESSABLE, response_type: 'COMPRESSABLE',
response_parameters: [ response_parameters: [
{size: response_sizes[index]} {size: response_sizes[index]}
], ],
payload: {body: zeroBuffer(payload_sizes[index])} payload: {body: zeroBuffer(payload_sizes[index])}
}); });
call.on('data', function(response) { call.on('data', function(response) {
assert.strictEqual(response.payload.type, assert.strictEqual(response.payload.type, 'COMPRESSABLE');
testProto.PayloadType.COMPRESSABLE); assert.equal(response.payload.body.length, response_sizes[index]);
assert.equal(response.payload.body.limit - response.payload.body.offset,
response_sizes[index]);
index += 1; index += 1;
if (index === 4) { if (index === 4) {
call.end(); call.end();
} else { } else {
call.write({ call.write({
response_type: testProto.PayloadType.COMPRESSABLE, response_type: 'COMPRESSABLE',
response_parameters: [ response_parameters: [
{size: response_sizes[index]} {size: response_sizes[index]}
], ],
@ -251,7 +248,7 @@ function cancelAfterBegin(client, done) {
function cancelAfterFirstResponse(client, done) { function cancelAfterFirstResponse(client, done) {
var call = client.fullDuplexCall(); var call = client.fullDuplexCall();
call.write({ call.write({
response_type: testProto.PayloadType.COMPRESSABLE, response_type: 'COMPRESSABLE',
response_parameters: [ response_parameters: [
{size: 31415} {size: 31415}
], ],
@ -281,7 +278,7 @@ function authTest(expected_user, client, done) {
} }
client.updateMetadata = grpc.getGoogleAuthDelegate(credential); client.updateMetadata = grpc.getGoogleAuthDelegate(credential);
var arg = { var arg = {
response_type: testProto.PayloadType.COMPRESSABLE, response_type: 'COMPRESSABLE',
response_size: 314159, response_size: 314159,
payload: { payload: {
body: zeroBuffer(271828) body: zeroBuffer(271828)
@ -291,9 +288,8 @@ function authTest(expected_user, client, done) {
}; };
var call = client.unaryCall(arg, function(err, resp) { var call = client.unaryCall(arg, function(err, resp) {
assert.ifError(err); assert.ifError(err);
assert.strictEqual(resp.payload.type, testProto.PayloadType.COMPRESSABLE); assert.strictEqual(resp.payload.type, 'COMPRESSABLE');
assert.strictEqual(resp.payload.body.limit - resp.payload.body.offset, assert.strictEqual(resp.payload.body.length, 314159);
314159);
assert.strictEqual(resp.username, expected_user); assert.strictEqual(resp.username, expected_user);
assert.strictEqual(resp.oauth_scope, AUTH_SCOPE_RESPONSE); assert.strictEqual(resp.oauth_scope, AUTH_SCOPE_RESPONSE);
}); });

@ -72,10 +72,9 @@ function handleUnary(call, callback) {
var req = call.request; var req = call.request;
var zeros = zeroBuffer(req.response_size); var zeros = zeroBuffer(req.response_size);
var payload_type = req.response_type; var payload_type = req.response_type;
if (payload_type === testProto.PayloadType.RANDOM) { if (payload_type === 'RANDOM') {
payload_type = [ payload_type = ['COMPRESSABLE',
testProto.PayloadType.COMPRESSABLE, 'UNCOMPRESSABLE'][Math.random() < 0.5 ? 0 : 1];
testProto.PayloadType.UNCOMPRESSABLE][Math.random() < 0.5 ? 0 : 1];
} }
callback(null, {payload: {type: payload_type, body: zeros}}); callback(null, {payload: {type: payload_type, body: zeros}});
} }
@ -89,7 +88,7 @@ function handleUnary(call, callback) {
function handleStreamingInput(call, callback) { function handleStreamingInput(call, callback) {
var aggregate_size = 0; var aggregate_size = 0;
call.on('data', function(value) { call.on('data', function(value) {
aggregate_size += value.payload.body.limit - value.payload.body.offset; aggregate_size += value.payload.body.length;
}); });
call.on('end', function() { call.on('end', function() {
callback(null, {aggregated_payload_size: aggregate_size}); callback(null, {aggregated_payload_size: aggregate_size});
@ -103,10 +102,9 @@ function handleStreamingInput(call, callback) {
function handleStreamingOutput(call) { function handleStreamingOutput(call) {
var req = call.request; var req = call.request;
var payload_type = req.response_type; var payload_type = req.response_type;
if (payload_type === testProto.PayloadType.RANDOM) { if (payload_type === 'RANDOM') {
payload_type = [ payload_type = ['COMPRESSABLE',
testProto.PayloadType.COMPRESSABLE, 'UNCOMPRESSABLE'][Math.random() < 0.5 ? 0 : 1];
testProto.PayloadType.UNCOMPRESSABLE][Math.random() < 0.5 ? 0 : 1];
} }
_.each(req.response_parameters, function(resp_param) { _.each(req.response_parameters, function(resp_param) {
call.write({ call.write({
@ -127,10 +125,9 @@ function handleStreamingOutput(call) {
function handleFullDuplex(call) { function handleFullDuplex(call) {
call.on('data', function(value) { call.on('data', function(value) {
var payload_type = value.response_type; var payload_type = value.response_type;
if (payload_type === testProto.PayloadType.RANDOM) { if (payload_type === 'RANDOM') {
payload_type = [ payload_type = ['COMPRESSABLE',
testProto.PayloadType.COMPRESSABLE, 'UNCOMPRESSABLE'][Math.random() < 0.5 ? 0 : 1];
testProto.PayloadType.UNCOMPRESSABLE][Math.random() < 0.5 ? 0 : 1];
} }
_.each(value.response_parameters, function(resp_param) { _.each(value.response_parameters, function(resp_param) {
call.write({ call.write({

Loading…
Cancel
Save