datahub/datahub-web-react/src/graphql-mock/fixtures/searchResult/dataJobSearchResult.ts

import { DataJob, SearchResult, SearchResults } from '../../../types.generated';
import { dataJobEntity } from '../entity/dataJobEntity';
import { generateData } from './dataGenerator';

const searchResult = (): SearchResult => {
    return {
        entity: dataJobEntity(),
        matchedFields: [],
        __typename: 'SearchResult',
    };
};

const generateSearchResults = (): SearchResult[] => {
    return generateData<SearchResult>({ generator: searchResult, count: 2 });
};

const searchResults = generateSearchResults();

export const dataJobSearchResult: SearchResults = {
    start: 0,
    count: 0,
    total: 0,
    searchResults,
    facets: [],
    __typename: 'SearchResults',
};

export const findDataJobByURN = (urn: string): DataJob => {
    return searchResults.find((r) => {
        return (r.entity as DataJob).urn === urn;
    })?.entity as DataJob;
};