fix: command upload test (#139)

screwdriver-cd · Apr 25, 2023 · f44cb72 · f44cb72
1 parent 3b003eb
commit f44cb72
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/test/helpers/aws.test.js b/test/helpers/aws.test.js
@@ -181,12 +181,14 @@ describe('aws helper test', () => {
     it('upload command as stream', () => {
         const uploadParam = {
             Bucket: testBucket,
-            Key: `caches/${cacheKey}`
+            Key: `command/${cacheKey}`
         };
         const uploadOption = {
             partSize
         };
 
+        awsClient.segment = 'command';
+
         return awsClient.uploadCommandAsStream({ cacheKey, payload: Buffer.from('hellow world', 'utf8') }).then(() => {
             assert.calledWith(clientMock.prototype.upload, sinon.match(uploadParam), sinon.match(uploadOption));
         });

diff --git a/test/plugins/commands.test.js b/test/plugins/commands.test.js
@@ -296,7 +296,6 @@ describe('commands plugin test using s3', () => {
     let uploadAsStreamMock;
     let deleteObjMock;
     let getDownloadMock;
-    let uploadDirectMock;
     let data;
 
     before(() => {
@@ -317,16 +316,13 @@ describe('commands plugin test using s3', () => {
         uploadAsStreamMock = sinon.stub().resolves(null);
         deleteObjMock = sinon.stub().resolves(null);
         getDownloadMock = sinon.stub().resolves(null);
-        uploadDirectMock = sinon.stub().resolves(null);
 
         awsClientMock = sinon.stub().returns({
             updateLastModified: sinon.stub().yields(null),
             removeObject: deleteObjMock,
             getDownloadStream: getDownloadStreamMock,
-            uploadCmdAsStream: uploadAsStreamMock,
             getDownloadObject: getDownloadMock,
-            uploadAsBuffer: uploadDirectMock,
-            uploadCommandAsStream: uploadDirectMock
+            uploadCommandAsStream: uploadAsStreamMock
         });
 
         data = {