Fixes the test case

Author: eelcocramer

docker/Dockerfile

@@ -1,7 +1,7 @@
 #
 # ----- Go Builder Image ------
 #
-FROM --platform=$BUILDPLATFORM golang:1.21 AS builder
+FROM golang:1.23 AS builder
 
 # curl git bash
 RUN apt-get update && apt-get install -y --no-install-recommends \
@@ -10,12 +10,12 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
         bash \
         tzdata \
         ca-certificates \
-    && rm -rf /var/lib/apt/lists/*
+        && rm -rf /var/lib/apt/lists/*
 
 #
 # ----- Build and Test Image -----
 #
-FROM builder as build-and-test
+FROM builder AS build-and-test
 
 # set working directory
 RUN mkdir -p /go/src/pumba
@@ -60,7 +60,7 @@ RUN --mount=type=cache,target=/root/.cache/go-build make build TARGETOS=${TARGET
 #
 # ------ Pumba Integration Tests ------
 #
-FROM --platform=$TARGETPLATFORM bats/bats:v1.10.0 as integration-tests
+FROM bats/bats:v1.10.0 AS integration-tests
 
 # install required packages
 RUN apk add --no-cache docker iproute2
@@ -89,4 +89,4 @@ COPY --from=builder /usr/share/zoneinfo /usr/share/zoneinfo
 # this is the last command since it's never cached
 COPY --from=build-and-test /go/src/pumba/.bin/github.com/alexei-led/pumba /pumba
 
-ENTRYPOINT ["/pumba"]
+ENTRYPOINT ["/pumba"]

pkg/chaos/iptables/iptables_test.go

@@ -129,8 +129,8 @@ func Test_runIPTables(t *testing.T) {
 			} else {
 				call.Return(nil)
 			}
-			// set StopIPTablesContainer mock call
-			call = mockClient.On("StopIPTablesContainer", context.Background(), tt.args.cmdPrefix, tt.args.cmdSuffix, tt.args.srcIPs, tt.args.dstIPs, tt.args.sports, tt.args.dports, tt.args.iptablesImage, tt.args.pull, tt.args.dryRun)
+			// set StopContainer mock call
+			call = mockClient.On("StopIPTablesContainer", context.Background(), tt.args.container, tt.args.cmdPrefix, tt.args.cmdSuffix, tt.args.srcIPs, tt.args.dstIPs, tt.args.sports, tt.args.dports, tt.args.iptablesImage, tt.args.pull, tt.args.dryRun)
 			if tt.errs.stopErr {
 				call.Return(errors.New("test error"))
 			} else {