From bdb6fee38c13ea1e3e50753e232416935de35ff2 Mon Sep 17 00:00:00 2001
From: unclejack <unclejacksons@gmail.com>
Date: Thu, 20 Nov 2014 19:33:15 +0200
Subject: [PATCH] pkg/system: fix cleanup in tests

Docker-DCO-1.1-Signed-off-by: Cristian Staretu <cristian.staretu@gmail.com> (github: unclejack)
---
 system/lstat_test.go  | 4 +++-
 system/stat_test.go   | 4 +++-
 system/utimes_test.go | 7 ++++---
 3 files changed, 10 insertions(+), 5 deletions(-)

diff --git a/system/lstat_test.go b/system/lstat_test.go
index 7e271ef..9bab4d7 100644
--- a/system/lstat_test.go
+++ b/system/lstat_test.go
@@ -1,11 +1,13 @@
 package system
 
 import (
+	"os"
 	"testing"
 )
 
 func TestLstat(t *testing.T) {
-	file, invalid, _ := prepareFiles(t)
+	file, invalid, _, dir := prepareFiles(t)
+	defer os.RemoveAll(dir)
 
 	statFile, err := Lstat(file)
 	if err != nil {
diff --git a/system/stat_test.go b/system/stat_test.go
index 0dcb239..abcc8ea 100644
--- a/system/stat_test.go
+++ b/system/stat_test.go
@@ -1,12 +1,14 @@
 package system
 
 import (
+	"os"
 	"syscall"
 	"testing"
 )
 
 func TestFromStatT(t *testing.T) {
-	file, _, _ := prepareFiles(t)
+	file, _, _, dir := prepareFiles(t)
+	defer os.RemoveAll(dir)
 
 	stat := &syscall.Stat_t{}
 	err := syscall.Lstat(file, stat)
diff --git a/system/utimes_test.go b/system/utimes_test.go
index 38e4020..1dea47c 100644
--- a/system/utimes_test.go
+++ b/system/utimes_test.go
@@ -8,7 +8,7 @@ import (
 	"testing"
 )
 
-func prepareFiles(t *testing.T) (string, string, string) {
+func prepareFiles(t *testing.T) (string, string, string, string) {
 	dir, err := ioutil.TempDir("", "docker-system-test")
 	if err != nil {
 		t.Fatal(err)
@@ -26,11 +26,12 @@ func prepareFiles(t *testing.T) (string, string, string) {
 		t.Fatal(err)
 	}
 
-	return file, invalid, symlink
+	return file, invalid, symlink, dir
 }
 
 func TestLUtimesNano(t *testing.T) {
-	file, invalid, symlink := prepareFiles(t)
+	file, invalid, symlink, dir := prepareFiles(t)
+	defer os.RemoveAll(dir)
 
 	before, err := os.Stat(file)
 	if err != nil {