upstream-twenty/packages/twenty-server/test/integration/graphql/suites/mutate-by-relation-field.integration-spec.ts

import { createManyOperationFactory } from 'test/integration/graphql/utils/create-many-operation-factory.util';
import { deleteManyOperationFactory } from 'test/integration/graphql/utils/delete-many-operation-factory.util';
import { destroyManyOperationFactory } from 'test/integration/graphql/utils/destroy-many-operation-factory.util';
import { findManyOperationFactory } from 'test/integration/graphql/utils/find-many-operation-factory.util';
import { makeGraphqlAPIRequest } from 'test/integration/graphql/utils/make-graphql-api-request.util';
import { restoreManyOperationFactory } from 'test/integration/graphql/utils/restore-many-operation-factory.util';
import { updateManyOperationFactory } from 'test/integration/graphql/utils/update-many-operation-factory.util';

// Distinct ID prefix (eeee / ffff) from filter-by-relation-field.integration-
// spec.ts so the two suites can share the workspace database without colliding.
const TEST_COMPANY_IDS = {
  AIRBNB: '20202020-eeee-4000-8000-000000000001',
  STRIPE: '20202020-eeee-4000-8000-000000000002',
  NOTION: '20202020-eeee-4000-8000-000000000003',
};

const TEST_PERSON_IDS = {
  AIRBNB_ENGINEER: '20202020-ffff-4000-8000-000000000001',
  AIRBNB_DESIGNER: '20202020-ffff-4000-8000-000000000002',
  STRIPE_ENGINEER: '20202020-ffff-4000-8000-000000000003',
  NOTION_ENGINEER: '20202020-ffff-4000-8000-000000000004',
  UNAFFILIATED: '20202020-ffff-4000-8000-000000000005',
};

const ALL_TEST_PERSON_IDS = Object.values(TEST_PERSON_IDS);

