diff --git a/docker/test_env/postgres.go b/docker/test_env/postgres.go index 73a6a636e..dd1249a95 100644 --- 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 != "" {