taigrr/go-fastping
1
0
Fork 0
mirror of https://github.com/taigrr/go-fastping synced 2025-01-18 05:03:15 -08:00
Code Issues Projects Releases Wiki Activity

Simplify test

Browse Source
This commit is contained in:
Tatsushi Demachi 2015-08-18 20:58:17 +09:00
parent 780383b35c
commit 7c5596df1e
1 changed files with 11 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
fastping_test.go
View File

@ -7,17 +7,6 @@ import (
"time" "time"
) )
type addHostTest struct {
host string
addr *net.IPAddr
expect bool
}
var addHostTests = []addHostTest{
{host: "127.0.0.1", addr: &net.IPAddr{IP: net.IPv4(127, 0, 0, 1)}, expect: true},
{host: "localhost", addr: &net.IPAddr{IP: net.IPv4(127, 0, 0, 1)}, expect: false},
}
func TestSource(t *testing.T) { func TestSource(t *testing.T) {
for i, tt := range []struct { for i, tt := range []struct {
firstAddr string firstAddr string
@ -75,16 +64,25 @@ func TestSource(t *testing.T) {
} }
func TestAddIP(t *testing.T) { func TestAddIP(t *testing.T) {
addIPTests := []struct {
host string
addr *net.IPAddr
expect bool
}{
{host: "127.0.0.1", addr: &net.IPAddr{IP: net.IPv4(127, 0, 0, 1)}, expect: true},
{host: "localhost", addr: &net.IPAddr{IP: net.IPv4(127, 0, 0, 1)}, expect: false},
}
p := NewPinger() p := NewPinger()
for _, tt := range addHostTests { for _, tt := range addIPTests {
if ok := p.AddIP(tt.host); ok != nil { if ok := p.AddIP(tt.host); ok != nil {
if tt.expect != false { if tt.expect != false {
t.Errorf("AddIP failed: got %v, expected %v", ok, tt.expect) t.Errorf("AddIP failed: got %v, expected %v", ok, tt.expect)
} }
} }
} }
for _, tt := range addHostTests { for _, tt := range addIPTests {
if tt.expect { if tt.expect {
if !p.addrs[tt.host].IP.Equal(tt.addr.IP) { if !p.addrs[tt.host].IP.Equal(tt.addr.IP) {
t.Errorf("AddIP didn't save IPAddr: %v", tt.host) t.Errorf("AddIP didn't save IPAddr: %v", tt.host)