Fix string padding removal from prefix

keybase · May 19, 2016 · f44b3a4 · f44b3a4
1 parent e222a80
commit f44b3a4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/util/rand.go b/util/rand.go
@@ -18,8 +18,9 @@ func RandomID(prefix string) (string, error) {
 	if err != nil {
 		return "", err
 	}
-	str := prefix + base32.StdEncoding.EncodeToString(buf)
+	str := base32.StdEncoding.EncodeToString(buf)
 	str = strings.Replace(str, "=", "", -1)
+	str = prefix + str
 	return str, nil
 }
 

diff --git a/util/rand_test.go b/util/rand_test.go
@@ -9,12 +9,12 @@ import (
 )
 
 func TestRandString(t *testing.T) {
-	s, err := RandomID("prefix.")
+	s, err := RandomID("prefix=")
 	t.Logf("Rand string: %s", s)
 	if err != nil {
 		t.Fatal(err)
 	}
-	if !strings.HasPrefix(s, "prefix.") {
+	if !strings.HasPrefix(s, "prefix=") {
 		t.Errorf("Invalid prefix: %s", s)
 	}
 	if len(s)-len("prefix.") != 52 {