Merge pull request #1101 from dotcloud/fix-unit_tests

add sleep in tests and go fmt

buildfile_test.go

@@ -1,10 +1,10 @@
 package docker
 
 import (
+	"fmt"
 	"io/ioutil"
 	"sync"
 	"testing"
-	"fmt"
 )
 
 // mkTestContext generates a build context from the contents of the provided dockerfile.

container_test.go

@@ -517,6 +517,9 @@ func TestKillDifferentUser(t *testing.T) {
 		}
 	})
 
+	// Even if the state is running, lets give some time to lxc to spawn the process
+	container.WaitTimeout(500 * time.Millisecond)
+
 	if err := container.Kill(); err != nil {
 		t.Fatal(err)
 	}

runtime_test.go

@@ -366,14 +366,14 @@ func TestAllocatePortLocalhost(t *testing.T) {
 	defer container.Kill()
 
 	setTimeout(t, "Waiting for the container to be started timed out", 2*time.Second, func() {
-		for {
+		for !container.State.Running {
-			if container.State.Running {
-				break
-			}
 			time.Sleep(10 * time.Millisecond)
 		}
 	})
 
+	// Even if the state is running, lets give some time to lxc to spawn the process
+	container.WaitTimeout(500 * time.Millisecond)
+
 	conn, err := net.Dial("tcp",
 		fmt.Sprintf(
 			"localhost:%s", container.NetworkSettings.PortMapping[strconv.Itoa(port)],