From 2cd49eb020061df846f0e20bc0cbe77020fc19b9 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Mart=C3=ADn=20Ferrari?= <tincho@debian.org>
Date: Sat, 19 Aug 2017 11:37:57 +0200
Subject: [PATCH] Fix path and timing issues with integration tests.

---
 node_exporter_test.go | 33 +++++++++++++++------------------
 1 file changed, 15 insertions(+), 18 deletions(-)

diff --git a/node_exporter_test.go b/node_exporter_test.go
index 538ef1af..a41ee971 100644
--- a/node_exporter_test.go
+++ b/node_exporter_test.go
@@ -13,8 +13,10 @@ import (
 	"github.com/prometheus/procfs"
 )
 
+var (
+	binary = filepath.Join(os.Getenv("GOPATH"), "bin/node_exporter")
+)
 const (
-	binary  = "./node_exporter"
 	address = "localhost:19100"
 )
 
@@ -54,7 +56,7 @@ func TestFileDescriptorLeak(t *testing.T) {
 		return nil
 	}
 
-	if err := runCommandAndTests(exporter, test); err != nil {
+	if err := runCommandAndTests(exporter, address, test); err != nil {
 		t.Error(err)
 	}
 }
@@ -83,7 +85,7 @@ func TestHandlingOfDuplicatedMetrics(t *testing.T) {
 		return queryExporter(address)
 	}
 
-	if err := runCommandAndTests(exporter, test); err != nil {
+	if err := runCommandAndTests(exporter, address, test); err != nil {
 		t.Error(err)
 	}
 }
@@ -106,27 +108,22 @@ func queryExporter(address string) error {
 	return nil
 }
 
-func runCommandAndTests(cmd *exec.Cmd, fn func(pid int) error) error {
+func runCommandAndTests(cmd *exec.Cmd, address string, fn func(pid int) error) error {
 	if err := cmd.Start(); err != nil {
 		return fmt.Errorf("failed to start command: %s", err)
 	}
-
-	errc := make(chan error)
-	go func() {
-		if err := cmd.Wait(); err != nil {
-			errc <- fmt.Errorf("execution of command failed: %s", err)
-		} else {
-			errc <- nil
+	time.Sleep(50 * time.Millisecond)
+	for i := 0; i < 10; i++ {
+		if err := queryExporter(address); err == nil {
+			break
+		}
+		time.Sleep(500 * time.Millisecond)
+		if cmd.Process== nil || i == 9 {
+			return fmt.Errorf("can't start command")
 		}
-	}()
-
-	// Allow the process to start before running any tests.
-	select {
-	case err := <-errc:
-		return err
-	case <-time.After(100 * time.Millisecond):
 	}
 
+	errc := make(chan error)
 	go func(pid int) {
 		errc <- fn(pid)
 	}(cmd.Process.Pid)