Merge pull request #8 from StableLib/zerolength

Fix hanging when requested zero-length string.

uniuri.go

@@ -53,6 +53,9 @@ func NewLen(length int) string {
 // NewLenChars returns a new random string of the provided length, consisting
 // of the provided byte slice of allowed characters (maximum 256).
 func NewLenChars(length int, chars []byte) string {
+	if length == 0 {
+		return ""
+	}
 	b := make([]byte, length)
 	r := make([]byte, length+(length/4)) // storage for random bytes.
 	clen := len(chars)

uniuri_test.go

@@ -46,6 +46,15 @@ func TestNew(t *testing.T) {
 	}
 }
 
+func TestNewLen(t *testing.T) {
+	for i := 0; i < 100; i++ {
+		u := NewLen(i)
+		if len(u) != i {
+			t.Fatalf("request length %d, got %d", i, len(u))
+		}
+	}
+}
+
 func TestNewLenChars(t *testing.T) {
 	length := 10
 	chars := []byte("01234567")