upd: megalodon to v7

1 files changed, 184 insertions, 0 deletions

diff --git a/packages/megalodon/test/unit/webo_socket.spec.ts b/packages/megalodon/test/unit/webo_socket.spec.ts
new file mode 100644
index 0000000000..bb9f997a57
--- /dev/null
+++ b/packages/megalodon/test/unit/webo_socket.spec.ts
@@ -0,0 +1,184 @@
+import { Parser } from '@/mastodon/web_socket'
+import Entity from '@/entity'
+
+const account: Entity.Account = {
+  id: '1',
+  username: 'h3poteto',
+  acct: 'h3poteto@pleroma.io',
+  display_name: 'h3poteto',
+  locked: false,
+  group: false,
+  noindex: null,
+  suspended: null,
+  limited: null,
+  created_at: '2019-03-26T21:30:32',
+  followers_count: 10,
+  following_count: 10,
+  statuses_count: 100,
+  note: 'engineer',
+  url: 'https://pleroma.io',
+  avatar: '',
+  avatar_static: '',
+  header: '',
+  header_static: '',
+  emojis: [],
+  moved: null,
+  fields: [],
+  bot: false
+}
+const status: Entity.Status = {
+  id: '1',
+  uri: 'http://example.com',
+  url: 'http://example.com',
+  account: account,
+  in_reply_to_id: null,
+  in_reply_to_account_id: null,
+  reblog: null,
+  content: 'hoge',
+  plain_content: 'hoge',
+  created_at: '2019-03-26T21:40:32',
+  emojis: [],
+  replies_count: 0,
+  reblogs_count: 0,
+  favourites_count: 0,
+  reblogged: null,
+  favourited: null,
+  muted: null,
+  sensitive: false,
+  spoiler_text: '',
+  visibility: 'public',
+  media_attachments: [],
+  mentions: [],
+  tags: [],
+  card: null,
+  poll: null,
+  application: {
+    name: 'Web'
+  } as Entity.Application,
+  language: null,
+  pinned: null,
+  emoji_reactions: [],
+  bookmarked: false,
+  quote: false
+}
+
+const notification: Entity.Notification = {
+  id: '1',
+  account: account,
+  status: status,
+  type: 'favourite',
+  created_at: '2019-04-01T17:01:32'
+}
+
+const conversation: Entity.Conversation = {
+  id: '1',
+  accounts: [account],
+  last_status: status,
+  unread: true
+}
+
+describe('Parser', () => {
+  let parser: Parser
+
+  beforeEach(() => {
+    parser = new Parser()
+  })
+
+  describe('parse', () => {
+    describe('message is heartbeat', () => {
+      describe('message is an object', () => {
+        const message = Buffer.alloc(0)
+
+        it('should be called', () => {
+          const spy = jest.fn()
+          parser.once('heartbeat', spy)
+          parser.parse(message, true)
+          expect(spy).toHaveBeenCalledWith({})
+        })
+      })
+      describe('message is empty string', () => {
+        const message: string = ''
+
+        it('should be called', () => {
+          const spy = jest.fn()
+          parser.once('heartbeat', spy)
+          parser.parse(Buffer.from(message), false)
+          expect(spy).toHaveBeenCalledWith({})
+        })
+      })
+    })
+
+    describe('message is not json', () => {
+      describe('event is delete', () => {
+        const message = JSON.stringify({
+          event: 'delete',
+          payload: '12asdf34'
+        })
+
+        it('should be called', () => {
+          const spy = jest.fn()
+          parser.once('delete', spy)
+          parser.parse(Buffer.from(message), false)
+          expect(spy).toHaveBeenCalledWith('12asdf34')
+        })
+      })
+      describe('event is not delete', () => {
+        const message = JSON.stringify({
+          event: 'event',
+          payload: '12asdf34'
+        })
+
+        it('should be called', () => {
+          const error = jest.fn()
+          const deleted = jest.fn()
+          parser.once('error', error)
+          parser.once('delete', deleted)
+          parser.parse(Buffer.from(message), false)
+          expect(error).toHaveBeenCalled()
+          expect(deleted).not.toHaveBeenCalled()
+        })
+      })
+    })
+
+    describe('message is json', () => {
+      describe('event is update', () => {
+        const message = JSON.stringify({
+          event: 'update',
+          payload: JSON.stringify(status)
+        })
+        it('should be called', () => {
+          const spy = jest.fn()
+          parser.once('update', spy)
+          parser.parse(Buffer.from(message), false)
+          expect(spy).toHaveBeenCalledWith(status)
+        })
+      })
+
+      describe('event is notification', () => {
+        const message = JSON.stringify({
+          event: 'notification',
+          payload: JSON.stringify(notification)
+        })
+        it('should be called', () => {
+          const spy = jest.fn()
+          parser.once('notification', spy)
+          parser.parse(Buffer.from(message), false)
+          expect(spy).toHaveBeenCalledWith(notification)
+        })
+      })
+
+      describe('event is conversation', () => {
+        const message = JSON.stringify({
+          event: 'conversation',
+          payload: JSON.stringify(conversation)
+        })
+        it('should be called', () => {
+          const spy = jest.fn()
+          parser.once('conversation', spy)
+          parser.parse(Buffer.from(message), false)
+          expect(spy).toHaveBeenCalledWith(conversation)
+        })
+      })
+    })
+  })
+})