datahub/datahub-web-react/src/graphql-mock/resolver/updateDataJobResolver.ts

import { DataJob, DataJobUpdateInput } from '../../types.generated';
import { findDataJobByURN } from '../fixtures/searchResult/dataJobSearchResult';
import { updateEntityOwners, updateEntityTag } from '../mutationHelper';

type UpdateDataJob = {
    data: { updateDataJob: DataJob };
};

export const updateDataJobResolver = {
    updateDataJob({ variables: { urn, input } }): UpdateDataJob {
        const { ownership, globalTags }: DataJobUpdateInput = input;
        const dataJob = findDataJobByURN(urn);

        if (ownership) {
            updateEntityOwners({ entity: dataJob, owners: ownership?.owners });
        } else if (globalTags) {
            updateEntityTag({ entity: dataJob, globalTags });
        }

        return {
            data: {
                updateDataJob: dataJob,
            },
        };
    },
};