monkeeShark/packages/megalodon/test/unit/webo_socket.spec.ts

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)
        })
      })
    })
  })
})