[ML] Update tests to have getCreateTransformSettingsRequestBody

qn895 · qn895 · commit b82513ee3b25 · 2020-09-30T16:20:36.000-05:00
diff --git a/x-pack/plugins/transform/public/app/common/request.test.ts b/x-pack/plugins/transform/public/app/common/request.test.ts
@@ -17,6 +17,7 @@ import {
   defaultQuery,
   getPreviewTransformRequestBody,
   getCreateTransformRequestBody,
+  getCreateTransformSettingsRequestBody,
   getPivotQuery,
   isDefaultQuery,
   isMatchAllQuery,
@@ -159,6 +160,7 @@ describe('Transform: Common', () => {
       transformDescription: 'the-transform-description',
       transformFrequency: '1m',
       transformSettingsMaxPageSearchSize: 100,
+      transformSettingsDocsPerSecond: 400,
       destinationIndex: 'the-destination-index',
       touched: true,
       valid: true,
@@ -180,11 +182,42 @@ describe('Transform: Common', () => {
       },
       settings: {
         max_page_search_size: 100,
+        docs_per_second: 400,
       },
       source: {
         index: ['the-index-pattern-title'],
         query: { query_string: { default_operator: 'AND', query: 'the-search-query' } },
       },
     });
   });
+
+  test('getCreateTransformSettingsRequestBody() with all settings', () => {
+    const transformDetailsState: StepDetailsExposedState = {
+      transformSettingsDocsPerSecond: 400,
+      transformSettingsMaxPageSearchSize: 100,
+    };
+
+    const request = getCreateTransformSettingsRequestBody(transformDetailsState);
+
+    expect(request).toEqual({
+      settings: {
+        docs_per_second: 400,
+        max_page_search_size: 100,
+      },
+    });
+  });
+
+  test('getCreateTransformSettingsRequestBody() with individual settings', () => {
+    const transformDetailsState: StepDetailsExposedState = {
+      transformSettingsDocsPerSecond: 400,
+    };
+
+    const request = getCreateTransformSettingsRequestBody(transformDetailsState);
+
+    expect(request).toEqual({
+      settings: {
+        docs_per_second: 400,
+      },
+    });
+  });
 });
