From 0cc7425dd7c4e0dfcb8e2113cd238f314aadd131 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?G=C3=A1bor=20Lipt=C3=A1k?= <gliptak@gmail.com>
Date: Sat, 11 Apr 2020 16:26:22 -0400
Subject: [PATCH] Additional mount unittest
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: Gábor Lipták <gliptak@gmail.com>
---
 pkg/driver/mount_test.go | 54 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 54 insertions(+)

diff --git a/pkg/driver/mount_test.go b/pkg/driver/mount_test.go
index 22fe3c947c..11b11db1ff 100644
--- a/pkg/driver/mount_test.go
+++ b/pkg/driver/mount_test.go
@@ -41,6 +41,10 @@ func TestMakeDir(t *testing.T) {
 		t.Fatalf("Expect no error but got: %v", err)
 	}
 
+	if mountObj.MakeDir(targetPath) != nil {
+		t.Fatalf("Expect no error but got: %v", err)
+	}
+
 	if exists, err := mountObj.ExistsPath(targetPath); !exists {
 		t.Fatalf("Expect no error but got: %v", err)
 	}
@@ -64,8 +68,58 @@ func TestMakeFile(t *testing.T) {
 		t.Fatalf("Expect no error but got: %v", err)
 	}
 
+	if mountObj.MakeFile(targetPath) != nil {
+		t.Fatalf("Expect no error but got: %v", err)
+	}
+
 	if exists, err := mountObj.ExistsPath(targetPath); !exists {
 		t.Fatalf("Expect no error but got: %v", err)
 	}
 
 }
+
+func TestExistsPath(t *testing.T) {
+	// Setup the full driver and its environment
+	dir, err := ioutil.TempDir("", "mount-ebs-csi")
+	if err != nil {
+		t.Fatalf("error creating directory %v", err)
+	}
+	defer os.RemoveAll(dir)
+
+	targetPath := filepath.Join(dir, "notafile")
+
+	var (
+		mountObj = newNodeMounter()
+	)
+
+	exists, err := mountObj.ExistsPath(targetPath)
+
+	if err != nil {
+		t.Fatalf("Expect no error but got: %v", err)
+	}
+
+	if exists {
+		t.Fatalf("Expected file %s to not exist", targetPath)
+	}
+
+}
+
+func TestGetDeviceName(t *testing.T) {
+	// Setup the full driver and its environment
+	dir, err := ioutil.TempDir("", "mount-ebs-csi")
+	if err != nil {
+		t.Fatalf("error creating directory %v", err)
+	}
+	defer os.RemoveAll(dir)
+
+	targetPath := filepath.Join(dir, "notafile")
+
+	var (
+		mountObj = newNodeMounter()
+	)
+
+	if _, _, err := mountObj.GetDeviceName(targetPath); err != nil {
+		t.Fatalf("Expect no error but got: %v", err)
+	}
+
+}