// Each of the four many-record mutations must support relation-traversal
// filters: the join survives into the generated UPDATE / DELETE / SoftDelete /
// Restore SQL via an `id IN (subquery)` rewrite.
describe('Mutate by relation field (e2e)', () => {
  const resetFixtures = async () => {
    const createCompanies = createManyOperationFactory({
      objectMetadataSingularName: 'company',
      objectMetadataPluralName: 'companies',
      gqlFields: 'id name',
      data: [
        { id: TEST_COMPANY_IDS.AIRBNB, name: 'AirbnbMutate' },
        { id: TEST_COMPANY_IDS.STRIPE, name: 'StripeMutate' },
        { id: TEST_COMPANY_IDS.NOTION, name: 'NotionMutate' },
      ],
      upsert: true,
    });

    await makeGraphqlAPIRequest(createCompanies);

    const createPeople = createManyOperationFactory({
      objectMetadataSingularName: 'person',
      objectMetadataPluralName: 'people',
      gqlFields: 'id',
      data: [
        {
          id: TEST_PERSON_IDS.AIRBNB_ENGINEER,
          companyId: TEST_COMPANY_IDS.AIRBNB,
          jobTitle: 'Original City',
        },
        {
          id: TEST_PERSON_IDS.AIRBNB_DESIGNER,
          companyId: TEST_COMPANY_IDS.AIRBNB,
          jobTitle: 'Original City',
        },
        {
          id: TEST_PERSON_IDS.STRIPE_ENGINEER,
          companyId: TEST_COMPANY_IDS.STRIPE,
          jobTitle: 'Original City',
        },
        {
          id: TEST_PERSON_IDS.NOTION_ENGINEER,
          companyId: TEST_COMPANY_IDS.NOTION,
          jobTitle: 'Original City',
        },
        {
          id: TEST_PERSON_IDS.UNAFFILIATED,
          companyId: null,
          jobTitle: 'Original City',
        },
      ],
      upsert: true,
    });

    await makeGraphqlAPIRequest(createPeople);
  };

  beforeEach(async () => {
    // Each test mutates state, so wipe + reseed before every one.
    await makeGraphqlAPIRequest(
      destroyManyOperationFactory({
        objectMetadataSingularName: 'person',
        objectMetadataPluralName: 'people',
        gqlFields: 'id',
        filter: { id: { in: ALL_TEST_PERSON_IDS } },
      }),
    );

    await makeGraphqlAPIRequest(
      destroyManyOperationFactory({
        objectMetadataSingularName: 'company',
        objectMetadataPluralName: 'companies',
        gqlFields: 'id',
        filter: { id: { in: Object.values(TEST_COMPANY_IDS) } },
      }),
    );

    await resetFixtures();
  });

  it('should updateMany via a relation traversal filter', async () => {
    const updateOperation = updateManyOperationFactory({
      objectMetadataSingularName: 'person',
      objectMetadataPluralName: 'people',
      gqlFields: 'id jobTitle',
      data: { jobTitle: 'Updated City' },
      filter: {
        and: [
          { id: { in: ALL_TEST_PERSON_IDS } },
          { company: { name: { eq: 'AirbnbMutate' } } },
        ],
      },
    });

    const updateResponse = await makeGraphqlAPIRequest(updateOperation);

    expect(updateResponse.body.errors).toBeUndefined();

    const updatedPeople = updateResponse.body.data.updatePeople;

    expect(updatedPeople).toHaveLength(2);
    expect(
      updatedPeople.map((person: { id: string }) => person.id).sort(),
    ).toEqual(
      [TEST_PERSON_IDS.AIRBNB_ENGINEER, TEST_PERSON_IDS.AIRBNB_DESIGNER].sort(),
    );

    const findOperation = findManyOperationFactory({
      objectMetadataSingularName: 'person',
      objectMetadataPluralName: 'people',
      gqlFields: 'id jobTitle',
      filter: { id: { in: ALL_TEST_PERSON_IDS } },
    });

    const findResponse = await makeGraphqlAPIRequest(findOperation);

    const jobTitleByPersonId = Object.fromEntries(
      findResponse.body.data.people.edges.map(
        (edge: { node: { id: string; jobTitle: string } }) => [
          edge.node.id,
          edge.node.jobTitle,
        ],
      ),
    );

    expect(jobTitleByPersonId[TEST_PERSON_IDS.AIRBNB_ENGINEER]).toEqual(
      'Updated City',
    );
    expect(jobTitleByPersonId[TEST_PERSON_IDS.AIRBNB_DESIGNER]).toEqual(
      'Updated City',
    );

    // Non-Airbnb rows must stay untouched — the JOIN must not widen the
    // mutation past the filter.
    expect(jobTitleByPersonId[TEST_PERSON_IDS.STRIPE_ENGINEER]).toEqual(
      'Original City',
    );
    expect(jobTitleByPersonId[TEST_PERSON_IDS.NOTION_ENGINEER]).toEqual(
      'Original City',
    );
    expect(jobTitleByPersonId[TEST_PERSON_IDS.UNAFFILIATED]).toEqual(
      'Original City',
    );
  });

  it('should deleteMany via a relation traversal filter (soft-delete)', async () => {
    const deleteOperation = deleteManyOperationFactory({
      objectMetadataSingularName: 'person',
      objectMetadataPluralName: 'people',
      gqlFields: 'id deletedAt',
      filter: {
        and: [
          { id: { in: ALL_TEST_PERSON_IDS } },
          { company: { name: { eq: 'NotionMutate' } } },
        ],
      },
    });

    const deleteResponse = await makeGraphqlAPIRequest(deleteOperation);

    expect(deleteResponse.body.errors).toBeUndefined();

    const deletedPeople = deleteResponse.body.data.deletePeople;

    expect(deletedPeople).toHaveLength(1);
    expect(deletedPeople[0].id).toEqual(TEST_PERSON_IDS.NOTION_ENGINEER);
    expect(deletedPeople[0].deletedAt).toBeTruthy();

    const findOperation = findManyOperationFactory({
      objectMetadataSingularName: 'person',
      objectMetadataPluralName: 'people',
      gqlFields: 'id',
      filter: { id: { in: ALL_TEST_PERSON_IDS } },
    });

    const findResponse = await makeGraphqlAPIRequest(findOperation);
    const remainingIds = findResponse.body.data.people.edges.map(
      (edge: { node: { id: string } }) => edge.node.id,
    );

    expect(remainingIds.sort()).toEqual(
      [
        TEST_PERSON_IDS.AIRBNB_ENGINEER,
        TEST_PERSON_IDS.AIRBNB_DESIGNER,
        TEST_PERSON_IDS.STRIPE_ENGINEER,
        TEST_PERSON_IDS.UNAFFILIATED,
      ].sort(),
    );
  });

  it('should restoreMany via a relation traversal filter', async () => {
    // Soft-delete via a scalar filter so restore is the only step exercising
    // the traversal path.
    const deleteOperation = deleteManyOperationFactory({
      objectMetadataSingularName: 'person',
      objectMetadataPluralName: 'people',
      gqlFields: 'id',
      filter: {
        id: {
          in: [
            TEST_PERSON_IDS.AIRBNB_ENGINEER,
            TEST_PERSON_IDS.AIRBNB_DESIGNER,
            TEST_PERSON_IDS.STRIPE_ENGINEER,
          ],
        },
      },
    });

    await makeGraphqlAPIRequest(deleteOperation);

    const restoreOperation = restoreManyOperationFactory({
      objectMetadataSingularName: 'person',
      objectMetadataPluralName: 'people',
      gqlFields: 'id deletedAt',
      filter: {
        and: [
          { id: { in: ALL_TEST_PERSON_IDS } },
          { company: { name: { eq: 'AirbnbMutate' } } },
        ],
      },
    });

    const restoreResponse = await makeGraphqlAPIRequest(restoreOperation);

    expect(restoreResponse.body.errors).toBeUndefined();

    const restoredPeople = restoreResponse.body.data.restorePeople;

    expect(restoredPeople).toHaveLength(2);
    expect(
      restoredPeople.map((person: { id: string }) => person.id).sort(),
    ).toEqual(
      [TEST_PERSON_IDS.AIRBNB_ENGINEER, TEST_PERSON_IDS.AIRBNB_DESIGNER].sort(),
    );
    restoredPeople.forEach((person: { deletedAt: string | null }) => {
      expect(person.deletedAt).toBeNull();
    });

    // The Stripe engineer was soft-deleted too but doesn't match the
    // company filter — must remain deleted.
    const findStripe = findManyOperationFactory({
      objectMetadataSingularName: 'person',
      objectMetadataPluralName: 'people',
      gqlFields: 'id',
      filter: { id: { in: [TEST_PERSON_IDS.STRIPE_ENGINEER] } },
    });

    const findStripeResponse = await makeGraphqlAPIRequest(findStripe);

    expect(findStripeResponse.body.data.people.edges).toEqual([]);
  });

  it('should destroyMany via a relation traversal filter (hard-delete)', async () => {
    const destroyOperation = destroyManyOperationFactory({
      objectMetadataSingularName: 'person',
      objectMetadataPluralName: 'people',
      gqlFields: 'id',
      filter: {
        and: [
          { id: { in: ALL_TEST_PERSON_IDS } },
          { company: { name: { eq: 'StripeMutate' } } },
        ],
      },
    });

    const destroyResponse = await makeGraphqlAPIRequest(destroyOperation);

    expect(destroyResponse.body.errors).toBeUndefined();

    const destroyedPeople = destroyResponse.body.data.destroyPeople;

    expect(destroyedPeople).toHaveLength(1);
    expect(destroyedPeople[0].id).toEqual(TEST_PERSON_IDS.STRIPE_ENGINEER);

    // destroy is a hard-delete — the row must be gone even when querying
    // with a deletedAt filter.
    const findOperation = findManyOperationFactory({
      objectMetadataSingularName: 'person',
      objectMetadataPluralName: 'people',
      gqlFields: 'id',
      filter: {
        id: { in: [TEST_PERSON_IDS.STRIPE_ENGINEER] },
        not: { deletedAt: { is: 'NULL' } },
      },
    });

    const findResponse = await makeGraphqlAPIRequest(findOperation);

    expect(findResponse.body.data.people.edges).toEqual([]);
  });

  it('should keep scalar-only mutations working unchanged', async () => {
    // Pins the no-traversal branch of the mutation builder helper: scalar
    // filters keep emitting a direct WHERE without the IN-subquery wrap.
    const updateOperation = updateManyOperationFactory({
      objectMetadataSingularName: 'person',
      objectMetadataPluralName: 'people',
      gqlFields: 'id jobTitle',
      data: { jobTitle: 'Scalar Updated City' },
      filter: { id: { eq: TEST_PERSON_IDS.UNAFFILIATED } },
    });

    const updateResponse = await makeGraphqlAPIRequest(updateOperation);

    expect(updateResponse.body.errors).toBeUndefined();
    expect(updateResponse.body.data.updatePeople).toHaveLength(1);
    expect(updateResponse.body.data.updatePeople[0].jobTitle).toEqual(
      'Scalar Updated City',
    );
  });
});