2020-03-01 16:06:34 +00:00
|
|
|
package testcontainers
|
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
|
|
|
|
|
|
|
"github.com/pkg/errors"
|
|
|
|
)
|
|
|
|
|
|
|
|
// GenericContainerRequest represents parameters to a generic container
|
|
|
|
type GenericContainerRequest struct {
|
|
|
|
ContainerRequest // embedded request for provider
|
|
|
|
Started bool // whether to auto-start the container
|
|
|
|
ProviderType ProviderType // which provider to use, Docker if empty
|
|
|
|
}
|
|
|
|
|
2020-09-06 12:31:23 +00:00
|
|
|
// GenericNetworkRequest represents parameters to a generic network
|
|
|
|
type GenericNetworkRequest struct {
|
|
|
|
NetworkRequest // embedded request for provider
|
|
|
|
ProviderType ProviderType // which provider to use, Docker if empty
|
|
|
|
}
|
|
|
|
|
|
|
|
// GenericNetwork creates a generic network with parameters
|
|
|
|
func GenericNetwork(ctx context.Context, req GenericNetworkRequest) (Network, error) {
|
|
|
|
provider, err := req.ProviderType.GetProvider()
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
network, err := provider.CreateNetwork(ctx, req.NetworkRequest)
|
|
|
|
if err != nil {
|
|
|
|
return nil, errors.Wrap(err, "failed to create network")
|
|
|
|
}
|
|
|
|
|
|
|
|
return network, nil
|
|
|
|
}
|
|
|
|
|
2020-03-01 16:06:34 +00:00
|
|
|
// GenericContainer creates a generic container with parameters
|
|
|
|
func GenericContainer(ctx context.Context, req GenericContainerRequest) (Container, error) {
|
|
|
|
provider, err := req.ProviderType.GetProvider()
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
c, err := provider.CreateContainer(ctx, req.ContainerRequest)
|
|
|
|
if err != nil {
|
|
|
|
return nil, errors.Wrap(err, "failed to create container")
|
|
|
|
}
|
|
|
|
|
|
|
|
if req.Started {
|
|
|
|
if err := c.Start(ctx); err != nil {
|
|
|
|
return c, errors.Wrap(err, "failed to start container")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return c, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// GenericProvider represents an abstraction for container and network providers
|
|
|
|
type GenericProvider interface {
|
|
|
|
ContainerProvider
|
|
|
|
NetworkProvider
|
|
|
|
}
|