Ocelot-Social/webapp/composables/useInviteCode.spec.js

import { useInviteCode } from './useInviteCode'

describe('useInviteCode', () => {
  let apollo, toast, t, store, generatePersonalInviteCode, invalidateInviteCode

  beforeEach(() => {
    // Mock apollo.mutate to invoke the update callback (simulating Apollo's behavior)
    apollo = {
      mutate: jest.fn().mockImplementation(({ update }) => {
        const data = { generatePersonalInviteCode: { code: 'new123' }, _invalidateInviteCode: true }
        if (update) update(null, { data })
        return Promise.resolve({ data })
      }),
    }
    toast = { success: jest.fn(), error: jest.fn() }
    t = jest.fn((key) => key)
    store = {
      getters: { 'auth/user': { inviteCodes: [{ code: 'abc', isValid: true }] } },
      commit: jest.fn(),
    }
    ;({ generatePersonalInviteCode, invalidateInviteCode } = useInviteCode({
      apollo,
      toast,
      t,
      store,
    }))
  })

  describe('generatePersonalInviteCode', () => {
    it('calls mutation with comment', async () => {
      await generatePersonalInviteCode('Hello')
      expect(apollo.mutate).toHaveBeenCalledWith(
        expect.objectContaining({ variables: { comment: 'Hello' } }),
      )
    })

    it('shows success toast', async () => {
      await generatePersonalInviteCode('Hello')
      expect(toast.success).toHaveBeenCalledWith('invite-codes.create-success')
    })

    it('returns success true', async () => {
      const result = await generatePersonalInviteCode('Hello')
      expect(result).toEqual({ success: true })
    })

    it('updates store with new invite code', async () => {
      await generatePersonalInviteCode('Hello')
      expect(store.commit).toHaveBeenCalledWith('auth/SET_USER_PARTIAL', {
        inviteCodes: [{ code: 'abc', isValid: true }, { code: 'new123' }],
      })
    })

    it('shows error toast on failure', async () => {
      apollo.mutate.mockRejectedValue(new Error('Ouch'))
      await generatePersonalInviteCode('Hello')
      expect(toast.error).toHaveBeenCalled()
    })

    it('returns success false on failure', async () => {
      apollo.mutate.mockRejectedValue(new Error('Ouch'))
      const result = await generatePersonalInviteCode('Hello')
      expect(result).toEqual({ success: false })
    })
  })

  describe('invalidateInviteCode', () => {
    it('calls mutation with code', async () => {
      await invalidateInviteCode('abc')
      expect(apollo.mutate).toHaveBeenCalledWith(
        expect.objectContaining({ variables: { code: 'abc' } }),
      )
    })

    it('shows success toast', async () => {
      await invalidateInviteCode('abc')
      expect(toast.success).toHaveBeenCalledWith('invite-codes.invalidate-success')
    })

    it('updates store to mark code as invalid', async () => {
      await invalidateInviteCode('abc')
      expect(store.commit).toHaveBeenCalledWith('auth/SET_USER_PARTIAL', {
        inviteCodes: [{ code: 'abc', isValid: false }],
      })
    })

    it('shows error toast on failure', async () => {
      apollo.mutate.mockRejectedValue(new Error('Ouch'))
      await invalidateInviteCode('abc')
      expect(toast.error).toHaveBeenCalled()
    })

    it('returns success false on failure', async () => {
      apollo.mutate.mockRejectedValue(new Error('Ouch'))
      const result = await invalidateInviteCode('abc')
      expect(result).toEqual({ success: false })
    })
  })
})