add sns publish

s2texperiments · Jun 4, 2018 · 554be0a · 554be0a
1 parent 8dcec1c
commit 554be0a
Show file tree

Hide file tree

Showing 6 changed files with 91 additions and 25 deletions.
diff --git a/deploy_s3.sh b/deploy_s3.sh
@@ -1,4 +1,4 @@
 #!/bin/bash
 
 mkdir deploy
-zip -r deploy/s2t-incoming-not-transcoded-file-eventhandler.zip index.js s3Api.js node_modules/
+zip -r deploy/s2t-incoming-not-transcoded-file-eventhandler.zip index.js s3Api.js snsApi.js node_modules/
diff --git a/index.js b/index.js
@@ -1,5 +1,6 @@
 const test = process.env['MODE'] === 'test';
 let s3Api = test ? require('./test/s3ApiMock.js') : require('./s3Api.js');
+let snsApi = test ? require('./test/snsApiMock.js') : require('./snsApi.js');
 
 exports.handler = async (event) => {
 
@@ -33,6 +34,33 @@ exports.handler = async (event) => {
         Key: key
     });
 
+    await snsApi.publish({
+        Message: 'placeholder',
+        MessageAttributes: {
+            bucket: {
+                DataType: 'String',
+                StringValue: bucketName
+            },
+            key: {
+                DataType: 'String',
+                StringValue: key
+            },
+            "transcribe-provider": {
+                DataType: 'String',
+                StringValue: provider
+            },
+            pid: {
+                DataType: 'String',
+                StringValue: pid
+            },
+            "api-key-id": {
+                DataType: 'String',
+                StringValue: apiKeyId
+            },
+        },
+        TopicArn: process.env['TOPIC_ARN']
+    });
+
     return {
         bucket: bucketName,
         key: key,

diff --git a/test/.env b/test/.env
@@ -1 +1,2 @@
-MODE=test
+MODE=test
+TOPIC_ARN=given:arn:from:env
diff --git a/test/snsApiMock.js b/test/snsApiMock.js
@@ -9,7 +9,7 @@ module.exports = {
     },
     publish: async (params) => {
         expect(params).to.deep.equal(expectedParams);
-        console.log(`expected head params: ${JSON.stringify(expectedParams)}`);
+        console.log(`expected publish params: ${JSON.stringify(expectedParams)}`);
         return new Promise((resolve, rejected) => resolve());
     }
 };
diff --git a/test/snsEventData.json b/test/snsEventData.json
diff --git a/test/test.js b/test/test.js
@@ -10,10 +10,12 @@ const event = JSON.parse(fs.readFileSync('test/s3EventData.json', 'utf8'));
 
 describe('eventhandler', () => {
 
-    let s3ApiMock;
+    let s3ApiMock, snsApiMock;
     beforeEach(() => {
         s3ApiMock = require('./s3ApiMock');
+        snsApiMock = require('./snsApiMock');
         s3ApiMock.reset();
+        snsApiMock.reset();
     });
 
     it('collect head request without meta data', async () => {
@@ -22,6 +24,34 @@ describe('eventhandler', () => {
             Key: "gcp/not-transcoded/sugr1km8s6/f423fbfb-6381-11e8-a23f-c7cbebde15f2.ogg"
         });
 
+        snsApiMock.expectedPublishParams(
+            {
+                Message: "placeholder",
+                MessageAttributes: {
+                    "api-key-id": {
+                        DataType: "String",
+                        StringValue: "sugr1km8s6"
+                    },
+                    bucket: {
+                        DataType: "String",
+                        StringValue: "s2t-base-s2tbucket-19xbw73dypb0s"
+                    },
+                    key: {
+                        DataType: "String",
+                        StringValue: "gcp/not-transcoded/sugr1km8s6/f423fbfb-6381-11e8-a23f-c7cbebde15f2.ogg"
+                    },
+                    pid: {
+                        DataType: "String",
+                        StringValue: "f423fbfb-6381-11e8-a23f-c7cbebde15f2"
+                    },
+                    "transcribe-provider": {
+                        DataType: "String",
+                        StringValue: "gcp"
+                    }
+                },
+                "TopicArn": "given:arn:from:env"
+            });
+
         s3ApiMock.givenResponseHeadObject({
             AcceptRanges: "bytes",
             ContentLength: 3191,
@@ -51,6 +81,34 @@ describe('eventhandler', () => {
             Key: "gcp/not-transcoded/sugr1km8s6/f423fbfb-6381-11e8-a23f-c7cbebde15f2.ogg"
         });
 
+        snsApiMock.expectedPublishParams(
+            {
+                Message: "placeholder",
+                MessageAttributes: {
+                    "api-key-id": {
+                        DataType: "String",
+                        StringValue: "sugr1km8s6"
+                    },
+                    bucket: {
+                        DataType: "String",
+                        StringValue: "s2t-base-s2tbucket-19xbw73dypb0s"
+                    },
+                    key: {
+                        DataType: "String",
+                        StringValue: "gcp/not-transcoded/sugr1km8s6/f423fbfb-6381-11e8-a23f-c7cbebde15f2.ogg"
+                    },
+                    pid: {
+                        DataType: "String",
+                        StringValue: "f423fbfb-6381-11e8-a23f-c7cbebde15f2"
+                    },
+                    "transcribe-provider": {
+                        DataType: "String",
+                        StringValue: "gcp"
+                    }
+                },
+                "TopicArn": "given:arn:from:env"
+            });
+
         s3ApiMock.givenResponseHeadObject({
             AcceptRanges: "bytes",
             ContentLength: 3191,