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

Add error handling codes to tests

Browse Source
This commit is contained in:
Tatsushi Demachi 2014-07-21 11:09:57 +09:00
parent 0e0b3516a6
commit a83423c361
1 changed files with 29 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
fastping_test.go
View File

@ -50,7 +50,7 @@ func TestRun(t *testing.T) {
found1, found100 := false, false found1, found100 := false, false
called, idle := false, false called, idle := false, false
p.AddHandler("receive", func(ip *net.IPAddr, d time.Duration) { err := p.AddHandler("receive", func(ip *net.IPAddr, d time.Duration) {
called = true called = true
if ip.String() == "127.0.0.1" { if ip.String() == "127.0.0.1" {
found1 = true found1 = true
@ -58,10 +58,18 @@ func TestRun(t *testing.T) {
found100 = true found100 = true
} }
}) })
p.AddHandler("idle", func() { if err != nil {
t.Fatalf("Failed to add receive hander: %v", err)
}
err = p.AddHandler("idle", func() {
idle = true idle = true
}) })
err := p.Run() if err != nil {
t.Fatalf("Failed to add idle handler: %v", err)
}
err = p.Run()
if err != nil { if err != nil {
t.Fatalf("Pinger returns error: %v", err) t.Fatalf("Pinger returns error: %v", err)
} }
@ -93,18 +101,25 @@ func TestMultiRun(t *testing.T) {
var mu sync.Mutex var mu sync.Mutex
res1 := 0 res1 := 0
p1.AddHandler("receive", func(*net.IPAddr, time.Duration) { err := p1.AddHandler("receive", func(*net.IPAddr, time.Duration) {
mu.Lock() mu.Lock()
res1++ res1++
mu.Unlock() mu.Unlock()
}) })
if err != nil {
t.Fatalf("Failed to add receive hander: %v", err)
}
res2 := 0 res2 := 0
p2.AddHandler("receive", func(*net.IPAddr, time.Duration) { err = p2.AddHandler("receive", func(*net.IPAddr, time.Duration) {
mu.Lock() mu.Lock()
res2++ res2++
mu.Unlock() mu.Unlock()
}) })
if err != nil {
t.Fatalf("Failed to add idle handler: %v", err)
}
p1.MaxRTT, p2.MaxRTT = time.Millisecond*100, time.Millisecond*100 p1.MaxRTT, p2.MaxRTT = time.Millisecond*100, time.Millisecond*100
if err := p1.Run(); err != nil { if err := p1.Run(); err != nil {
@ -170,12 +185,19 @@ func TestRunLoop(t *testing.T) {
p.MaxRTT = time.Millisecond * 100 p.MaxRTT = time.Millisecond * 100
recvCount, idleCount := 0, 0 recvCount, idleCount := 0, 0
p.AddHandler("receive", func(*net.IPAddr, time.Duration) { err := p.AddHandler("receive", func(*net.IPAddr, time.Duration) {
recvCount++ recvCount++
}) })
p.AddHandler("idle", func() { if err != nil {
t.Fatalf("Failed to add receive handler: %v", err)
}
err = p.AddHandler("idle", func() {
idleCount++ idleCount++
}) })
if err != nil {
t.Fatalf("Failed to add idle handler: %v", err)
}
wait := make(chan bool) wait := make(chan bool)
quit, errch := p.RunLoop() quit, errch := p.RunLoop()