mirror of
https://github.com/elastic/kibana.git
synced 2025-04-24 01:38:56 -04:00
4b7d18b5c3
## Summary Part of https://github.com/elastic/kibana/issues/157095. Uses the new versioned router capabilities for the bfetch plugin. ### Checklist - [ ] [Documentation](https://www.elastic.co/guide/en/kibana/master/development-documentation.html) was added for features that require explanation or tutorials - [ ] [Unit or functional tests](https://www.elastic.co/guide/en/kibana/master/development-tests.html) were updated or added to match the most common scenarios --------- Co-authored-by: kibanamachine <42973632+kibanamachine@users.noreply.github.com>
109 lines
3.3 KiB
TypeScript
109 lines
3.3 KiB
TypeScript
/*
|
|
* 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 and the Server Side Public License, v 1; you may not use this file except
|
|
* in compliance with, at your election, the Elastic License 2.0 or the Server
|
|
* Side Public License, v 1.
|
|
*/
|
|
|
|
import expect from '@kbn/expect';
|
|
import request from 'superagent';
|
|
import type SuperTest from 'supertest';
|
|
import { IEsSearchResponse } from '@kbn/data-plugin/common';
|
|
import { ELASTIC_HTTP_VERSION_HEADER } from '@kbn/core-http-common';
|
|
import { BFETCH_ROUTE_VERSION_LATEST } from '@kbn/bfetch-plugin/common';
|
|
import { FtrService } from '../ftr_provider_context';
|
|
|
|
/**
|
|
* Function copied from here:
|
|
* test/api_integration/apis/search/bsearch.ts without the compress
|
|
*
|
|
* Splits the JSON lines from bsearch
|
|
*/
|
|
const parseBfetchResponse = (resp: request.Response): Array<Record<string, any>> => {
|
|
return resp.text
|
|
.trim()
|
|
.split('\n')
|
|
.map((item) => JSON.parse(item));
|
|
};
|
|
|
|
/**
|
|
* Function copied from here:
|
|
* x-pack/test/rule_registry/common/lib/authentication/spaces.ts
|
|
* @param spaceId The space id we want to utilize
|
|
*/
|
|
const getSpaceUrlPrefix = (spaceId?: string): string => {
|
|
return spaceId && spaceId !== 'default' ? `/s/${spaceId}` : ``;
|
|
};
|
|
|
|
/**
|
|
* Options for the send method
|
|
*/
|
|
interface SendOptions {
|
|
supertest: SuperTest.SuperTest<SuperTest.Test>;
|
|
options: object;
|
|
strategy: string;
|
|
space?: string;
|
|
}
|
|
|
|
/**
|
|
* Bsearch Service that can reduce flake on the CI systems when they are under
|
|
* pressure and bsearch returns an async search response or a sync response.
|
|
*
|
|
* @example
|
|
* const supertest = getService('supertest');
|
|
* const bsearch = getService('bsearch');
|
|
* const response = await bsearch.send<MyType>({
|
|
* supertest,
|
|
* options: {
|
|
* defaultIndex: ['large_volume_dns_data'],
|
|
* },
|
|
* strategy: 'securitySolutionSearchStrategy',
|
|
* });
|
|
* expect(response).eql({ ... your value ... });
|
|
*/
|
|
export class BsearchService extends FtrService {
|
|
private readonly retry = this.ctx.getService('retry');
|
|
|
|
/** Send method to send in your supertest, url, options, and strategy name */
|
|
async send<T extends IEsSearchResponse>({ supertest, options, strategy, space }: SendOptions) {
|
|
const spaceUrl = getSpaceUrlPrefix(space);
|
|
const { body } = await this.retry.try(async () => {
|
|
return supertest
|
|
.post(`${spaceUrl}/internal/search/${strategy}`)
|
|
.set(ELASTIC_HTTP_VERSION_HEADER, '1')
|
|
.set('kbn-xsrf', 'true')
|
|
.send(options)
|
|
.expect(200);
|
|
});
|
|
|
|
if (!body.isRunning) {
|
|
return body;
|
|
}
|
|
|
|
const result = await this.retry.try(async () => {
|
|
const resp = await supertest
|
|
.post(`${spaceUrl}/internal/bsearch`)
|
|
.set('kbn-xsrf', 'true')
|
|
.set(ELASTIC_HTTP_VERSION_HEADER, BFETCH_ROUTE_VERSION_LATEST)
|
|
.send({
|
|
batch: [
|
|
{
|
|
request: {
|
|
id: body.id,
|
|
...options,
|
|
},
|
|
options: {
|
|
strategy,
|
|
},
|
|
},
|
|
],
|
|
})
|
|
.expect(200);
|
|
const [parsedResponse] = parseBfetchResponse(resp);
|
|
expect(parsedResponse.result.isRunning).equal(false);
|
|
return parsedResponse.result as T;
|
|
});
|
|
return result;
|
|
}
|
|
}
|