samples: add contents variable to streamFileUpload sample (#2068)

ddelgrosso1 · web-flow · commit 48e4bc9f077e · 2022-09-15T14:59:54.000-07:00
* samples: add contents variable to streamFileUpload sample

* check that contents of upload match
diff --git a/samples/streamFileUpload.js b/samples/streamFileUpload.js
@@ -20,7 +20,11 @@
  * at https://cloud.google.com/storage/docs.
  */
 
-function main(bucketName = 'my-bucket', destFileName = 'file.txt') {
+function main(
+  bucketName = 'my-bucket',
+  destFileName = 'file.txt',
+  contents = 'this is the file content'
+) {
   // [START storage_stream_file_upload]
   /**
    * TODO(developer): Uncomment the following lines before running the sample
@@ -31,6 +35,9 @@ function main(bucketName = 'my-bucket', destFileName = 'file.txt') {
   // The new ID for your GCS file
   // const destFileName = 'your-new-file-name';
 
+  // The content to be uploaded in the GCS file
+  // const contents = 'your file content';
+
   // Imports the Google Cloud client library
   const {Storage} = require('@google-cloud/storage');
 
@@ -48,7 +55,7 @@ function main(bucketName = 'my-bucket', destFileName = 'file.txt') {
 
   // Create a pass through stream from a string
   const passthroughStream = new stream.PassThrough();
-  passthroughStream.write('input text');
+  passthroughStream.write(contents);
   passthroughStream.end();
 
   async function streamFileUpload() {
diff --git a/samples/system-test/files.test.js b/samples/system-test/files.test.js
@@ -103,11 +103,13 @@ describe('file', () => {
 
   it('should upload a file using a stream', async () => {
     const output = execSync(
-      `node streamFileUpload.js ${bucketName} ${fileName}`
+      `node streamFileUpload.js ${bucketName} ${fileName} ${fileContents}`
     );
     assert.match(output, new RegExp(`${fileName} uploaded to ${bucketName}`));
     const [exists] = await bucket.file(fileName).exists();
     assert.strictEqual(exists, true);
+    const response = await bucket.file(fileName).download();
+    assert.strictEqual(response[0].toString(), fileContents);
   });
 
   it('should upload a file with a kms key', async () => {
