test: added unit tests

googleapis · May 6, 2021 · bef8232 · bef8232
1 parent 0886c75
commit bef8232
Show file tree

Hide file tree

Showing 3 changed files with 53 additions and 0 deletions.
diff --git a/test/index.ts b/test/index.ts
@@ -557,6 +557,7 @@ describe('Spanner', () => {
             name: PATH,
             displayName: NAME,
             nodeCount: 1,
+            processingUnits: undefined,
             config: `projects/project-id/instanceConfigs/${CONFIG.config}`,
           },
         });

diff --git a/test/mockserver/mockinstanceadmin.ts b/test/mockserver/mockinstanceadmin.ts
@@ -209,6 +209,9 @@ export class MockInstanceAdmin {
       nodeCount: call.request!.instance
         ? call.request!.instance.nodeCount
         : undefined,
+      processingUnits: call.request!.instance
+        ? call.request!.instance.processingUnits
+        : undefined,
       labels: call.request!.instance
         ? call.request!.instance.labels
         : undefined,

diff --git a/test/spanner.ts b/test/spanner.ts
@@ -2710,6 +2710,55 @@ describe('Spanner with mock server', () => {
       );
     });
 
+    it('should create an instance with processing units', async () => {
+      const [createdInstance] = await spanner
+        .createInstance('new-instance', {
+          config: 'test-instance-config',
+          processingUnits: 500,
+        })
+        .then(data => {
+          const operation = data[1];
+          return operation.promise() as Promise<
+            [Instance, CreateInstanceMetadata, object]
+          >;
+        })
+        .then(response => {
+          return response;
+        });
+      assert.strictEqual(
+        createdInstance.name,
+        `projects/${spanner.projectId}/instances/new-instance`
+      );
+      assert.strictEqual(createdInstance.processingUnits, 500);
+      assert.strictEqual(createdInstance.nodeCount, 0);
+    });
+
+    it('should fail if both nodes and processingUnits are given', async () => {
+      try {
+        await spanner
+          .createInstance('new-instance', {
+            config: 'test-instance-config',
+            nodes: 1,
+            processingUnits: 500,
+          })
+          .then(data => {
+            const operation = data[1];
+            return operation.promise() as Promise<
+              [Instance, CreateInstanceMetadata, object]
+            >;
+          })
+          .then(response => {
+            return response;
+          });
+        assert.fail('missing expected error');
+      } catch (err) {
+        assert.strictEqual(
+          err.message,
+          'Only one of nodeCount or processingUnits can be specified.'
+        );
+      }
+    });
+
     it('should update an instance', async () => {
       const instance = spanner.instance(mockInstanceAdmin.PROD_INSTANCE_NAME);
       const [updatedInstance] = await instance