Update scenario.go

Author: stephanos

dockerfiles/cli.Dockerfile

@@ -18,7 +18,6 @@ COPY cmd ./cmd
 COPY loadgen ./loadgen
 COPY scenarios ./scenarios
 COPY workers ./workers/
-COPY common ./common
 COPY go.mod go.sum ./
 
 # Install Antithesis SDK and instrumentor
@@ -36,22 +35,20 @@ COPY --from=source /go/pkg/mod /go/pkg/mod
 WORKDIR /omes_source
 
 # Instrument OMES with Antithesis
-RUN echo "🔧 Instrumenting OMES..." && \
+RUN echo "🔧 Instrumenting..." && \
     mkdir /omes_transformed && \
     antithesis-go-instrumentor /omes_source /omes_transformed && \
-    cp -r /omes_source/.git /omes_transformed/ 2>/dev/null || true && \
-    mv /omes_transformed /instrumented && \
+    cp -r /omes_source/.git /omes_transformed/customer/ 2>/dev/null || true && \
+    cp /omes_source/go.mod /omes_source/go.sum /omes_transformed/customer/ && \
     mkdir -p /notifier /symbols && \
-    mv /instrumented/notifier.json /notifier/ 2>/dev/null || true && \
-    mv /instrumented/symbols.json /symbols/ 2>/dev/null || true
+    cp /omes_transformed/notifier.json /notifier/ 2>/dev/null || true && \
+    cp /omes_transformed/symbols.json /symbols/ 2>/dev/null || true
 
 # Build stage: compile the instrumented code
 FROM --platform=linux/$TARGETARCH golang:1.25 AS build
 
 ARG TARGETARCH
 
-WORKDIR /app
-
 # Install protobuf compiler and git
 RUN apt-get update \
     && DEBIAN_FRONTEND=noninteractive \
@@ -72,8 +69,11 @@ RUN wget -q -O - https://sh.rustup.rs | sh -s -- -y \
     fi
 ENV PATH="$PATH:/root/.cargo/bin"
 
-# Copy instrumented source code
-COPY --from=instrumented /instrumented /app
+# Copy entire instrumented structure (includes customer code + notifier package)
+COPY --from=instrumented /omes_transformed /omes_transformed
+
+# Set working directory to the customer code
+WORKDIR /omes_transformed/customer
 
 # Build the CLI
 RUN CGO_ENABLED=0 go build -o temporal-omes ./cmd

go.mod

@@ -31,6 +31,7 @@ require (
 )
 
 require (
+	github.com/antithesishq/antithesis-sdk-go v0.5.1-0.20250924165633-f60b0222f1b6 // indirect
 	github.com/davecgh/go-spew v1.1.1 // indirect
 	github.com/facebookgo/clock v0.0.0-20150410010913-600d898af40a // indirect
 	github.com/golang/mock v1.6.0 // indirect

go.sum

@@ -1,3 +1,5 @@
+github.com/antithesishq/antithesis-sdk-go v0.5.1-0.20250924165633-f60b0222f1b6 h1:qSD74Vz3scN2SrfML8dy2Whcv0C3pNkfqYZXeL4SIq0=
+github.com/antithesishq/antithesis-sdk-go v0.5.1-0.20250924165633-f60b0222f1b6/go.mod h1:IUpT2DPAKh6i/YhSbt6Gl3v2yvUZjmKncl7U91fup7E=
 github.com/beorn7/perks v1.0.1 h1:VlbKKnNfV8bJzeqoa4cOKqO6bYr3WgKZxO8Z16+hsOM=
 github.com/beorn7/perks v1.0.1/go.mod h1:G2ZrVWU2WbWT9wwq4/hrbKbnv/1ERSJQ0ibhJ6rlkpw=
 github.com/cespare/xxhash/v2 v2.3.0 h1:UL815xU9SqsFlibzuggzjXhog7bL6oX9BbNZnL2UFvs=

loadgen/scenario.go

@@ -18,7 +18,7 @@ import (
 	"go.temporal.io/sdk/temporal"
 	"go.uber.org/zap"
 
-	"github.com/temporalio/omes/common/softassert"
+	"github.com/antithesishq/antithesis-sdk-go/assert"
 	"github.com/temporalio/omes/loadgen/kitchensink"
 )
 
@@ -434,8 +434,7 @@ func (r *Run) ExecuteAnyWorkflow(ctx context.Context, options client.StartWorkfl
 	}
 	if err := execution.Get(ctx, valuePtr); err != nil {
 		// Antithesis assertion: Workflow execution should never fail
-		softassert.Always(
-			false,
+		assert.Unreachable(
 			"Workflow execution should never return an error",
 			map[string]any{
 				"workflow_id": execution.GetID(),

workers/go/go.mod

@@ -13,6 +13,7 @@ require (
 )
 
 require (
+	github.com/antithesishq/antithesis-sdk-go v0.5.1-0.20250924165633-f60b0222f1b6 // indirect
 	github.com/beorn7/perks v1.0.1 // indirect
 	github.com/cespare/xxhash/v2 v2.3.0 // indirect
 	github.com/davecgh/go-spew v1.1.1 // indirect

workers/go/go.sum

@@ -1,3 +1,5 @@
+github.com/antithesishq/antithesis-sdk-go v0.5.1-0.20250924165633-f60b0222f1b6 h1:qSD74Vz3scN2SrfML8dy2Whcv0C3pNkfqYZXeL4SIq0=
+github.com/antithesishq/antithesis-sdk-go v0.5.1-0.20250924165633-f60b0222f1b6/go.mod h1:IUpT2DPAKh6i/YhSbt6Gl3v2yvUZjmKncl7U91fup7E=
 github.com/beorn7/perks v1.0.1 h1:VlbKKnNfV8bJzeqoa4cOKqO6bYr3WgKZxO8Z16+hsOM=
 github.com/beorn7/perks v1.0.1/go.mod h1:G2ZrVWU2WbWT9wwq4/hrbKbnv/1ERSJQ0ibhJ6rlkpw=
 github.com/cespare/xxhash/v2 v2.3.0 h1:UL815xU9SqsFlibzuggzjXhog7bL6oX9BbNZnL2UFvs=