From 9078c8f2047d6e289a9637f84d05568547546421 Mon Sep 17 00:00:00 2001 From: Murilo Santana Date: Wed, 22 Apr 2015 10:48:52 -0300 Subject: [PATCH] check length after adding as well --- fastping_test.go | 17 ++++++++++------- 1 file changed, 10 insertions(+), 7 deletions(-) diff --git a/fastping_test.go b/fastping_test.go index 2b1330c..5f8b5bd 100644 --- a/fastping_test.go +++ b/fastping_test.go @@ -43,14 +43,15 @@ func TestRemoveIP(t *testing.T) { if err := p.AddIP("127.0.0.1"); err != nil { t.Fatalf("AddIP failed: %v", err) } + if len(p.addrs) != 1 { + t.Fatalf("AddIP length check failed") + } if err := p.RemoveIP("127.0.0.1"); err != nil { t.Fatalf("RemoveIP failed: %v", err) } - - length := len(p.addrs) - if length != 0 { - t.Fatalf("RemoveIP failed: got %v, expected %v", length, 0) + if len(p.addrs) != 0 { + t.Fatalf("RemoveIP length check failed") } } @@ -60,11 +61,13 @@ func TestRemoveIPAddr(t *testing.T) { if err := p.AddIP("127.0.0.1"); err != nil { t.Fatalf("AddIP failed: %v", err) } + if len(p.addrs) != 1 { + t.Fatalf("AddIP length check failed") + } p.RemoveIPAddr(&net.IPAddr{IP: net.IPv4(127, 0, 0, 1)}) - length := len(p.addrs) - if length != 0 { - t.Fatalf("RemoveIP failed: got %v, expected %v", length, 0) + if len(p.addrs) != 0 { + t.Fatalf("RemoveIPAddr length check failed") } }