From 8aa9d20932665eaf95a53ad420eb85559a4dd5e2 Mon Sep 17 00:00:00 2001 From: Mike Goelzer Date: Sun, 12 Apr 2015 08:14:48 -0700 Subject: [PATCH] Additional unit tests; raises generator.go test coverage to 100% Signed-off-by: Mike Goelzer --- discovery/generator_test.go | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/discovery/generator_test.go b/discovery/generator_test.go index 827d1ef55d..747334452f 100644 --- a/discovery/generator_test.go +++ b/discovery/generator_test.go @@ -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) +}