twenty/packages/twenty-server/src/api/rest/api-rest.controller.utils.spec.ts

import { cleanGraphQLResponse } from 'src/api/rest/api-rest.controller.utils';

describe('cleanGraphQLResponse', () => {
  it('should remove edges/node from results', () => {
    const data = {
      companies: {
        edges: [
          {
            node: { id: 'id', createdAt: '2023-01-01' },
          },
        ],
      },
    };
    const expectedResult = {
      companies: [{ id: 'id', createdAt: '2023-01-01' }],
    };

    expect(cleanGraphQLResponse(data)).toEqual(expectedResult);
  });
  it('should remove nested edges/node from results', () => {
    const data = {
      companies: {
        edges: [
          {
            node: {
              id: 'id',
              createdAt: '2023-01-01',
              people: {
                edges: [{ node: { id: 'id1' } }, { node: { id: 'id2' } }],
              },
            },
          },
        ],
      },
    };
    const expectedResult = {
      companies: [
        {
          id: 'id',
          createdAt: '2023-01-01',
          people: [{ id: 'id1' }, { id: 'id2' }],
        },
      ],
    };

    expect(cleanGraphQLResponse(data)).toEqual(expectedResult);
  });
  it('should not format when no list returned', () => {
    const data = { company: { id: 'id' } };

    expect(cleanGraphQLResponse(data)).toEqual(data);
  });
});