Merge pull request #45 from go-tstr/dep-improvements

Dep improvements

Author: heppu

dep/cmd/cmd.go

@@ -96,6 +96,19 @@ func WithCommand(name string, args ...string) Opt {
 	}
 }
 
+// WithCommandFn creates a new command using the given function.
+// This is useful for lazy loading of the command and using arguments from other dependencies.
+func WithCommandFn(fn func() (*exec.Cmd, error)) Opt {
+	return func(c *Cmd) error {
+		cmd, err := fn()
+		if err != nil {
+			return err
+		}
+		c.cmd = cmd
+		return nil
+	}
+}
+
 // WithReadyFn allows user to provide custom readiness function.
 // Given fn should block until the command is ready.
 func WithReadyFn(fn func(*exec.Cmd) error) Opt {

dep/container/container.go

@@ -1,3 +1,4 @@
+// Package container provides a wrapper around the testcontainers-go library to simplify container management in tests.
 package container
 
 import (
@@ -45,13 +46,20 @@ func (c *Container) Stop() error {
 	return testcontainers.TerminateContainer(c.c)
 }
 
+// Container returns the underlying testcontainers.Container.
+func (c *Container) Container() testcontainers.Container {
+	return c.c
+}
+
+// WithReadyFn sets a custom readiness function which should block until ready.
 func WithReadyFn(fn func(testcontainers.Container) error) Opt {
 	return func(c *Container) error {
 		c.ready = fn
 		return nil
 	}
 }
 
+// WithModule creates a container using the testcontainers-go modules.
 func WithModule[T testcontainers.Container](
 	runFn func(ctx context.Context, img string, opts ...testcontainers.ContainerCustomizer) (T, error),
 	img string,
@@ -67,6 +75,7 @@ func WithModule[T testcontainers.Container](
 	}
 }
 
+// WithGenericContainer creates a container using the testcontainers.GenericContainer function.
 func WithGenericContainer(req testcontainers.GenericContainerRequest) Opt {
 	return func(c *Container) (err error) {
 		c.c, err = testcontainers.GenericContainer(context.Background(), req)