Adds a few more tests around allowed 'type's and 'value's

elastic · Jun 12, 2023 · 302a153 · 302a153
1 parent 2448f1c
commit 302a153
Showing 1 changed file with 68 additions and 0 deletions.
diff --git a/x-pack/plugins/security_solution/common/risk_engine/risk_weights/schema.test.ts b/x-pack/plugins/security_solution/common/risk_engine/risk_weights/schema.test.ts
@@ -15,6 +15,44 @@ import { RiskCategories, RiskWeightTypes } from './types';
 describe('risk weight schema', () => {
   let type: string;
 
+  describe('allowed types', () => {
+    it('allows the global weight type', () => {
+      const payload = {
+        type: RiskWeightTypes.global,
+        host: 0.1,
+      };
+      const decoded = riskWeightSchema.decode(payload);
+      const message = pipe(decoded, foldLeftRight);
+
+      expect(getPaths(left(message.errors))).toEqual([]);
+      expect(message.schema).toEqual(payload);
+    });
+
+    it('allows the risk category weight type', () => {
+      const payload = {
+        type: RiskWeightTypes.global,
+        host: 0.1,
+      };
+      const decoded = riskWeightSchema.decode(payload);
+      const message = pipe(decoded, foldLeftRight);
+
+      expect(getPaths(left(message.errors))).toEqual([]);
+      expect(message.schema).toEqual(payload);
+    });
+
+    it('rejects an unknown weight type', () => {
+      const payload = {
+        type: 'unknown',
+        host: 0.1,
+      };
+      const decoded = riskWeightSchema.decode(payload);
+      const message = pipe(decoded, foldLeftRight);
+
+      expect(getPaths(left(message.errors)).length).toBeGreaterThan(0);
+      expect(message.schema).toEqual({});
+    });
+  });
+
   describe('conditional fields', () => {
     describe('global weights', () => {
       beforeEach(() => {
@@ -161,6 +199,36 @@ describe('risk weight schema', () => {
         expect(getPaths(left(message.errors))).toEqual([]);
         expect(message.schema).toEqual({ type, value: RiskCategories.alerts, host: 0.1 });
       });
+
+      describe('allowed category values', () => {
+        it('allows the alerts type for a category', () => {
+          const payload = {
+            type,
+            value: RiskCategories.alerts,
+            host: 0.1,
+          };
+          const decoded = riskWeightSchema.decode(payload);
+          const message = pipe(decoded, foldLeftRight);
+
+          expect(getPaths(left(message.errors))).toEqual([]);
+          expect(message.schema).toEqual(payload);
+        });
+
+        it('rejects an unknown category value', () => {
+          const payload = {
+            type,
+            value: 'unknown',
+            host: 0.1,
+          };
+          const decoded = riskWeightSchema.decode(payload);
+          const message = pipe(decoded, foldLeftRight);
+
+          expect(getPaths(left(message.errors))).toContain(
+            'Invalid value "unknown" supplied to "value"'
+          );
+          expect(message.schema).toEqual({});
+        });
+      });
     });
   });
 });