Skip to content
Snippets Groups Projects
Commit 20afbb91 authored by murgatroid99's avatar murgatroid99
Browse files

Updated interop tests to handle proto3 changes

parent c73a2be6
No related branches found
No related tags found
No related merge requests found
...@@ -86,7 +86,7 @@ function emptyUnary(client, done) { ...@@ -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) { ...@@ -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) { ...@@ -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) { ...@@ -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) { ...@@ -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) { ...@@ -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) { ...@@ -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) { ...@@ -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) { ...@@ -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) { ...@@ -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) { ...@@ -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) { ...@@ -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({
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment