fix quickstart.test.js using new export structure (#348)

googleapis · Sep 4, 2018 · d6cb9b2 · d6cb9b2
1 parent f323ab3
commit d6cb9b2
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/samples/system-test/quickstart.test.js b/samples/system-test/quickstart.test.js
@@ -21,7 +21,6 @@ const test = require(`ava`);
 const tools = require(`@google-cloud/nodejs-repo-tools`);
 const uuid = require(`uuid`);
 
-
 const {Storage} = proxyquire(`@google-cloud/storage`, {});
 
 const storage = new Storage();
@@ -39,8 +38,8 @@ test.after.always(async () => {
 test.cb(`should create a bucket`, t => {
   const expectedBucketName = `my-new-bucket`;
 
-  const storageMock = {
-    createBucket: _bucketName => {
+  const StorageMock = class {
+    createBucket(_bucketName) {
       t.is(_bucketName, expectedBucketName);
 
       return bucket.create().then(([bucket]) => {
@@ -61,10 +60,12 @@ test.cb(`should create a bucket`, t => {
 
         return [bucket];
       });
-    },
+    }
   };
 
   proxyquire(`../quickstart`, {
-    '@google-cloud/storage': sinon.stub().returns(storageMock),
+    '@google-cloud/storage': {
+      Storage: StorageMock,
+    },
   });
 });