test: update test

.eslintignore

@@ -1,2 +1,3 @@
 docs/
 examples/
+__tests__/

__tests__/index.test.ts

@@ -1,8 +1,76 @@
 
-import func from '../libs/index'
+import CloudFrontUpdator, {
+  interfaces
+} from '../libs/index'
 
+const condition: interfaces.Config = {
+  debugMode: true,
+  taskType: 'parallel',
+  allowSensitiveAction: false
+}
+class DummyClient {
+  // @ts-ignore
+  getDistributionConfig (param: any) {
+    return {
+      promise (): AWS.CloudFront.GetDistributionConfigResult {
+        const DistributionConfig: AWS.CloudFront.DistributionConfig = {
+          Enabled: true
+        } as AWS.CloudFront.DistributionConfig
+        return {
+          ETag: 'tag',
+          DistributionConfig
+        }
+      }
+    }
+  }
+
+  // @ts-ignore
+  updateDistribution (param: any) {
+    return {
+      promise () {
+        return {}
+      }
+    }
+  }
+
+  // @ts-ignore
+  listDistributions (param: any) {
+    return {
+      promise () {
+        return {}
+      }
+    }
+  }
+}
 describe('dummy', () => {
-    it('test', () => {
-        expect(func('John')).toEqual('Hello John !')
+  // @ts-ignore
+  const updator: interfaces.UpdatorFunction = (id, conf) => conf
+  const filter: interfaces.FilterCondition = () => true
+  let client = new CloudFrontUpdator({
+    updator,
+    filter
+  }, condition, {
+    cfClient: new DummyClient() as any as AWS.CloudFront
+  })
+  it('should work', async () => {
+    const dist = {} as AWS.CloudFront.Distribution
+    await expect(client.updateDistribution(dist)).resolves.toBe(undefined)
+  })
+  it('should throw error', async () => {
+    client = new CloudFrontUpdator({
+      updator: (id, conf) => {
+        console.log(id)
+        conf.Enabled = false
+        return conf
+      },
+      filter
+    }, condition, {
+      cfClient: new DummyClient() as any as AWS.CloudFront
     })
+    const dist = {
+      Id: 'ETEST'
+    } as AWS.CloudFront.Distribution
+    const promise = client.updateDistribution(dist)
+    await expect(promise).rejects.toMatch("You can not allow the action, please set 'allowSensitiveAction' option in the constructor")
+  })
 })

libs/index.ts

@@ -111,9 +111,10 @@ export class CloudFrontUpdator {
       }).promise()
       if (!DistributionConfig) throw new Error('No such distribution')
       if (!ETag) throw new Error('no ETag')
+      const beforeConfig = Object.assign({}, DistributionConfig)
       const config = await this.updator(distribution.Id, DistributionConfig)
       if (!this.allowSensitiveAction && config) {
-        if (DistributionConfig.Enabled !== config.Enabled) {
+        if (beforeConfig.Enabled !== config.Enabled) {
           const err = 'You can not allow the action, please set \'allowSensitiveAction\' option in the constructor'
           this.log(err)
           throw err