1
0
mirror of https://github.com/goreleaser/goreleaser.git synced 2025-02-03 13:11:48 +02:00

test: fix clear env on s3 tests

This commit is contained in:
Carlos Alexandro Becker 2018-08-25 17:47:05 -03:00
parent 661f56ac15
commit 235fff7b8a
No known key found for this signature in database
GPG Key ID: E61E2F7DC14AB940

View File

@ -4,14 +4,15 @@ import (
"testing"
"fmt"
"github.com/aws/aws-sdk-go/aws"
"github.com/aws/aws-sdk-go/aws/session"
"github.com/stretchr/testify/assert"
"net/http"
"net/http/httptest"
"os"
"path/filepath"
"time"
"github.com/aws/aws-sdk-go/aws"
"github.com/aws/aws-sdk-go/aws/session"
"github.com/stretchr/testify/assert"
)
func setEnv() {
@ -19,6 +20,13 @@ func setEnv() {
os.Setenv("AWS_SECRET_ACCESS_KEY", "secret")
}
func clearnEnv() {
os.Unsetenv("AWS_ACCESS_KEY_ID")
os.Unsetenv("AWS_SECRET_ACCESS_KEY")
os.Unsetenv("AWS_SHARED_CREDENTIALS_FILE")
os.Unsetenv("AWS_CONFIG_FILE")
}
func Test_awsSession(t *testing.T) {
type args struct {
profile string
@ -55,7 +63,7 @@ func Test_awsSession(t *testing.T) {
{
name: "test default shared credentials provider",
before: func() {
os.Clearenv()
clearnEnv()
os.Setenv("AWS_SHARED_CREDENTIALS_FILE", filepath.Join("testdata", "credentials.ini"))
},
expectToken: "token",
@ -63,7 +71,7 @@ func Test_awsSession(t *testing.T) {
{
name: "test default shared credentials provider",
before: func() {
os.Clearenv()
clearnEnv()
os.Setenv("AWS_SHARED_CREDENTIALS_FILE", filepath.Join("testdata", "credentials.ini"))
},
expectToken: "token",
@ -71,7 +79,7 @@ func Test_awsSession(t *testing.T) {
{
name: "test profile with shared credentials provider",
before: func() {
os.Clearenv()
clearnEnv()
os.Setenv("AWS_SHARED_CREDENTIALS_FILE", filepath.Join("testdata", "credentials.ini"))
},
args: args{
@ -83,8 +91,8 @@ func Test_awsSession(t *testing.T) {
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
os.Clearenv()
defer os.Clearenv()
clearnEnv()
defer clearnEnv()
if tt.before != nil {
tt.before()
}
@ -130,8 +138,8 @@ const assumeRoleRespMsg = `
`
func Test_awsSession_mfa(t *testing.T) {
os.Clearenv()
defer os.Clearenv()
clearnEnv()
defer clearnEnv()
os.Setenv("AWS_SHARED_CREDENTIALS_FILE", filepath.Join("testdata", "credentials.ini"))
os.Setenv("AWS_CONFIG_FILE", filepath.Join("testdata", "config.ini"))
@ -180,8 +188,8 @@ func Test_awsSession_fail(t *testing.T) {
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
os.Clearenv()
defer os.Clearenv()
clearnEnv()
defer clearnEnv()
builder := newSessionBuilder()
sess := builder.Build()