Method SetupTest

pkg/blobmanager/s3/backend_test.go:277–331

()

const testBucket = "test-bucket"

func (s *testSuite) SetupTest() {
	s.minio = newMinioInstance(s.T())
	location := fmt.Sprintf("http://%s/%s", s.minio.ConnectionString(s.T()), testBucket)

	// Create backend
	backend, err := NewBackend(&Credentials{
		AccessKeyID:     "root",
		SecretAccessKey: "test-password",
		Region:          "us-east-1",
		Location:        location,
	})
	require.NoError(s.T(), err)
	s.backend = backend

	invalidBackend, err := NewBackend(&Credentials{
		AccessKeyID:     "root",
		SecretAccessKey: "wrong-password",
		Region:          "us-east-1",
		BucketName:      testBucket,
		Location:        location,
	})

	require.NoError(s.T(), err)
	s.invalidBackend = invalidBackend

	// create bucket
	minioClient, err := minio.New(s.minio.ConnectionString(s.T()), &minio.Options{
		Creds: credentials.NewStaticV4("root", "test-password", ""), Secure: false,
	})
	require.NoError(s.T(), err)
	require.NoError(s.T(), minioClient.MakeBucket(context.TODO(), testBucket, minio.MakeBucketOptions{}))

	// upload a valid artifact
	buf := bytes.NewBuffer([]byte("test"))
	s.ownedObjectDigest = fmt.Sprintf("%x", sha256.Sum256(buf.Bytes()))
	// calculate sha256 of the content in the buffer
	err = s.backend.Upload(context.Background(), buf, &pb.CASResource{Digest: s.ownedObjectDigest, FileName: "test.txt"})
	require.NoError(s.T(), err)

	// Copy an existing object but reference it from somewhere else
	s.tamperedObjectDigest = "b5bb9d8014a0f9b1d61e21e796d78dccdf1352f23cd32812f4850b878ae4944c"
	_, err = minioClient.CopyObject(context.Background(), minio.CopyDestOptions{
		Bucket: testBucket, Object: fmt.Sprintf("sha256:%s", s.tamperedObjectDigest),
	}, minio.CopySrcOptions{
		Bucket: testBucket, Object: fmt.Sprintf("sha256:%s", s.ownedObjectDigest),
	})
	require.NoError(s.T(), err)

	// upload another one but by the client directly
	reader := bytes.NewReader([]byte("hello world"))
	s.externalObjectDigest = "external-deadbeef"
	_, err = minioClient.PutObject(context.Background(), testBucket, fmt.Sprintf("sha256:%s", s.externalObjectDigest), reader, reader.Size(),
		minio.PutObjectOptions{})
	require.NoError(s.T(), err)
}

func (s *testSuite) TearDownTest() {
	if s.minio == nil {