Fix tests

elastic · Dec 6, 2023 · 3f985ac · 3f985ac
1 parent 0b90335
commit 3f985ac
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/x-pack/plugins/fleet/common/services/output_helpers.test.ts b/x-pack/plugins/fleet/common/services/output_helpers.test.ts
@@ -5,6 +5,8 @@
  * 2.0.
  */
 
+import { safeLoad } from 'js-yaml';
+
 import {
   getAllowedOutputTypeForPolicy,
   outputYmlIncludesReservedPerformanceKey,
@@ -54,13 +56,13 @@ describe('outputYmlIncludesReservedPerformanceKey', () => {
     it('returns true when reserved key is present', () => {
       const configYml = `queue.mem.events: 1000`;
 
-      expect(outputYmlIncludesReservedPerformanceKey(configYml)).toBe(true);
+      expect(outputYmlIncludesReservedPerformanceKey(configYml, safeLoad)).toBe(true);
     });
 
     it('returns false when no reserved key is present', () => {
       const configYml = `some.random.key: 1000`;
 
-      expect(outputYmlIncludesReservedPerformanceKey(configYml)).toBe(false);
+      expect(outputYmlIncludesReservedPerformanceKey(configYml, safeLoad)).toBe(false);
     });
   });
 
@@ -72,7 +74,7 @@ describe('outputYmlIncludesReservedPerformanceKey', () => {
             events: 1000
     `;
 
-      expect(outputYmlIncludesReservedPerformanceKey(configYml)).toBe(true);
+      expect(outputYmlIncludesReservedPerformanceKey(configYml, safeLoad)).toBe(true);
     });
 
     it('returns false when no reserved key is present', () => {
@@ -82,21 +84,21 @@ describe('outputYmlIncludesReservedPerformanceKey', () => {
             key: 1000
       `;
 
-      expect(outputYmlIncludesReservedPerformanceKey(configYml)).toBe(false);
+      expect(outputYmlIncludesReservedPerformanceKey(configYml, safeLoad)).toBe(false);
     });
   });
 
   describe('plain string', () => {
     it('returns true when reserved key is present', () => {
       const configYml = `bulk_max_size`;
 
-      expect(outputYmlIncludesReservedPerformanceKey(configYml)).toBe(true);
+      expect(outputYmlIncludesReservedPerformanceKey(configYml, safeLoad)).toBe(true);
     });
 
     it('returns false when no reserved key is present', () => {
       const configYml = `just a string`;
 
-      expect(outputYmlIncludesReservedPerformanceKey(configYml)).toBe(false);
+      expect(outputYmlIncludesReservedPerformanceKey(configYml, safeLoad)).toBe(false);
     });
   });
 });