fix test config

influxdata · Mar 11, 2016 · c5d4a4e · c5d4a4e
1 parent a1fa1be
commit c5d4a4e
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/cmd/kapacitord/run/server_test.go b/cmd/kapacitord/run/server_test.go
@@ -25,7 +25,6 @@ import (
 	"github.com/influxdata/kapacitor"
 	"github.com/influxdata/kapacitor/cmd/kapacitord/run"
 	"github.com/influxdata/kapacitor/services/udf"
-	"github.com/influxdata/kapacitor/vendor/github.com/influxdata/influxdb/toml"
 )
 
 func TestServer_Ping(t *testing.T) {
@@ -955,7 +954,7 @@ func TestServer_UDFBatchAgents(t *testing.T) {
 			},
 			config: udf.FunctionConfig{
 				Prog:    filepath.Join(tdir, "outliers"),
-				Timeout: toml.Duration(time.Minute),
+				Timeout: config.Duration(time.Minute),
 			},
 		},
 		// Python
@@ -964,7 +963,7 @@ func TestServer_UDFBatchAgents(t *testing.T) {
 			config: udf.FunctionConfig{
 				Prog:    "python2",
 				Args:    []string{"-u", filepath.Join(udfDir, "agent/examples/outliers/outliers.py")},
-				Timeout: toml.Duration(time.Minute),
+				Timeout: config.Duration(time.Minute),
 				Env: map[string]string{
 					"PYTHONPATH": strings.Join(
 						[]string{filepath.Join(udfDir, "agent/py"), os.Getenv("PYTHONPATH")},