sushi-bot: Add reaction to add x-logo on 'twitter'

lib/slackMock.js

@@ -1,12 +1,47 @@
+/* eslint-env node, jest */
+
 const EventEmitter = require('events');
 const noop = require('lodash/noop');
+const last = require('lodash/last');
+
+// https://jestjs.io/docs/mock-function-api
+const mockMethodCalls = [
+	'mockImplementation',
+	'mockImplementationOnce',
+	'mockReturnThis',
+	'mockReturnValue',
+	'mockReturnValueOnce',
+	'mockResolvedValue',
+	'mockResolvedValueOnce',
+	'mockRejectedValue',
+	'mockRejectedValueOnce',
+	'mockRestore',
+	'mockClear',
+	'mockReset',
+	'mockName',
+];
 
-const createWebClient = (callback) => {
+const createWebClient = (fallbackFn, registeredMocks) => {
 	const handler = (stack) => {
 		return new Proxy(
-			(...args) => callback(stack, ...args),
+			(...args) => {
+				const path = stack.join('.');
+				if (registeredMocks.has(path)) {
+					return registeredMocks.get(path)(...args);
+				}
+				if (mockMethodCalls.includes(last(stack))) {
+					const mock = jest.fn();
+					registeredMocks.set(stack.slice(0, -1).join('.'), mock);
+					return mock[last(stack)](...args);
+				}
+				return fallbackFn(stack, ...args)
+			},
 			{
 				get: (name, property, receiver) => {
+					const path = [...stack, property].join('.');
+					if (registeredMocks.has(path)) {
+						return registeredMocks.get(path);
+					}
 					if (typeof property === 'string' && property !== 'then') {
 						return handler([...stack, property])
 					}
@@ -26,7 +61,8 @@ module.exports = class SlackMock extends EventEmitter {
 		this.fakeUser = 'U00000000';
 		this.fakeTimestamp = '1234567890.123456';
 		this.eventClient = new EventEmitter();
-		this.webClient = createWebClient((...args) => this.handleWebcall(...args));
+		this.registeredMocks = new Map();
+		this.webClient = createWebClient((...args) => this.handleWebcall(...args), this.registeredMocks);
 		this.messageClient = {
 			action: noop,
 			viewSubmission: noop,
@@ -95,4 +131,11 @@ module.exports = class SlackMock extends EventEmitter {
 		this.postMessage(message, options);
 		return res;
 	}
+
+	// Not recommended. Instanciate a new SlackMock instead.
+	reset() {
+		this.eventClient.removeAllListeners();
+		this.removeAllListeners();
+		this.registeredMocks.clear();
+	}
 };

sushi-bot/index.test.js

@@ -306,3 +306,49 @@ it('reacts to "エクササイズランキング 確認"', () => new Promise((re
 	})();
 }));
 
+it('reacts to "twitter" with :x-logo:', () => new Promise((resolve) => {
+	slack.on('reactions.add', ({name, channel, timestamp}) => {
+		expect(name).toBe('x-logo');
+		expect(channel).toBe(slack.fakeChannel);
+		expect(timestamp).toBe(slack.fakeTimestamp);
+		resolve();
+	});
+
+	slack.eventClient.emit('message', {
+		channel: slack.fakeChannel,
+		text: '私のtwitterアカウントは@hakatashiです',
+		user: slack.fakeUser,
+		ts: slack.fakeTimestamp,
+	});
+}));
+
+it('reacts to "twitter" with :x-logo: case-insensitively', () => new Promise((resolve) => {
+	slack.on('reactions.add', ({name, channel, timestamp}) => {
+		expect(name).toBe('x-logo');
+		expect(channel).toBe(slack.fakeChannel);
+		expect(timestamp).toBe(slack.fakeTimestamp);
+		resolve();
+	});
+
+	slack.eventClient.emit('message', {
+		channel: slack.fakeChannel,
+		text: '私のTwIttERアカウントは@hakatashiです',
+		user: slack.fakeUser,
+		ts: slack.fakeTimestamp,
+	});
+}));
+
+it('does not react to "twitter.com"', async () => {
+	slack.webClient.reactions.add.mockReturnValue(null);
+
+	slack.eventClient.emit('message', {
+		channel: slack.fakeChannel,
+		text: '<https://twitter.com/hakatashi/status/1539187440476246017>',
+		user: slack.fakeUser,
+		ts: slack.fakeTimestamp,
+	});
+
+	await new Promise((resolve) => process.nextTick(resolve));
+
+	expect(slack.webClient.reactions.add).not.toHaveBeenCalled();
+});

sushi-bot/index.ts

@@ -336,6 +336,12 @@ export default async function ({eventClient, webClient: slack}: SlackInterface)
 				}
 			}
 		}
+
+		{
+			if (text.match(/twitter(?!\.com)/i)) {
+				slack.reactions.add({name: 'x-logo', channel, timestamp})
+			}
+		}
 	});
 
 	schedule.scheduleJob('0 19 * * *', async (date) => {