@sphereon/ssi-sdk.data-store/src/__tests__/pd-manager.entities.test.ts

import { DataSources } from '@sphereon/ssi-sdk.agent-config'
import { DataSource } from 'typeorm'
import { PresentationDefinitionItemEntity } from '../entities/presentationDefinition/PresentationDefinitionItemEntity'
import { DataStorePresentationDefinitionMigrations } from '../migrations'
import { DataStorePresentationDefinitionEntities } from '../index'
import { afterEach, beforeEach, describe, expect, it } from 'vitest'

describe('PresentationDefinitionItemEntity tests', (): void => {
  let dbConnection: DataSource

  beforeEach(async (): Promise<void> => {
    DataSources.singleInstance().defaultDbType = 'sqlite'
    dbConnection = await new DataSource({
      type: 'sqlite',
      database: ':memory:',
      logging: ['info'],
      synchronize: true,
      migrationsRun: false,
      migrations: DataStorePresentationDefinitionMigrations,
      entities: DataStorePresentationDefinitionEntities,
    }).initialize()
  })

  afterEach(async (): Promise<void> => {
    await dbConnection.destroy()
  })

  it('should create and retrieve PresentationDefinitionItemEntity with dcqlPayload', async (): Promise<void> => {
    const repository = dbConnection.getRepository(PresentationDefinitionItemEntity)
    const entity = new PresentationDefinitionItemEntity()
    entity.definitionId = 'definition1'
    entity.version = '1.0'
    entity.definitionPayload = JSON.stringify({ id: 'definition1', input_descriptors: [] })
    entity.dcqlPayload = JSON.stringify({
      credentials: [
        {
          id: 'credential1',
          format: 'jwt_vc',
          claims: [
            {
              namespace: 'test',
              claim_name: 'testClaim',
            },
          ],
        },
      ],
    })

    const savedEntity = await repository.save(entity)
    expect(savedEntity).toBeDefined()
    expect(savedEntity.id).toBeDefined()
    expect(savedEntity.dcqlPayload).toBeDefined()

    const retrievedEntity = await repository.findOneBy({ id: savedEntity.id })
    expect(retrievedEntity).toBeDefined()
    expect(retrievedEntity!.dcqlPayload).toBeDefined()
    const parsedDcql = JSON.parse(retrievedEntity!.dcqlPayload)
    expect(parsedDcql.credentials[0].id).toEqual('credential1')
  })

  it('should update PresentationDefinitionItemEntity dcqlPayload', async (): Promise<void> => {
    const repository = dbConnection.getRepository(PresentationDefinitionItemEntity)
    const entity = new PresentationDefinitionItemEntity()
    entity.definitionId = 'definition1'
    entity.version = '1.0'
    entity.definitionPayload = JSON.stringify({ id: 'definition1', input_descriptors: [] })
    entity.dcqlPayload = JSON.stringify({
      credentials: [
        {
          id: 'credential1',
          format: 'jwt_vc',
          claims: [
            {
              namespace: 'test',
              claim_name: 'testClaim',
            },
          ],
        },
      ],
    })

    const savedEntity = await repository.save(entity)
    expect(savedEntity).toBeDefined()

    const updatedDcql = {
      credentials: [
        {
          id: 'credential2',
          format: 'jwt_vc',
          claims: [
            {
              namespace: 'test',
              claim_name: 'updatedClaim',
            },
          ],
        },
      ],
    }
    savedEntity.dcqlPayload = JSON.stringify(updatedDcql)
    const updatedEntity = await repository.save(savedEntity)
    expect(updatedEntity).toBeDefined()
    expect(JSON.parse(updatedEntity.dcqlPayload).credentials[0].id).toEqual('credential2')
  })

  it('should create and retrieve PresentationDefinitionItemEntity', async (): Promise<void> => {
    const repository = dbConnection.getRepository(PresentationDefinitionItemEntity)
    const entity = new PresentationDefinitionItemEntity()
    entity.definitionId = 'definition1'
    entity.version = '1.0'
    entity.definitionPayload = JSON.stringify({ id: 'definition1', input_descriptors: [] })

    const savedEntity = await repository.save(entity)
    expect(savedEntity).toBeDefined()
    expect(savedEntity.id).toBeDefined()

    const retrievedEntity = await repository.findOneBy({ id: savedEntity.id })
    expect(retrievedEntity).toBeDefined()
    expect(retrievedEntity!.definitionId).toEqual('definition1')
  })

  it('should update PresentationDefinitionItemEntity', async (): Promise<void> => {
    const repository = dbConnection.getRepository(PresentationDefinitionItemEntity)
    const entity = new PresentationDefinitionItemEntity()
    entity.definitionId = 'definition1'
    entity.version = '1.0'
    entity.definitionPayload = JSON.stringify({ id: 'definition1', input_descriptors: [] })

    const savedEntity = await repository.save(entity)
    expect(savedEntity).toBeDefined()

    savedEntity.version = '1.1'
    const updatedEntity = await repository.save(savedEntity)
    expect(updatedEntity).toBeDefined()
    expect(updatedEntity.version).toEqual('1.1')
  })

  it('should delete PresentationDefinitionItemEntity', async (): Promise<void> => {
    const repository = dbConnection.getRepository(PresentationDefinitionItemEntity)
    const entity = new PresentationDefinitionItemEntity()
    entity.definitionId = 'definition1'
    entity.version = '1.0'
    entity.definitionPayload = JSON.stringify({ id: 'definition1', input_descriptors: [] })

    const savedEntity = await repository.save(entity)
    expect(savedEntity).toBeDefined()

    await repository.delete(savedEntity.id)
    const retrievedEntity = await repository.findOneBy({ id: savedEntity.id })
    expect(retrievedEntity).toBeNull()
  })
})