mirror of
https://github.com/c9s/bbgo.git
synced 2024-11-23 15:25:14 +00:00
39 lines
1.2 KiB
Go
39 lines
1.2 KiB
Go
package testutil
|
|
|
|
import (
|
|
"os"
|
|
"regexp"
|
|
"testing"
|
|
)
|
|
|
|
func maskSecret(s string) string {
|
|
re := regexp.MustCompile(`\b(\w{4})\w+\b`)
|
|
s = re.ReplaceAllString(s, "$1******")
|
|
return s
|
|
}
|
|
|
|
func IntegrationTestConfigured(t *testing.T, prefix string) (key, secret string, ok bool) {
|
|
var hasKey, hasSecret bool
|
|
key, hasKey = os.LookupEnv(prefix + "_API_KEY")
|
|
secret, hasSecret = os.LookupEnv(prefix + "_API_SECRET")
|
|
ok = hasKey && hasSecret && os.Getenv("TEST_"+prefix) == "1"
|
|
if ok {
|
|
t.Logf(prefix+" api integration test enabled, key = %s, secret = %s", maskSecret(key), maskSecret(secret))
|
|
}
|
|
|
|
return key, secret, ok
|
|
}
|
|
|
|
func IntegrationTestWithPassphraseConfigured(t *testing.T, prefix string) (key, secret, passphrase string, ok bool) {
|
|
var hasKey, hasSecret, hasPassphrase bool
|
|
key, hasKey = os.LookupEnv(prefix + "_API_KEY")
|
|
secret, hasSecret = os.LookupEnv(prefix + "_API_SECRET")
|
|
passphrase, hasPassphrase = os.LookupEnv(prefix + "_API_PASSPHRASE")
|
|
ok = hasKey && hasSecret && hasPassphrase && os.Getenv("TEST_"+prefix) == "1"
|
|
if ok {
|
|
t.Logf(prefix+" api integration test enabled, key = %s, secret = %s, passphrase= %s", maskSecret(key), maskSecret(secret), maskSecret(passphrase))
|
|
}
|
|
|
|
return key, secret, passphrase, ok
|
|
}
|