Skip to content

Commit 3213175

Browse files
swcloudswcloud
authored
Dlp (#2347)
* Fix the long running operation method * fix lint error
1 parent 9c2a6ad commit 3213175

1 file changed

Lines changed: 341 additions & 0 deletions

File tree

‎packages/dlp/test/gapic_v2beta1.js‎

Lines changed: 341 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,341 @@
1+
/*
2+
* Copyright 2017, Google Inc. All rights reserved.
3+
*
4+
* Licensed under the Apache License, Version 2.0 (the "License");
5+
* you may not use this file except in compliance with the License.
6+
* You may obtain a copy of the License at
7+
*
8+
* http://www.apache.org/licenses/LICENSE-2.0
9+
*
10+
* Unless required by applicable law or agreed to in writing, software
11+
* distributed under the License is distributed on an "AS IS" BASIS,
12+
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13+
* See the License for the specific language governing permissions and
14+
* limitations under the License.
15+
*/
16+
'use strict';
17+
18+
var assert = require('assert');
19+
var dlpV2beta1 = require('../src/v2beta1')();
20+
21+
var FAKE_STATUS_CODE = 1;
22+
var error = new Error();
23+
error.code = FAKE_STATUS_CODE;
24+
describe('DlpServiceClient', function() {
25+
describe('inspectContent', function() {
26+
it('invokes inspectContent without error', function(done) {
27+
var client = dlpV2beta1.dlpServiceClient();
28+
// Mock request
29+
var inspectConfig = {};
30+
var items = [];
31+
var request = {
32+
inspectConfig : inspectConfig,
33+
items : items
34+
};
35+
36+
// Mock response
37+
var expectedResponse = {};
38+
39+
// Mock Grpc layer
40+
client._inspectContent = mockSimpleGrpcMethod(request, expectedResponse);
41+
42+
client.inspectContent(request, function(err, response) {
43+
assert.ifError(err);
44+
assert.deepStrictEqual(response, expectedResponse);
45+
done();
46+
});
47+
});
48+
49+
it('invokes inspectContent with error', function(done) {
50+
var client = dlpV2beta1.dlpServiceClient();
51+
// Mock request
52+
var inspectConfig = {};
53+
var items = [];
54+
var request = {
55+
inspectConfig : inspectConfig,
56+
items : items
57+
};
58+
59+
// Mock Grpc layer
60+
client._inspectContent = mockSimpleGrpcMethod(request, null, error);
61+
62+
client.inspectContent(request, function(err, response) {
63+
assert(err instanceof Error);
64+
assert.equal(err.code, FAKE_STATUS_CODE);
65+
done();
66+
});
67+
});
68+
});
69+
70+
describe('redactContent', function() {
71+
it('invokes redactContent without error', function(done) {
72+
var client = dlpV2beta1.dlpServiceClient();
73+
// Mock request
74+
var inspectConfig = {};
75+
var items = [];
76+
var replaceConfigs = [];
77+
var request = {
78+
inspectConfig : inspectConfig,
79+
items : items,
80+
replaceConfigs : replaceConfigs
81+
};
82+
83+
// Mock response
84+
var expectedResponse = {};
85+
86+
// Mock Grpc layer
87+
client._redactContent = mockSimpleGrpcMethod(request, expectedResponse);
88+
89+
client.redactContent(request, function(err, response) {
90+
assert.ifError(err);
91+
assert.deepStrictEqual(response, expectedResponse);
92+
done();
93+
});
94+
});
95+
96+
it('invokes redactContent with error', function(done) {
97+
var client = dlpV2beta1.dlpServiceClient();
98+
// Mock request
99+
var inspectConfig = {};
100+
var items = [];
101+
var replaceConfigs = [];
102+
var request = {
103+
inspectConfig : inspectConfig,
104+
items : items,
105+
replaceConfigs : replaceConfigs
106+
};
107+
108+
// Mock Grpc layer
109+
client._redactContent = mockSimpleGrpcMethod(request, null, error);
110+
111+
client.redactContent(request, function(err, response) {
112+
assert(err instanceof Error);
113+
assert.equal(err.code, FAKE_STATUS_CODE);
114+
done();
115+
});
116+
});
117+
});
118+
119+
describe('createInspectOperation', function() {
120+
it('invokes createInspectOperation without error', function(done) {
121+
var client = dlpV2beta1.dlpServiceClient();
122+
// Mock request
123+
var inspectConfig = {};
124+
var storageConfig = {};
125+
var outputConfig = {};
126+
var request = {
127+
inspectConfig : inspectConfig,
128+
storageConfig : storageConfig,
129+
outputConfig : outputConfig
130+
};
131+
132+
// Mock response
133+
var name = 'name3373707';
134+
var expectedResponse = {
135+
name : name
136+
};
137+
138+
// Mock Grpc layer
139+
client._createInspectOperation = mockLongRunningGrpcMethod(request, expectedResponse);
140+
141+
client.createInspectOperation(request).then(function(responses) {
142+
var operation = responses[0];
143+
return operation.promise();
144+
}).then(function(responses) {
145+
assert.deepStrictEqual(responses[0], expectedResponse);
146+
done();
147+
}).catch(function(err) {
148+
done(err);
149+
});
150+
});
151+
152+
it('invokes createInspectOperation with error', function(done) {
153+
var client = dlpV2beta1.dlpServiceClient();
154+
// Mock request
155+
var inspectConfig = {};
156+
var storageConfig = {};
157+
var outputConfig = {};
158+
var request = {
159+
inspectConfig : inspectConfig,
160+
storageConfig : storageConfig,
161+
outputConfig : outputConfig
162+
};
163+
164+
// Mock Grpc layer
165+
client._createInspectOperation = mockLongRunningGrpcMethod(request, null, error);
166+
167+
client.createInspectOperation(request).then(function(responses) {
168+
var operation = responses[0];
169+
return operation.promise();
170+
}).then(function(responses) {
171+
assert.fail();
172+
}).catch(function(err) {
173+
assert(err instanceof Error);
174+
assert.equal(err.code, FAKE_STATUS_CODE);
175+
done();
176+
});
177+
});
178+
});
179+
180+
describe('listInspectFindings', function() {
181+
it('invokes listInspectFindings without error', function(done) {
182+
var client = dlpV2beta1.dlpServiceClient();
183+
// Mock request
184+
var formattedName = client.resultPath("[RESULT]");
185+
var request = {
186+
name : formattedName
187+
};
188+
189+
// Mock response
190+
var nextPageToken = 'nextPageToken-1530815211';
191+
var expectedResponse = {
192+
nextPageToken : nextPageToken
193+
};
194+
195+
// Mock Grpc layer
196+
client._listInspectFindings = mockSimpleGrpcMethod(request, expectedResponse);
197+
198+
client.listInspectFindings(request, function(err, response) {
199+
assert.ifError(err);
200+
assert.deepStrictEqual(response, expectedResponse);
201+
done();
202+
});
203+
});
204+
205+
it('invokes listInspectFindings with error', function(done) {
206+
var client = dlpV2beta1.dlpServiceClient();
207+
// Mock request
208+
var formattedName = client.resultPath("[RESULT]");
209+
var request = {
210+
name : formattedName
211+
};
212+
213+
// Mock Grpc layer
214+
client._listInspectFindings = mockSimpleGrpcMethod(request, null, error);
215+
216+
client.listInspectFindings(request, function(err, response) {
217+
assert(err instanceof Error);
218+
assert.equal(err.code, FAKE_STATUS_CODE);
219+
done();
220+
});
221+
});
222+
});
223+
224+
describe('listInfoTypes', function() {
225+
it('invokes listInfoTypes without error', function(done) {
226+
var client = dlpV2beta1.dlpServiceClient();
227+
// Mock request
228+
var category = 'category50511102';
229+
var languageCode = 'languageCode-412800396';
230+
var request = {
231+
category : category,
232+
languageCode : languageCode
233+
};
234+
235+
// Mock response
236+
var expectedResponse = {};
237+
238+
// Mock Grpc layer
239+
client._listInfoTypes = mockSimpleGrpcMethod(request, expectedResponse);
240+
241+
client.listInfoTypes(request, function(err, response) {
242+
assert.ifError(err);
243+
assert.deepStrictEqual(response, expectedResponse);
244+
done();
245+
});
246+
});
247+
248+
it('invokes listInfoTypes with error', function(done) {
249+
var client = dlpV2beta1.dlpServiceClient();
250+
// Mock request
251+
var category = 'category50511102';
252+
var languageCode = 'languageCode-412800396';
253+
var request = {
254+
category : category,
255+
languageCode : languageCode
256+
};
257+
258+
// Mock Grpc layer
259+
client._listInfoTypes = mockSimpleGrpcMethod(request, null, error);
260+
261+
client.listInfoTypes(request, function(err, response) {
262+
assert(err instanceof Error);
263+
assert.equal(err.code, FAKE_STATUS_CODE);
264+
done();
265+
});
266+
});
267+
});
268+
269+
describe('listRootCategories', function() {
270+
it('invokes listRootCategories without error', function(done) {
271+
var client = dlpV2beta1.dlpServiceClient();
272+
// Mock request
273+
var languageCode = 'languageCode-412800396';
274+
var request = {
275+
languageCode : languageCode
276+
};
277+
278+
// Mock response
279+
var expectedResponse = {};
280+
281+
// Mock Grpc layer
282+
client._listRootCategories = mockSimpleGrpcMethod(request, expectedResponse);
283+
284+
client.listRootCategories(request, function(err, response) {
285+
assert.ifError(err);
286+
assert.deepStrictEqual(response, expectedResponse);
287+
done();
288+
});
289+
});
290+
291+
it('invokes listRootCategories with error', function(done) {
292+
var client = dlpV2beta1.dlpServiceClient();
293+
// Mock request
294+
var languageCode = 'languageCode-412800396';
295+
var request = {
296+
languageCode : languageCode
297+
};
298+
299+
// Mock Grpc layer
300+
client._listRootCategories = mockSimpleGrpcMethod(request, null, error);
301+
302+
client.listRootCategories(request, function(err, response) {
303+
assert(err instanceof Error);
304+
assert.equal(err.code, FAKE_STATUS_CODE);
305+
done();
306+
});
307+
});
308+
});
309+
310+
});
311+
312+
function mockSimpleGrpcMethod(expectedRequest, response, error) {
313+
return function(actualRequest, options, callback) {
314+
assert.deepStrictEqual(actualRequest, expectedRequest);
315+
if (error) {
316+
callback(error);
317+
} else if (response) {
318+
callback(null, response);
319+
} else {
320+
callback(null);
321+
}
322+
};
323+
}
324+
325+
function mockLongRunningGrpcMethod(expectedRequest, response, error) {
326+
return function(request) {
327+
assert.deepStrictEqual(request, expectedRequest);
328+
var mockOperation = {
329+
promise: function() {
330+
return new Promise(function(resolve, reject) {
331+
if (error) {
332+
reject(error);
333+
} else {
334+
resolve([response]);
335+
}
336+
});
337+
}
338+
};
339+
return Promise.resolve([mockOperation]);
340+
};
341+
}

0 commit comments

Comments
 (0)