Compass/backend/api/tests/unit/block-user.unit.test.ts

jest.mock('shared/supabase/init')
jest.mock('shared/supabase/users')
jest.mock('shared/supabase/utils')

import * as blockUserModule from 'api/block-user'
import {AuthedUser} from 'api/helpers/endpoint'
import * as supabaseInit from 'shared/supabase/init'
import * as supabaseUsers from 'shared/supabase/users'
import * as supabaseUtils from 'shared/supabase/utils'

describe('blockUser', () => {
  let mockPg: any

  beforeEach(() => {
    jest.resetAllMocks()
    mockPg = {
      tx: jest.fn(async (cb) => {
        const mockTx = {}
        await cb(mockTx)
      }),
    }
    ;(supabaseInit.createSupabaseDirectClient as jest.Mock).mockReturnValue(mockPg)
  })
  afterEach(() => {
    jest.restoreAllMocks()
  })

  describe('when given valid input', () => {
    it('block the user successfully', async () => {
      const mockParams = {id: '123'}
      const mockAuth = {uid: '321'} as AuthedUser
      const mockReq = {} as any

      await blockUserModule.blockUser(mockParams, mockAuth, mockReq)

      expect(mockPg.tx).toHaveBeenCalledTimes(1)
      expect(supabaseUsers.updatePrivateUser).toBeCalledTimes(2)
      expect(supabaseUsers.updatePrivateUser).toHaveBeenNthCalledWith(
        1,
        expect.any(Object),
        mockAuth.uid,
        {blockedByUserIds: supabaseUtils.FieldVal.arrayConcat(mockParams.id)},
      )
      expect(supabaseUsers.updatePrivateUser).toHaveBeenNthCalledWith(
        2,
        expect.any(Object),
        mockParams.id,
        {blockedByUserIds: supabaseUtils.FieldVal.arrayConcat(mockAuth.uid)},
      )
    })
  })
  describe('when an error occurs', () => {
    it('throw an error if the user tries to block themselves', async () => {
      const mockParams = {id: '123'}
      const mockAuth = {uid: '123'} as AuthedUser
      const mockReq = {} as any

      expect(blockUserModule.blockUser(mockParams, mockAuth, mockReq)).rejects.toThrowError(
        'You cannot block yourself',
      )
    })
  })
})

describe('unblockUser', () => {
  let mockPg: any

  beforeEach(() => {
    jest.resetAllMocks()
    mockPg = {
      tx: jest.fn(async (cb) => {
        const mockTx = {}
        await cb(mockTx)
      }),
    }
    ;(supabaseInit.createSupabaseDirectClient as jest.Mock).mockReturnValue(mockPg)
  })
  afterEach(() => {
    jest.restoreAllMocks()
  })

  describe('when given valid input', () => {
    it('should block the user successfully', async () => {
      const mockParams = {id: '123'}
      const mockAuth = {uid: '321'} as AuthedUser
      const mockReq = {} as any

      await blockUserModule.unblockUser(mockParams, mockAuth, mockReq)

      expect(mockPg.tx).toHaveBeenCalledTimes(1)
      expect(supabaseUsers.updatePrivateUser).toBeCalledTimes(2)
      expect(supabaseUsers.updatePrivateUser).toHaveBeenNthCalledWith(
        1,
        expect.any(Object),
        mockAuth.uid,
        {blockedByUserIds: supabaseUtils.FieldVal.arrayConcat(mockParams.id)},
      )
      expect(supabaseUsers.updatePrivateUser).toHaveBeenNthCalledWith(
        2,
        expect.any(Object),
        mockParams.id,
        {blockedByUserIds: supabaseUtils.FieldVal.arrayConcat(mockAuth.uid)},
      )
    })
  })
})