diff --git a/modules/pinecone/pinecone.go b/modules/pinecone/pinecone.go index d4a26b5245..62c4b63f66 100644 --- a/modules/pinecone/pinecone.go +++ b/modules/pinecone/pinecone.go @@ -14,30 +14,20 @@ type Container struct { // Run creates an instance of the Pinecone container type func Run(ctx context.Context, img string, opts ...testcontainers.ContainerCustomizer) (*Container, error) { - req := testcontainers.ContainerRequest{ - Image: img, - ExposedPorts: []string{"5080/tcp"}, + moduleOpts := []testcontainers.ContainerCustomizer{ + testcontainers.WithExposedPorts("5080/tcp"), } - genericContainerReq := testcontainers.GenericContainerRequest{ - ContainerRequest: req, - Started: true, - } - - for _, opt := range opts { - if err := opt.Customize(&genericContainerReq); err != nil { - return nil, fmt.Errorf("customize: %w", err) - } - } + moduleOpts = append(moduleOpts, opts...) - container, err := testcontainers.GenericContainer(ctx, genericContainerReq) + ctr, err := testcontainers.Run(ctx, img, moduleOpts...) var c *Container - if container != nil { - c = &Container{Container: container} + if ctr != nil { + c = &Container{Container: ctr} } if err != nil { - return c, fmt.Errorf("generic container: %w", err) + return c, fmt.Errorf("run pinecone: %w", err) } return c, nil