utils.go 1023 B

12345678910111213141516171819202122232425262728293031323334353637383940414243444546
  1. package integration
  2. import (
  3. "fmt"
  4. "os"
  5. "testing"
  6. )
  7. const InstanceNamePrefix = "SdkIntegrationTestInstance"
  8. type Config struct {
  9. AccessKeyId string
  10. AccessKeySecret string
  11. PublicKeyId string
  12. PrivateKey string
  13. RoleArn string
  14. ChildAK string
  15. ChildSecret string
  16. }
  17. func getConfigFromEnv() *Config {
  18. config := &Config{
  19. AccessKeyId: os.Getenv("ACCESS_KEY_ID"),
  20. AccessKeySecret: os.Getenv("ACCESS_KEY_SECRET"),
  21. PublicKeyId: os.Getenv("PUBLIC_KEY_ID"),
  22. PrivateKey: os.Getenv("PRIVATE_KEY"),
  23. RoleArn: os.Getenv("ROLE_ARN"),
  24. ChildAK: os.Getenv("CHILD_AK"),
  25. ChildSecret: os.Getenv("CHILD_SECRET"),
  26. }
  27. if config.AccessKeyId == "" {
  28. panic("Get ACCESS_KEY_ID from environment variables failed")
  29. } else {
  30. return config
  31. }
  32. }
  33. func getEcsDemoInstanceId() string {
  34. return os.Getenv("DEMO_ECS_INSTANCE_ID")
  35. }
  36. func assertErrorNil(t *testing.T, err error, message string) {
  37. if err != nil {
  38. fmt.Fprintf(os.Stderr, message+": %v\n", err)
  39. }
  40. }