import type { StorageService, EventEmitter } from '@aries-framework/core'

import { Key } from '@aries-framework/core'

import { getAgentContext, mockFunction } from '../../../../core/tests/helpers'
import { TenantRoutingRecord } from '../TenantRoutingRecord'
import { TenantRoutingRepository } from '../TenantRoutingRepository'

const storageServiceMock = {
  findByQuery: jest.fn(),
} as unknown as StorageService<TenantRoutingRecord>
const eventEmitter = jest.fn() as unknown as EventEmitter
const agentContext = getAgentContext()

const tenantRoutingRepository = new TenantRoutingRepository(storageServiceMock, eventEmitter)

describe('TenantRoutingRepository', () => {
  afterEach(() => {
    jest.clearAllMocks()
  })

  describe('findByRecipientKey', () => {
    test('it should correctly transform the key to a fingerprint and return the routing record', async () => {
      const key = Key.fromFingerprint('z6Mkk7yqnGF3YwTrLpqrW6PGsKci7dNqh1CjnvMbzrMerSeL')
      const tenantRoutingRecord = new TenantRoutingRecord({
        recipientKeyFingerprint: key.fingerprint,
        tenantId: 'tenant-id',
      })

      mockFunction(storageServiceMock.findByQuery).mockResolvedValue([tenantRoutingRecord])
      const returnedRecord = await tenantRoutingRepository.findByRecipientKey(agentContext, key)

      expect(storageServiceMock.findByQuery).toHaveBeenCalledWith(agentContext, TenantRoutingRecord, {
        recipientKeyFingerprint: 'z6Mkk7yqnGF3YwTrLpqrW6PGsKci7dNqh1CjnvMbzrMerSeL',
      })
      expect(returnedRecord).toBe(tenantRoutingRecord)
    })
  })
})