Additional unit tests; raises generator.go test coverage to 100%

Signed-off-by: Mike Goelzer <mike@goelzer.com>
This commit is contained in:
Mike Goelzer 2015-04-12 08:14:48 -07:00
parent ac4d776abb
commit 8aa9d20932
1 changed files with 14 additions and 0 deletions

View File

@ -39,3 +39,17 @@ func TestGeneratorWithPort(t *testing.T) {
assert.Equal(t, ips[9], "127.0.0.10:2375")
assert.Equal(t, ips[10], "127.0.0.11:2375")
}
func TestGenerateWithMalformedInputAtRangeStart(t *testing.T) {
malformedInput := "127.0.0.[x:11]:2375"
ips := Generate(malformedInput)
assert.Equal(t, len(ips), 1)
assert.Equal(t, ips[0], malformedInput)
}
func TestGenerateWithMalformedInputAtRangeEnd(t *testing.T) {
malformedInput := "127.0.0.[1:x]:2375"
ips := Generate(malformedInput)
assert.Equal(t, len(ips), 1)
assert.Equal(t, ips[0], malformedInput)
}