add tag to generic template and text message

ShellMonkeys · Jun 6, 2017 · adc7f3a · adc7f3a
1 parent 303cc32
commit adc7f3a
Show file tree

Hide file tree

Showing 5 changed files with 29 additions and 1 deletion.
diff --git a/src/client/client.js b/src/client/client.js
@@ -79,6 +79,10 @@ export default class Client {
             notification_type: notificationType,
         };
 
+        if (message.tag) {
+            facebookEnvelope.tag = message.tag;
+        }
+
         return this.proxyFetchFacebook(`${this.baseURL}/me/messages?access_token=${this.pageAccessToken}`, { body: JSON.stringify(facebookEnvelope) })
             .catch((error) => {
                 log.error(error);

diff --git a/src/message/contentTypes/textMessage.js b/src/message/contentTypes/textMessage.js
@@ -13,4 +13,10 @@ export default class TextMessage extends Message {
         this.state.text = text;
         return this;
     }
+
+    withTag(tag) {
+        validate.oneOf(tag, ['ISSUE_RESOLUTION'], 'tag', 'TextMessage.withTag');
+        this.tag = tag;
+        return this;
+    }
 }
diff --git a/src/message/templates/genericTemplate.js b/src/message/templates/genericTemplate.js
@@ -43,4 +43,10 @@ export default class GenericTemplate extends Template {
         this.state.attachment.payload.elements.push(element);
         return this;
     }
+
+    withTag(tag) {
+        validate.oneOf(tag, ['SHIPPING_UPDATE', 'RESERVATION_UPDATE', 'ISSUE_RESOLUTION'], 'tag', 'GenericTemplate.withTag');
+        this.tag = tag;
+        return this;
+    }
 }
diff --git a/.../message/contentTypes/testMessage.test.js → .../message/contentTypes/textMessage.test.js b/.../message/contentTypes/testMessage.test.js → .../message/contentTypes/textMessage.test.js
@@ -10,3 +10,14 @@ test('TextMessage.constructor', (expect) => {
     }, 'should have the correct structure');
     expect.end();
 });
+
+test('TextMessage.withTag', (expect) => {
+    const testTextMsg = new TextMessage('hello, world!').withTag('ISSUE_RESOLUTION');
+    expect.same(testTextMsg, {
+        state: {
+            text: 'hello, world!',
+        },
+        tag: 'ISSUE_RESOLUTION',
+    }, 'should have the correct structure');
+    expect.end();
+});
diff --git a/test/message/templates/genericTemplate.test.js b/test/message/templates/genericTemplate.test.js
@@ -13,7 +13,7 @@ test('GenericTemplate', (expect) => {
             .setImageUrl('https//bot.peters-hats.com/img/hats/fez.jpg')
             .setDefaultAction(new DefaultAction('http://m.me/petershats?ref=invited_by_24601'))
             .addButton(new UrlButton('Take Quiz', 'http://m.me/petershats?ref=invited_by_24601')),
-    ]).addElement(new GenericElement('Additional'));
+    ]).addElement(new GenericElement('Additional')).withTag('SHIPPING_UPDATE');
     expect.same(testTemplate, {
         state: {
             attachment: {
@@ -46,6 +46,7 @@ test('GenericTemplate', (expect) => {
                 },
             },
         },
+        tag: 'SHIPPING_UPDATE',
     }, 'should have the correct structure');
     expect.end();
 });