зеркало из https://github.com/github/s3gof3r.git
78 строки
1.6 KiB
Go
78 строки
1.6 KiB
Go
package s3gof3r
|
|
|
|
import (
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
type authT struct {
|
|
env []string
|
|
}
|
|
|
|
// save current environment for restoration
|
|
func (s *authT) saveEnv() {
|
|
s.env = os.Environ()
|
|
os.Clearenv()
|
|
}
|
|
|
|
// restore environment after each test
|
|
func (s *authT) restoreEnv() {
|
|
os.Clearenv()
|
|
for _, kv := range s.env {
|
|
l := strings.SplitN(kv, "=", 2)
|
|
os.Setenv(l[0], l[1])
|
|
}
|
|
}
|
|
|
|
func TestEnvKeysWithoutToken(t *testing.T) {
|
|
testKeys := Keys{
|
|
AccessKey: "AKIDEXAMPLE",
|
|
SecretKey: "wJalrXUtnFEMI/K7MDENG+bPxRfiCYEXAMPLEKEY",
|
|
}
|
|
s := authT{}
|
|
s.saveEnv()
|
|
os.Setenv("AWS_ACCESS_KEY_ID", testKeys.AccessKey)
|
|
os.Setenv("AWS_SECRET_ACCESS_KEY", testKeys.SecretKey)
|
|
keys, err := EnvKeys()
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
if keys != testKeys {
|
|
t.Errorf("Keys do not match. Expected: %v. Actual: %v", testKeys, keys)
|
|
}
|
|
s.restoreEnv()
|
|
}
|
|
|
|
func TestEnvKeyWithToken(t *testing.T) {
|
|
testKeys := Keys{
|
|
AccessKey: "AKIDEXAMPLE",
|
|
SecretKey: "wJalrXUtnFEMI/K7MDENG+bPxRfiCYEXAMPLEKEY",
|
|
SecurityToken: "testtoken",
|
|
}
|
|
s := authT{}
|
|
s.saveEnv()
|
|
os.Setenv("AWS_ACCESS_KEY_ID", testKeys.AccessKey)
|
|
os.Setenv("AWS_SECRET_ACCESS_KEY", testKeys.SecretKey)
|
|
os.Setenv("AWS_SECURITY_TOKEN", testKeys.SecurityToken)
|
|
keys, err := EnvKeys()
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
if keys != testKeys {
|
|
t.Errorf("Keys do not match. Expected: %v. Actual: %v", testKeys, keys)
|
|
}
|
|
s.restoreEnv()
|
|
}
|
|
|
|
func TestEnvKeysNotSet(t *testing.T) {
|
|
s := authT{}
|
|
s.saveEnv()
|
|
_, err := EnvKeys()
|
|
expErr := "keys not set in environment: AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY"
|
|
if err.Error() != expErr {
|
|
t.Errorf("Expected error: %v. Actual: %v", expErr, err)
|
|
}
|
|
s.restoreEnv()
|
|
}
|