From b9cfc706d49580b59a4ae94873fd1779fac599e8 Mon Sep 17 00:00:00 2001
From: Vincent Demeester <vincent@sbr.pm>
Date: Mon, 2 Nov 2015 19:44:09 +0100
Subject: [PATCH] Add TESTDIRS back

Makes it simpler to test only one package.
Was removed by #17491.

Signed-off-by: Vincent Demeester <vincent@sbr.pm>
Upstream-commit: 4a54ab387f4b11310476414f575fb2502970b65d
Component: engine
---
 components/engine/hack/make/test-unit | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/components/engine/hack/make/test-unit b/components/engine/hack/make/test-unit
index c70f6dd97c..9590525372 100644
--- a/components/engine/hack/make/test-unit
+++ b/components/engine/hack/make/test-unit
@@ -9,11 +9,16 @@ set -e
 #
 bundle_test_unit() {
 	date
+	if [ -z "$TESTDIRS" ]; then
+		TEST_PATH=./...
+	else
+		TEST_PATH=./${TESTDIRS}
+	fi
 	pkg_list=$(go list -e \
 		-f '{{if ne .Name "github.com/docker/docker"}}
-				{{.ImportPath}}
-			{{end}}' \
-		"${BUILDFLAGS[@]}" ./... \
+			{{.ImportPath}}
+		    {{end}}' \
+		"${BUILDFLAGS[@]}" $TEST_PATH \
 		| grep -v github.com/docker/docker/vendor \
 		| grep -v github.com/docker/docker/integration-cli)
 	go test -cover $GCCGOFLAGS -ldflags "$LDFLAGS" "${BUILDFLAGS[@]}" $TESTFLAGS $pkg_list