49 lines
1.1 KiB
Go
49 lines
1.1 KiB
Go
|
package ddl
|
||
|
|
|
||
|
|
import (
|
||
|
|
"crypto/rand"
|
||
|
|
"math/big"
|
||
|
|
"os"
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/davecgh/go-spew/spew"
|
||
|
|
)
|
||
|
|
|
||
|
|
func RequireEnvString(name string, t *testing.T) string {
|
||
|
|
if strVal, isSet := os.LookupEnv(name); isSet {
|
||
|
|
return strVal
|
||
|
|
} else {
|
||
|
|
t.Errorf("Required environment variable %q not set", name)
|
||
|
|
return ""
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func DumpStruct(a ...interface{}) string {
|
||
|
|
dump := spew.Sdump(a...)
|
||
|
|
|
||
|
|
if strings.HasSuffix(dump, "\n") && len(dump) > 2 {
|
||
|
|
dump = dump[0 : len(dump)-len("\n")]
|
||
|
|
}
|
||
|
|
|
||
|
|
return dump
|
||
|
|
}
|
||
|
|
|
||
|
|
// GenerateRandomString returns a securely generated random string.
|
||
|
|
// It will return an error if the system's secure random
|
||
|
|
// number generator fails to function correctly, in which
|
||
|
|
// case the caller should not continue.
|
||
|
|
func GenerateRandomString(n int) (string, error) {
|
||
|
|
const letters = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"
|
||
|
|
ret := make([]byte, n)
|
||
|
|
for i := 0; i < n; i++ {
|
||
|
|
num, err := rand.Int(rand.Reader, big.NewInt(int64(len(letters))))
|
||
|
|
if err != nil {
|
||
|
|
return "", err
|
||
|
|
}
|
||
|
|
ret[i] = letters[num.Int64()]
|
||
|
|
}
|
||
|
|
|
||
|
|
return string(ret), nil
|
||
|
|
}
|