diff --git a/app/vminsert/netstorage/netstorage_test.go b/app/vminsert/netstorage/netstorage_test.go index 0a3ce9e38d..5d670c9a4a 100644 --- a/app/vminsert/netstorage/netstorage_test.go +++ b/app/vminsert/netstorage/netstorage_test.go @@ -1,11 +1,15 @@ package netstorage import ( + "flag" "runtime" "testing" ) func TestInitStopNodes(t *testing.T) { + if err := flag.Set("vmstorageDialTimeout", "1ms"); err != nil { + t.Fatalf("cannot set vmstorageDialTimeout flag: %s", err) + } for i := 0; i < 3; i++ { Init([]string{"host1", "host2"}, 0) runtime.Gosched() diff --git a/app/vmselect/netstorage/netstorage_test.go b/app/vmselect/netstorage/netstorage_test.go index 19d05060e6..364ca55947 100644 --- a/app/vmselect/netstorage/netstorage_test.go +++ b/app/vmselect/netstorage/netstorage_test.go @@ -1,12 +1,16 @@ package netstorage import ( + "flag" "reflect" "runtime" "testing" ) func TestInitStopNodes(t *testing.T) { + if err := flag.Set("vmstorageDialTimeout", "1ms"); err != nil { + t.Fatalf("cannot set vmstorageDialTimeout flag: %s", err) + } for i := 0; i < 3; i++ { Init([]string{"host1", "host2"}) runtime.Gosched()