Skip to content

Commit

Permalink
chore: improve mock patch file
Browse files Browse the repository at this point in the history
Replaces the old mock.patch file with a new one that ensures people can
also use ENVIRONMENT variables directly. See
livepeer#116.
  • Loading branch information
rickstaa authored and eliteprox committed Jul 26, 2024
1 parent a8b18e2 commit 5b9150c
Showing 1 changed file with 24 additions and 0 deletions.
24 changes: 24 additions & 0 deletions runner/dev/patches/mock.patch
Original file line number Diff line number Diff line change
Expand Up @@ -103,3 +103,27 @@ index bd2e4f2..830d27c 100644
seed = kwargs.pop("seed", None)
safety_check = kwargs.pop("safety_check", True)

diff --git a/worker/docker.go b/worker/docker.go
index 48131a7..c822efb 100644
--- a/worker/docker.go
+++ b/worker/docker.go
@@ -4,6 +4,7 @@ import (
"context"
"errors"
"log/slog"
+ "os"
"strings"
"sync"
"time"
@@ -168,6 +169,11 @@ func (m *DockerManager) createContainer(ctx context.Context, pipeline string, mo
envVars = append(envVars, key+"="+value.String())
}

+ if value, ok := os.LookupEnv("MOCK_PIPELINE"); ok {
+ envVars = append(envVars, "MOCK_PIPELINE="+value)
+ slog.Info("MOCK_PIPELINE set to " + value + ", passing to runner container")
+ }
+
containerConfig := &container.Config{
Image: m.containerImageID,
Env: envVars,

0 comments on commit 5b9150c

Please sign in to comment.