Fix hanging when requested zero-length string.
Add test for NewLen from 0 to 100.
This commit is contained in:
parent
e80eb311f3
commit
7668ee6861
@ -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)
|
||||
|
@ -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")
|
||||
|
Loading…
Reference in New Issue
Block a user