Expose pod termination fault in JS API

Signed-off-by: Pablo Chacin <[email protected]>

pkg/api/api.go

@@ -128,9 +128,36 @@ func (p *jsProtocolFaultInjector) InjectGrpcFaults(args ...goja.Value) {
 	}
 }
 
+// jsPodFaultInjector implements methods for injecting faults into Pods
+type jsPodFaultInjector struct {
+	ctx context.Context
+	rt  *goja.Runtime
+	disruptors.PodFaultInjector
+}
+
+// TerminatePods is a proxy method. Validates parameters and delegates to the Pod Fault Injector method
+func (p *jsPodFaultInjector) TerminatePods(args ...goja.Value) {
+	if len(args) == 0 {
+		common.Throw(p.rt, fmt.Errorf("PodTermination fault is required"))
+	}
+
+	fault := disruptors.PodTerminationFault{}
+	err := convertValue(p.rt, args[0], &fault)
+	if err != nil {
+		common.Throw(p.rt, fmt.Errorf("invalid fault argument: %w", err))
+	}
+
+	// TODO: return list of pods terminated
+	_, err = p.PodFaultInjector.TerminatePods(p.ctx, fault)
+	if err != nil {
+		common.Throw(p.rt, fmt.Errorf("error injecting fault: %w", err))
+	}
+}
+
 type jsPodDisruptor struct {
 	jsDisruptor
 	jsProtocolFaultInjector
+	jsPodFaultInjector
 }
 
 // buildJsPodDisruptor builds a goja object that implements the PodDisruptor API
@@ -150,6 +177,11 @@ func buildJsPodDisruptor(
 			rt:                    rt,
 			ProtocolFaultInjector: disruptor,
 		},
+		jsPodFaultInjector: jsPodFaultInjector{
+			ctx:              ctx,
+			rt:               rt,
+			PodFaultInjector: disruptor,
+		},
 	}
 
 	return buildObject(rt, d)
@@ -158,6 +190,7 @@ func buildJsPodDisruptor(
 type jsServiceDisruptor struct {
 	jsDisruptor
 	jsProtocolFaultInjector
+	jsPodFaultInjector
 }
 
 // buildJsServiceDisruptor builds a goja object that implements the ServiceDisruptor API
@@ -177,6 +210,11 @@ func buildJsServiceDisruptor(
 			rt:                    rt,
 			ProtocolFaultInjector: disruptor,
 		},
+		jsPodFaultInjector: jsPodFaultInjector{
+			ctx:              ctx,
+			rt:               rt,
+			PodFaultInjector: disruptor,
+		},
 	}
 
 	return buildObject(rt, d)

pkg/api/api_test.go

@@ -428,6 +428,47 @@ func Test_JsPodDisruptor(t *testing.T) {
 			`,
 			expectError: true,
 		},
+		{
+			description: "Terminate Pods (integer count)",
+			script: `
+			const fault = {
+				count: 1
+			}
+
+			d.terminatePods(fault)
+			`,
+			expectError: false,
+		},
+		{
+			description: "Terminate Pods (percentage count)",
+			script: `
+			const fault = {
+				count: '100%'
+			}
+
+			d.terminatePods(fault)
+			`,
+			expectError: false,
+		},
+		{
+			description: "Terminate Pods (invalid percentage count)",
+			script: `
+			const fault = {
+				count: '100'
+			}
+
+			d.terminatePods(fault)
+			`,
+			expectError: true,
+		},
+		{
+			description: "Terminate Pods (missing argument)",
+			script: `
+
+			d.terminatePods()
+			`,
+			expectError: true,
+		},
 	}
 
 	for _, tc := range testCases {