Skip to content

Commit

Permalink
Fix tests for travis
Browse files Browse the repository at this point in the history
  • Loading branch information
Julien Duchesne committed Oct 4, 2018
1 parent fc4e8e7 commit cbb05ac
Show file tree
Hide file tree
Showing 2 changed files with 16 additions and 10 deletions.
5 changes: 5 additions & 0 deletions Makefile
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,11 @@ install:
glide install
go install

coveralls:
wget https://raw.githubusercontent.com/coveo/terragrunt/master/scripts/coverage.sh
@sh ./coverage.sh --coveralls
rm coverage.sh

.PHONY: test
test:
go test ./...
Expand Down
21 changes: 11 additions & 10 deletions config_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -104,10 +104,7 @@ docker-image-build-folder: my-folder`)

func writeSSMConfig(parameterFolder, parameterKey, parameterValue string) {
fullParameterKey := fmt.Sprintf("%s/%s", parameterFolder, parameterKey)
awsSession := session.Must(session.NewSessionWithOptions(session.Options{
SharedConfigState: session.SharedConfigEnable,
}))
svc := ssm.New(awsSession)
client := getSSMClient()

putParameterInput := &ssm.PutParameterInput{
Name: aws.String(fullParameterKey),
Expand All @@ -116,27 +113,31 @@ func writeSSMConfig(parameterFolder, parameterKey, parameterValue string) {
Type: aws.String(ssm.ParameterTypeString),
}

if _, err := svc.PutParameter(putParameterInput); err != nil {
if _, err := client.PutParameter(putParameterInput); err != nil {
panic(err)
}
}

func deleteSSMConfig(parameterFolder, parameterKey string) {
fullParameterKey := fmt.Sprintf("%s/%s", parameterFolder, parameterKey)
awsSession := session.Must(session.NewSessionWithOptions(session.Options{
SharedConfigState: session.SharedConfigEnable,
}))
svc := ssm.New(awsSession)
client := getSSMClient()

deleteParameterInput := &ssm.DeleteParameterInput{
Name: aws.String(fullParameterKey),
}

if _, err := svc.DeleteParameter(deleteParameterInput); err != nil {
if _, err := client.DeleteParameter(deleteParameterInput); err != nil {
panic(err)
}
}

func getSSMClient() *ssm.SSM {
awsSession := session.Must(session.NewSessionWithOptions(session.Options{
SharedConfigState: session.SharedConfigEnable,
}))
return ssm.New(awsSession, &aws.Config{Region: aws.String("us-east-1")})
}

func randInt() int {
source := rand.NewSource(time.Now().UnixNano())
random := rand.New(source)
Expand Down

0 comments on commit cbb05ac

Please sign in to comment.