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