[TT-1259] postgresdb logstrem functional option (#980)

smartcontractkit · Jun 7, 2024 · 71ff056 · 71ff056
1 parent 404d898
commit 71ff056
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/docker/test_env/postgres.go b/docker/test_env/postgres.go
@@ -18,6 +18,7 @@ import (
 
 	"github.com/smartcontractkit/chainlink-testing-framework/docker"
 	"github.com/smartcontractkit/chainlink-testing-framework/logging"
+	"github.com/smartcontractkit/chainlink-testing-framework/logstream"
 	"github.com/smartcontractkit/chainlink-testing-framework/mirror"
 	"github.com/smartcontractkit/chainlink-testing-framework/utils/testcontext"
 )
@@ -56,6 +57,12 @@ func WithPostgresImageName(imageName string) PostgresDbOption {
 	}
 }
 
+func WithPostgresDbLogStream(ls *logstream.LogStream) PostgresDbOption {
+	return func(c *PostgresDb) {
+		c.LogStream = ls
+	}
+}
+
 func WithPostgresImageVersion(version string) PostgresDbOption {
 	return func(c *PostgresDb) {
 		if version != "" {