From d6cb9b2a41422df3cc86209ae1a2642ca8b7b2c2 Mon Sep 17 00:00:00 2001
From: Jonathan Lui <jonathanlui@google.com>
Date: Mon, 3 Sep 2018 21:12:19 -0700
Subject: [PATCH] fix quickstart.test.js using new export structure (#348)

---
 samples/system-test/quickstart.test.js | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/samples/system-test/quickstart.test.js b/samples/system-test/quickstart.test.js
index a478957b7..88448984a 100644
--- 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,
+    },
   });
 });