vendure/packages/elasticsearch-plugin/e2e/elasticsearch-plugin-uuid.e2e-spec.ts

import { DefaultJobQueuePlugin, mergeConfig, UuidIdStrategy } from '@vendure/core';
import { createTestEnvironment } from '@vendure/testing';
import gql from 'graphql-tag';
import path from 'path';
import { afterAll, beforeAll, describe, expect, it } from 'vitest';

import { initialData } from '../../../e2e-common/e2e-initial-data';
import { TEST_SETUP_TIMEOUT_MS, testConfig } from '../../../e2e-common/test-config';
import {
    SearchProductsShopQuery,
    SearchProductsShopQueryVariables,
} from '../../core/e2e/graphql/generated-e2e-shop-types';
import { SEARCH_PRODUCTS_SHOP } from '../../core/e2e/graphql/shop-definitions';
import { awaitRunningJobs } from '../../core/e2e/utils/await-running-jobs';
import { ElasticsearchPlugin } from '../src/plugin';

import { GetCollectionListQuery } from './graphql/generated-e2e-elasticsearch-plugin-types';

// eslint-disable-next-line @typescript-eslint/no-var-requires
const { elasticsearchHost, elasticsearchPort } = require('./constants');

// https://github.com/vendurehq/vendure/issues/494
describe('Elasticsearch plugin with UuidIdStrategy', () => {
    const { server, adminClient, shopClient } = createTestEnvironment(
        mergeConfig(testConfig(), {
            entityOptions: { entityIdStrategy: new UuidIdStrategy() },
            // logger: new DefaultLogger({ level: LogLevel.Info }),
            plugins: [
                ElasticsearchPlugin.init({
                    indexPrefix: 'e2e-uuid-tests',
                    port: elasticsearchPort,
                    host: elasticsearchHost,
                }),
                DefaultJobQueuePlugin,
            ],
        }),
    );

    beforeAll(async () => {
        await server.init({
            initialData,
            productsCsvPath: path.join(__dirname, 'fixtures/e2e-products-full.csv'),
            customerCount: 1,
        });
        await adminClient.asSuperAdmin();
        // We have extra time here because a lot of jobs are
        // triggered from all the product updates
        await awaitRunningJobs(adminClient, 10_000, 1000);
        await adminClient.query(REINDEX);
        await awaitRunningJobs(adminClient);
    }, TEST_SETUP_TIMEOUT_MS);

    afterAll(async () => {
        await server.destroy();
    });

    it('no term or filters', async () => {
        const { search } = await shopClient.query<SearchProductsShopQuery, SearchProductsShopQueryVariables>(
            SEARCH_PRODUCTS_SHOP,
            {
                input: {
                    groupByProduct: true,
                },
            },
        );
        expect(search.totalItems).toBe(21);
    });

    it('no term or filters grouped by SKU', async () => {
        const { search } = await shopClient.query<SearchProductsShopQuery, SearchProductsShopQueryVariables>(
            SEARCH_PRODUCTS_SHOP,
            {
                input: {
                    groupBySKU: true,
                },
            },
        );
        expect(search.totalItems).toBe(34);
    });

    it('with search term', async () => {
        const { search } = await shopClient.query<SearchProductsShopQuery, SearchProductsShopQueryVariables>(
            SEARCH_PRODUCTS_SHOP,
            {
                input: {
                    groupByProduct: true,
                    term: 'laptop',
                },
            },
        );
        expect(search.totalItems).toBe(1);
    });

    it('with search term grouped by SKU', async () => {
        const { search } = await shopClient.query<SearchProductsShopQuery, SearchProductsShopQueryVariables>(
            SEARCH_PRODUCTS_SHOP,
            {
                input: {
                    groupBySKU: true,
                    term: 'bonsai',
                },
            },
        );
        expect(search.totalItems).toBe(1);
    });

    it('with collectionId filter term', async () => {
        const { collections } = await shopClient.query<GetCollectionListQuery>(GET_COLLECTION_LIST);
        const { search } = await shopClient.query<SearchProductsShopQuery, SearchProductsShopQueryVariables>(
            SEARCH_PRODUCTS_SHOP,
            {
                input: {
                    groupByProduct: true,
                    collectionId: collections.items[0].id,
                },
            },
        );
        expect(search.items).not.toEqual([]);
    });
});

const REINDEX = gql`
    mutation Reindex {
        reindex {
            id
            queueName
            state
            progress
            duration
            result
        }
    }
`;

const GET_COLLECTION_LIST = gql`
    query GetCollectionList {
        collections {
            items {
                id
                name
            }
        }
    }
`;