kibana/x-pack/test/upgrade_assistant_integration/upgrade_assistant/reindexing.js

/*
 * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
 * or more contributor license agreements. Licensed under the Elastic License
 * 2.0; you may not use this file except in compliance with the Elastic License
 * 2.0.
 */

import expect from '@kbn/expect';

import { ReindexStatus, REINDEX_OP_TYPE } from '../../../plugins/upgrade_assistant/common/types';
import { generateNewIndexName } from '../../../plugins/upgrade_assistant/server/lib/reindexing/index_settings';
import { getIndexState } from '../../../plugins/upgrade_assistant/common/get_index_state';

export default function ({ getService }) {
  const supertest = getService('supertest');
  const esArchiver = getService('esArchiver');
  const es = getService('es');

  // Utility function that keeps polling API until reindex operation has completed or failed.
  const waitForReindexToComplete = async (indexName) => {
    console.log(`Waiting for reindex to complete...`);
    let lastState;

    while (true) {
      lastState = (await supertest.get(`/api/upgrade_assistant/reindex/${indexName}`).expect(200))
        .body.reindexOp;
      // Once the operation is completed or failed and unlocked, stop polling.
      if (lastState.status !== ReindexStatus.inProgress && lastState.locked === null) {
        break;
      }
      await new Promise((resolve) => setTimeout(resolve, 1000));
    }

    return lastState;
  };

  describe('reindexing', () => {
    afterEach(() => {
      // Cleanup saved objects
      return es.deleteByQuery({
        index: '.kibana',
        refresh: true,
        body: {
          query: {
            simple_query_string: {
              query: REINDEX_OP_TYPE,
              fields: ['type'],
            },
          },
        },
      });
    });

    it('should create a new index with the same documents', async () => {
      await esArchiver.load('x-pack/test/functional/es_archives/upgrade_assistant/reindex');
      const { body } = await supertest
        .post(`/api/upgrade_assistant/reindex/dummydata`)
        .set('kbn-xsrf', 'xxx')
        .expect(200);

      expect(body.indexName).to.equal('dummydata');
      expect(body.status).to.equal(ReindexStatus.inProgress);

      const lastState = await waitForReindexToComplete('dummydata');
      expect(lastState.errorMessage).to.equal(null);
      expect(lastState.status).to.equal(ReindexStatus.completed);

      const { newIndexName } = lastState;
      const { body: indexSummary } = await es.indices.get({ index: 'dummydata' });

      // The new index was created
      expect(indexSummary[newIndexName]).to.be.an('object');
      // The original index name is aliased to the new one
      expect(indexSummary[newIndexName].aliases.dummydata).to.be.an('object');
      // Verify mappings exist on new index
      expect(indexSummary[newIndexName].mappings.properties).to.be.an('object');
      // The number of documents in the new index matches what we expect
      expect((await es.count({ index: lastState.newIndexName })).body.count).to.be(3);

      // Cleanup newly created index
      await es.indices.delete({
        index: lastState.newIndexName,
      });
    });

    it('should update any aliases', async () => {
      await esArchiver.load('x-pack/test/functional/es_archives/upgrade_assistant/reindex');

      // Add aliases and ensure each returns the right number of docs
      await es.indices.updateAliases({
        body: {
          actions: [
            { add: { index: 'dummydata', alias: 'myAlias' } },
            { add: { index: 'dummy*', alias: 'wildcardAlias' } },
            {
              add: { index: 'dummydata', alias: 'myHttpsAlias', filter: { term: { https: true } } },
            },
          ],
        },
      });
      expect((await es.count({ index: 'myAlias' })).body.count).to.be(3);
      expect((await es.count({ index: 'wildcardAlias' })).body.count).to.be(3);
      expect((await es.count({ index: 'myHttpsAlias' })).body.count).to.be(2);

      // Reindex
      await supertest
        .post(`/api/upgrade_assistant/reindex/dummydata`)
        .set('kbn-xsrf', 'xxx')
        .expect(200);
      const lastState = await waitForReindexToComplete('dummydata');

      // The regular aliases should still return 3 docs
      expect((await es.count({ index: 'myAlias' })).body.count).to.be(3);
      expect((await es.count({ index: 'wildcardAlias' })).body.count).to.be(3);
      // The filtered alias should still return 2 docs
      expect((await es.count({ index: 'myHttpsAlias' })).body.count).to.be(2);

      // Cleanup newly created index
      await es.indices.delete({
        index: lastState.newIndexName,
      });
    });

    it('shows no warnings', async () => {
      const resp = await supertest.get(`/api/upgrade_assistant/reindex/7.0-data`);
      expect(resp.body.warnings.length).to.be(0);
    });

    it('reindexes old 7.0 index', async () => {
      const { body } = await supertest
        .post(`/api/upgrade_assistant/reindex/7.0-data`)
        .set('kbn-xsrf', 'xxx')
        .expect(200);

      expect(body.indexName).to.equal('7.0-data');
      expect(body.status).to.equal(ReindexStatus.inProgress);

      const lastState = await waitForReindexToComplete('7.0-data');
      expect(lastState.errorMessage).to.equal(null);
      expect(lastState.status).to.equal(ReindexStatus.completed);
    });

    it('should reindex a batch in order and report queue state', async () => {
      const assertQueueState = async (firstInQueueIndexName, queueLength) => {
        const response = await supertest
          .get(`/api/upgrade_assistant/reindex/batch/queue`)
          .set('kbn-xsrf', 'xxx')
          .expect(200);

        const { queue } = response.body;

        const [firstInQueue] = queue;

        if (!firstInQueueIndexName) {
          expect(firstInQueueIndexName).to.be(undefined);
        } else {
          expect(firstInQueue.indexName).to.be(firstInQueueIndexName);
        }

        expect(queue.length).to.be(queueLength);
      };

      const test1 = 'batch-reindex-test1';
      const test2 = 'batch-reindex-test2';
      const test3 = 'batch-reindex-test3';

      const cleanupReindex = async (indexName) => {
        try {
          await es.indices.delete({ index: generateNewIndexName(indexName) });
        } catch (e) {
          try {
            await es.indices.delete({ index: indexName });
          } catch (e) {
            // Ignore
          }
        }
      };

      try {
        // Set up indices for the batch
        await es.indices.create({ index: test1 });
        await es.indices.create({ index: test2 });
        await es.indices.create({ index: test3 });

        await es.indices.close({ index: test1 });

        const result = await supertest
          .post(`/api/upgrade_assistant/reindex/batch`)
          .set('kbn-xsrf', 'xxx')
          .send({ indexNames: [test1, test2, test3] })
          .expect(200);

        expect(result.body.enqueued.length).to.equal(3);
        expect(result.body.errors.length).to.equal(0);

        const [{ newIndexName: nameOfIndexThatShouldBeClosed }] = result.body.enqueued;

        await assertQueueState(test1, 3);
        await waitForReindexToComplete(test1);

        await assertQueueState(test2, 2);
        await waitForReindexToComplete(test2);

        await assertQueueState(test3, 1);
        await waitForReindexToComplete(test3);

        await assertQueueState(undefined, 0);

        // Check that the closed index is still closed after reindexing
        const { body: resolvedIndices } = await es.indices.resolveIndex({
          name: nameOfIndexThatShouldBeClosed,
        });

        const test1ReindexedState = getIndexState(nameOfIndexThatShouldBeClosed, resolvedIndices);
        expect(test1ReindexedState).to.be('closed');
      } finally {
        await cleanupReindex(test1);
        await cleanupReindex(test2);
        await cleanupReindex(test3);
      }
    });
  });
}