Using [0X-XX] in hostname pattern (#6)

Support hostnames contains 01:10

Author: kamynina

cmd/generate.go

@@ -185,13 +185,14 @@ func parseHostPattern(pattern string) []string {
 		from, _ := strconv.Atoi(pieces[2])
 		to, _ := strconv.Atoi(pieces[3])
 		tail := pieces[4]
+		hostnameFmt := fmt.Sprintf("%%s%%0%dd%%s", len(pieces[2]))
 		if from <= to {
 			ret := []string{}
 			for i := from; i <= to; i++ {
-				ret = append(ret, fmt.Sprintf("%s%d%s", head, i, tail))
+				ret = append(ret, fmt.Sprintf(hostnameFmt, head, i, tail))
 			}
 			return ret
 		}
 	}
 	return []string{pattern}
-}
+}

cmd/generate_test.go

@@ -18,6 +18,7 @@ func TestParseHostPattern(t *testing.T) {
 	AssertThat(t, parseHostPattern("test[1:3]-host"), EqualTo{[]string{"test1-host", "test2-host", "test3-host"}})
 	AssertThat(t, parseHostPattern("[1:3]-host"), EqualTo{[]string{"1-host", "2-host", "3-host"}})
 	AssertThat(t, parseHostPattern("host-[1:3]"), EqualTo{[]string{"host-1", "host-2", "host-3"}})
+	AssertThat(t, parseHostPattern("host-[01:03]"), EqualTo{[]string{"host-01", "host-02", "host-03"}})
 }
 
 func TestConvert(t *testing.T) {