8e6e53cf63
When running the test suite in CI (where requests are centralized from relatively few IPs), we'd occasionally hit Dockerhub's rate limits. Luckily Hashicorp runs a (limited) public mirror of the containers we need, so we can switch to them here in the tests. For consistency between developer and CI, we've opted to have the tests always pull from the Hashicorp mirror, rather than updating the CI runner to prefer the mirror. We exclude nomad and influxdb as we don't presently mirror these repos. Signed-off-by: Alexander Scheel <alex.scheel@hashicorp.com> Signed-off-by: Alexander Scheel <alex.scheel@hashicorp.com>
112 lines
2.5 KiB
Go
112 lines
2.5 KiB
Go
package minio
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/url"
|
|
"testing"
|
|
|
|
"github.com/aws/aws-sdk-go/aws"
|
|
"github.com/aws/aws-sdk-go/aws/credentials"
|
|
"github.com/aws/aws-sdk-go/aws/defaults"
|
|
"github.com/aws/aws-sdk-go/aws/session"
|
|
"github.com/aws/aws-sdk-go/service/s3"
|
|
"github.com/hashicorp/vault/helper/testhelpers/docker"
|
|
)
|
|
|
|
type Config struct {
|
|
Endpoint string
|
|
AccessKeyID string
|
|
SecretAccessKey string
|
|
Region string
|
|
}
|
|
|
|
const (
|
|
accessKeyID = "min-access-key"
|
|
secretKey = "min-secret-key"
|
|
)
|
|
|
|
func PrepareTestContainer(t *testing.T, version string) (func(), *Config) {
|
|
if version == "" {
|
|
version = "latest"
|
|
}
|
|
runner, err := docker.NewServiceRunner(docker.RunOptions{
|
|
ContainerName: "minio",
|
|
ImageRepo: "docker.mirror.hashicorp.services/minio/minio",
|
|
ImageTag: version,
|
|
Env: []string{
|
|
"MINIO_ACCESS_KEY=" + accessKeyID,
|
|
"MINIO_SECRET_KEY=" + secretKey,
|
|
},
|
|
Cmd: []string{"server", "/data"},
|
|
Ports: []string{"9000/tcp"},
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("Could not start docker Minio: %s", err)
|
|
}
|
|
|
|
svc, err := runner.StartService(context.Background(), connectMinio)
|
|
if err != nil {
|
|
t.Fatalf("Could not start docker Minio: %s", err)
|
|
}
|
|
|
|
return svc.Cleanup, &Config{
|
|
Endpoint: svc.Config.URL().Host,
|
|
AccessKeyID: accessKeyID,
|
|
SecretAccessKey: secretKey,
|
|
Region: "us-east-1",
|
|
}
|
|
}
|
|
|
|
func connectMinio(ctx context.Context, host string, port int) (docker.ServiceConfig, error) {
|
|
u := url.URL{
|
|
Scheme: "s3",
|
|
Host: fmt.Sprintf("%s:%d", host, port),
|
|
}
|
|
|
|
c := &Config{
|
|
Endpoint: u.Host,
|
|
AccessKeyID: accessKeyID,
|
|
SecretAccessKey: secretKey,
|
|
Region: "us-east-1",
|
|
}
|
|
s3conn, err := c.Conn()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
_, err = s3conn.ListBuckets(&s3.ListBucketsInput{})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return docker.NewServiceURL(u), nil
|
|
}
|
|
|
|
func (c *Config) Conn() (*s3.S3, error) {
|
|
cfg := &aws.Config{
|
|
DisableSSL: aws.Bool(true),
|
|
Region: aws.String("us-east-1"),
|
|
Endpoint: aws.String(c.Endpoint),
|
|
S3ForcePathStyle: aws.Bool(true),
|
|
Credentials: credentials.NewChainCredentials(
|
|
[]credentials.Provider{
|
|
&credentials.StaticProvider{
|
|
Value: credentials.Value{
|
|
AccessKeyID: accessKeyID,
|
|
SecretAccessKey: secretKey,
|
|
},
|
|
},
|
|
&credentials.EnvProvider{},
|
|
&credentials.SharedCredentialsProvider{},
|
|
defaults.RemoteCredProvider(*(defaults.Config()), defaults.Handlers()),
|
|
}),
|
|
}
|
|
|
|
sess, err := session.NewSession(cfg)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return s3.New(sess), nil
|
|
}
|