Skip to content

Commit cf8e417

Browse files
Joanna Grycziennae
Joanna Grycz
authored and
iennae
committed
Use mocked TPUClient
1 parent 27b4479 commit cf8e417

File tree

9 files changed

+251
-228
lines changed

9 files changed

+251
-228
lines changed

‎tpu/createVM.js‎

Lines changed: 20 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -16,19 +16,19 @@
1616

1717
'use strict';
1818

19-
async function main(nodeName, zone, tpuType, tpuSoftwareVersion) {
19+
async function main(tpuClient) {
2020
// [START tpu_vm_create]
21-
// Import the TPU library
22-
const {TpuClient} = require('@google-cloud/tpu').v2;
21+
// Import the TPUClient
22+
// TODO(developer): Uncomment below line before running the sample.
23+
// const {TpuClient} = require('@google-cloud/tpu').v2;
2324
const {Node, NetworkConfig} =
2425
require('@google-cloud/tpu').protos.google.cloud.tpu.v2;
2526

2627
// Instantiate a tpuClient
27-
const tpuClient = new TpuClient();
28+
// TODO(developer): Uncomment below line before running the sample.
29+
// tpuClient = new TpuClient();
2830

29-
/**
30-
* TODO(developer): Update/uncomment these variables before running the sample.
31-
*/
31+
// TODO(developer): Update below line before running the sample.
3232
// Project ID or project number of the Google Cloud project you want to create a node.
3333
const projectId = await tpuClient.getProjectId();
3434

@@ -39,21 +39,21 @@ async function main(nodeName, zone, tpuType, tpuSoftwareVersion) {
3939
const region = 'europe-west4';
4040

4141
// The name for your TPU.
42-
// nodeName = 'node-name-1';
42+
const nodeName = 'node-name-1';
4343

4444
// The zone in which to create the TPU.
4545
// For more information about supported TPU types for specific zones,
4646
// see https://cloud.google.com/tpu/docs/regions-zones
47-
// zone = 'europe-west4-a';
47+
const zone = 'europe-west4-a';
4848

4949
// The accelerator type that specifies the version and size of the Cloud TPU you want to create.
5050
// For more information about supported accelerator types for each TPU version,
5151
// see https://cloud.google.com/tpu/docs/system-architecture-tpu-vm#versions.
52-
// tpuType = 'v2-8';
52+
const tpuType = 'v2-8';
5353

5454
// Software version that specifies the version of the TPU runtime to install. For more information,
5555
// see https://cloud.google.com/tpu/docs/runtimes
56-
// tpuSoftwareVersion = 'tpu-vm-tf-2.14.1';
56+
const tpuSoftwareVersion = 'tpu-vm-tf-2.14.1';
5757

5858
async function callCreateTpuVM() {
5959
// Create a node
@@ -78,14 +78,17 @@ async function main(nodeName, zone, tpuType, tpuSoftwareVersion) {
7878
// Wait for the create operation to complete.
7979
const [response] = await operation.promise();
8080

81-
console.log(JSON.stringify(response));
8281
console.log(`TPU VM: ${nodeName} created.`);
82+
return response;
8383
}
84-
await callCreateTpuVM();
84+
return await callCreateTpuVM();
8585
// [END tpu_vm_create]
8686
}
8787

88-
main(...process.argv.slice(2)).catch(err => {
89-
console.error(err);
90-
process.exitCode = 1;
91-
});
88+
module.exports = main;
89+
90+
// TODO(developer): Uncomment below lines before running the sample.
91+
// main(...process.argv.slice(2)).catch(err => {
92+
// console.error(err);
93+
// process.exitCode = 1;
94+
// });

‎tpu/deleteVM.js‎

Lines changed: 19 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -16,25 +16,25 @@
1616

1717
'use strict';
1818

19-
async function main(nodeName, zone) {
19+
async function main(tpuClient) {
2020
// [START tpu_vm_delete]
21-
// Import TpuClient
22-
const {TpuClient} = require('@google-cloud/tpu').v2;
21+
// Import the TPUClient
22+
// TODO(developer): Uncomment below line before running the sample.
23+
// const {TpuClient} = require('@google-cloud/tpu').v2;
2324

2425
// Instantiate a tpuClient
25-
const tpuClient = new TpuClient();
26+
// TODO(developer): Uncomment below line before running the sample.
27+
// tpuClient = new TpuClient();
2628

27-
/**
28-
* TODO(developer): Update/uncomment these variables before running the sample.
29-
*/
29+
// TODO(developer): Update these variables before running the sample.
3030
// Project ID or project number of the Google Cloud project you want to delete a node.
3131
const projectId = await tpuClient.getProjectId();
3232

3333
// The name of TPU to delete.
34-
// nodeName = 'node-name-1';
34+
const nodeName = 'node-name-1';
3535

3636
// The zone, where the TPU is created.
37-
// zone = 'europe-west4-a';
37+
const zone = 'europe-west4-a';
3838

3939
async function callDeleteTpuVM() {
4040
const request = {
@@ -44,16 +44,20 @@ async function main(nodeName, zone) {
4444
const [operation] = await tpuClient.deleteNode(request);
4545

4646
// Wait for the delete operation to complete.
47-
await operation.promise();
47+
const [response] = await operation.promise();
4848

4949
console.log(`Node: ${nodeName} deleted.`);
50+
return response;
5051
}
5152

52-
await callDeleteTpuVM();
53+
return await callDeleteTpuVM();
5354
// [END tpu_vm_delete]
5455
}
5556

56-
main(...process.argv.slice(2)).catch(err => {
57-
console.error(err);
58-
process.exitCode = 1;
59-
});
57+
module.exports = main;
58+
59+
// TODO(developer): Uncomment below lines before running the sample.
60+
// main(...process.argv.slice(2)).catch(err => {
61+
// console.error(err);
62+
// process.exitCode = 1;
63+
// });

‎tpu/getVM.js‎

Lines changed: 18 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -16,25 +16,25 @@
1616

1717
'use strict';
1818

19-
async function main(nodeName, zone) {
19+
async function main(tpuClient) {
2020
// [START tpu_vm_get]
21-
// Import TpuClient
22-
const {TpuClient} = require('@google-cloud/tpu').v2;
21+
// Import the TPUClient
22+
// TODO(developer): Uncomment below line before running the sample.
23+
// const {TpuClient} = require('@google-cloud/tpu').v2;
2324

2425
// Instantiate a tpuClient
25-
const tpuClient = new TpuClient();
26+
// TODO(developer): Uncomment below line before running the sample.
27+
// tpuClient = new TpuClient();
2628

27-
/**
28-
* TODO(developer): Update/uncomment these variables before running the sample.
29-
*/
29+
// TODO(developer): Update these variables before running the sample.
3030
// Project ID or project number of the Google Cloud project you want to retrive a node.
3131
const projectId = await tpuClient.getProjectId();
3232

3333
// The name of TPU to retrive.
34-
// nodeName = 'node-name-1';
34+
const nodeName = 'node-name-1';
3535

3636
// The zone, where the TPU is created.
37-
// zone = 'europe-west4-a';
37+
const zone = 'europe-west4-a';
3838

3939
async function callGetTpuVM() {
4040
const request = {
@@ -44,14 +44,17 @@ async function main(nodeName, zone) {
4444
const [response] = await tpuClient.getNode(request);
4545

4646
console.log(`Node: ${nodeName} retrived.`);
47-
console.log(JSON.stringify(response));
47+
return response;
4848
}
4949

50-
await callGetTpuVM();
50+
return await callGetTpuVM();
5151
// [END tpu_vm_get]
5252
}
5353

54-
main(...process.argv.slice(2)).catch(err => {
55-
console.error(err);
56-
process.exitCode = 1;
57-
});
54+
module.exports = main;
55+
56+
// TODO(developer): Uncomment below lines before running the sample.
57+
// main(...process.argv.slice(2)).catch(err => {
58+
// console.error(err);
59+
// process.exitCode = 1;
60+
// });

‎tpu/getVMList.js‎

Lines changed: 17 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -16,22 +16,22 @@
1616

1717
'use strict';
1818

19-
async function main(zone) {
19+
async function main(tpuClient) {
2020
// [START tpu_vm_list]
21-
// Import TpuClient
22-
const {TpuClient} = require('@google-cloud/tpu').v2;
21+
// Import the TPUClient
22+
// TODO(developer): Uncomment below line before running the sample.
23+
// const {TpuClient} = require('@google-cloud/tpu').v2;
2324

2425
// Instantiate a tpuClient
25-
const tpuClient = new TpuClient();
26+
// TODO(developer): Uncomment below line before running the sample.
27+
// tpuClient = new TpuClient();
2628

27-
/**
28-
* TODO(developer): Update/uncomment these variables before running the sample.
29-
*/
29+
// TODO(developer): Update these variables before running the sample.
3030
// Project ID or project number of the Google Cloud project you want to retrive a list of TPU nodes.
3131
const projectId = await tpuClient.getProjectId();
3232

3333
// The zone from which the TPUs are retrived.
34-
// zone = 'europe-west4-a';
34+
const zone = 'europe-west4-a';
3535

3636
async function callTpuVMList() {
3737
const request = {
@@ -40,14 +40,17 @@ async function main(zone) {
4040

4141
const [response] = await tpuClient.listNodes(request);
4242

43-
console.log(JSON.stringify(response));
43+
return response;
4444
}
4545

46-
await callTpuVMList();
46+
return await callTpuVMList();
4747
// [END tpu_vm_list]
4848
}
4949

50-
main(...process.argv.slice(2)).catch(err => {
51-
console.error(err);
52-
process.exitCode = 1;
53-
});
50+
module.exports = main;
51+
52+
// TODO(developer): Uncomment below lines before running the sample.
53+
// main(...process.argv.slice(2)).catch(err => {
54+
// console.error(err);
55+
// process.exitCode = 1;
56+
// });

‎tpu/package.json‎

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,8 @@
1414
"test": "c8 mocha -p -j 2 test --timeout 1200000"
1515
},
1616
"dependencies": {
17-
"@google-cloud/tpu": "^3.5.0"
17+
"@google-cloud/tpu": "^3.5.0",
18+
"sinon": "^19.0.2"
1819
},
1920
"devDependencies": {
2021
"c8": "^10.0.0",

‎tpu/startVM.js‎

Lines changed: 20 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -16,43 +16,48 @@
1616

1717
'use strict';
1818

19-
async function main(nodeName, zone) {
19+
async function main(tpuClient) {
2020
// [START tpu_vm_start]
21-
// Import TpuClient
22-
const {TpuClient} = require('@google-cloud/tpu').v2;
21+
// Import the TPUClient
22+
// TODO(developer): Uncomment below line before running the sample.
23+
// const {TpuClient} = require('@google-cloud/tpu').v2;
2324

2425
// Instantiate a tpuClient
25-
const tpuClient = new TpuClient();
26+
// TODO(developer): Uncomment below line before running the sample.
27+
// tpuClient = new TpuClient();
2628

27-
/**
28-
* TODO(developer): Update/uncomment these variables before running the sample.
29-
*/
29+
// TODO(developer): Update these variables before running the sample.
3030
// Project ID or project number of the Google Cloud project you want to start a node.
3131
const projectId = await tpuClient.getProjectId();
3232

3333
// The name of TPU to start.
34-
// nodeName = 'node-name-1';
34+
const nodeName = 'node-name-1';
3535

3636
// The zone, where the TPU is created.
37-
// zone = 'europe-west4-a';
37+
const zone = 'europe-west4-a';
3838

3939
async function callStartTpuVM() {
4040
const request = {
4141
name: `projects/${projectId}/locations/${zone}/nodes/${nodeName}`,
4242
};
4343

4444
const [operation] = await tpuClient.startNode(request);
45+
4546
// Wait for the operation to complete.
46-
await operation.promise();
47+
const [response] = await operation.promise();
4748

4849
console.log(`Node: ${nodeName} started.`);
50+
return response;
4951
}
5052

51-
await callStartTpuVM();
53+
return await callStartTpuVM();
5254
// [END tpu_vm_start]
5355
}
5456

55-
main(...process.argv.slice(2)).catch(err => {
56-
console.error(err);
57-
process.exitCode = 1;
58-
});
57+
module.exports = main;
58+
59+
// TODO(developer): Uncomment below lines before running the sample.
60+
// main(...process.argv.slice(2)).catch(err => {
61+
// console.error(err);
62+
// process.exitCode = 1;
63+
// });

‎tpu/stopVM.js‎

Lines changed: 19 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -16,25 +16,25 @@
1616

1717
'use strict';
1818

19-
async function main(nodeName, zone) {
19+
async function main(tpuClient) {
2020
// [START tpu_vm_stop]
21-
// Import TpuClient
22-
const {TpuClient} = require('@google-cloud/tpu').v2;
21+
// Import the TPUClient
22+
// TODO(developer): Uncomment below line before running the sample.
23+
// const {TpuClient} = require('@google-cloud/tpu').v2;
2324

2425
// Instantiate a tpuClient
25-
const tpuClient = new TpuClient();
26+
// TODO(developer): Uncomment below line before running the sample.
27+
// tpuClient = new TpuClient();
2628

27-
/**
28-
* TODO(developer): Update/uncomment these variables before running the sample.
29-
*/
29+
// TODO(developer): Update these variables before running the sample.
3030
// Project ID or project number of the Google Cloud project you want to stop a node.
3131
const projectId = await tpuClient.getProjectId();
3232

3333
// The name of TPU to stop.
34-
// nodeName = 'node-name-1';
34+
const nodeName = 'node-name-1';
3535

3636
// The zone, where the TPU is created.
37-
// zone = 'europe-west4-a';
37+
const zone = 'europe-west4-a';
3838

3939
async function callStopTpuVM() {
4040
const request = {
@@ -43,16 +43,20 @@ async function main(nodeName, zone) {
4343

4444
const [operation] = await tpuClient.stopNode(request);
4545
// Wait for the operation to complete.
46-
await operation.promise();
46+
const [response] = await operation.promise();
4747

4848
console.log(`Node: ${nodeName} stopped.`);
49+
return response;
4950
}
5051

51-
await callStopTpuVM();
52+
return await callStopTpuVM();
5253
// [END tpu_vm_stop]
5354
}
5455

55-
main(...process.argv.slice(2)).catch(err => {
56-
console.error(err);
57-
process.exitCode = 1;
58-
});
56+
module.exports = main;
57+
58+
// TODO(developer): Uncomment below lines before running the sample.
59+
// main(...process.argv.slice(2)).catch(err => {
60+
// console.error(err);
61+
// process.exitCode = 1;
62+
// });

0 commit comments

Comments
 (0)