update iam test

Author: horike37

lib/iam.js

@@ -157,12 +157,6 @@ module.exports = {
     });
 
     return BbPromise.map(promises, (state) => this.deleteIamRole(state))
-    .then(() => BbPromise.resolve())
-    .catch((error) => {
-      if (error.statusCode === 404) {
-        return BbPromise.resolve();
-      }
-      return BbPromise.reject(error.message);
-    });
+    .then(() => BbPromise.resolve());
   },
 };

lib/iam.test.js

@@ -151,13 +151,68 @@ describe('iam', () => {
         serverlessStepFunctions.provider.request.restore();
       })
     );
+    
+    it('should deleteIamRole when 404 error', () => {
+      serverlessStepFunctions.provider.request.restore();
+      deleteIamRoleStub = sinon.stub(serverlessStepFunctions.provider, 'request')
+      .returns(BbPromise.reject({ statusCode: 404 }));
+      serverlessStepFunctions.deleteIamRole()
+      .then(() => {
+        expect(deleteIamRoleStub.callCount).to.be.equal(1);
+        serverlessStepFunctions.provider.request.restore();
+      })
+    });
+
+    it('should other when other error', () => {
+      serverlessStepFunctions.provider.request.restore();
+      deleteIamRoleStub = sinon.stub(serverlessStepFunctions.provider, 'request')
+      .returns(BbPromise.reject({ statusCode: 500 }));
+      serverlessStepFunctions.deleteIamRole()
+      .catch(() => {
+        expect(deleteIamRoleStub.callCount).to.be.equal(1);
+        serverlessStepFunctions.provider.request.restore();
+      })
+    });
   });
 
   describe('#getIamRoles()', () => {
-    // @todo
+    let getIamRoleStub;
+    beforeEach(() => {
+      serverless.service.stepFunctions = {
+        helloHello: 'value',
+        hogeHoge: 'value',
+        sssss: 'value',
+      };
+      getIamRoleStub = sinon
+        .stub(serverlessStepFunctions, 'getIamRole').returns(BbPromise.resolve());
+    });
+
+    it('should getIamRoles with correct params'
+    , () => serverlessStepFunctions.getIamRoles()
+      .then(() => {
+        expect(getIamRoleStub.calledThrice).to.be.equal(true);
+        serverlessStepFunctions.getIamRole.restore();
+      })
+    );
   });
 
   describe('#deleteIamRoles()', () => {
-    // @todo
+    let deleteIamRoleStub;
+    beforeEach(() => {
+      serverless.service.stepFunctions = {
+        helloHello: 'value',
+        hogeHoge: 'value',
+      };
+      deleteIamRoleStub = sinon
+        .stub(serverlessStepFunctions, 'deleteIamRole').returns(BbPromise.resolve());
+    });
+
+    it('should deleteIamRoles with correct params'
+    , () => serverlessStepFunctions.deleteIamRoles()
+      .then(() => {
+        expect(deleteIamRoleStub.calledTwice).to.be.equal(true);
+        serverlessStepFunctions.deleteIamRole.restore();
+      })
+    );
   });
 });