add Probers to Probe pkgs.

2015-01-27 10:00:21 -08:00
parent a298402bd4
commit 6eb0b89cbd
7 changed files with 38 additions and 11 deletions
--- a/pkg/probe/tcp/tcp_test.go
+++ b/pkg/probe/tcp/tcp_test.go
@@ -28,6 +28,7 @@ import (
 )

 func TestTcpHealthChecker(t *testing.T) {
+	prober := New()
 	tests := []struct {
 		expectedStatus probe.Status
 		usePort        bool
@@ -58,7 +59,7 @@ func TestTcpHealthChecker(t *testing.T) {
 		if !test.usePort {
 			p = -1
 		}
-		status, err := Probe(host, p)
+		status, err := prober.Probe(host, p)
 		if status != test.expectedStatus {
 			t.Errorf("expected: %v, got: %v", test.expectedStatus, status)
 		}