kubernetes
diff --git a/‎pkg/volume/util/volumepathhandler/volume_path_handler_test.go
Copy file name to clipboard
+309Lines changed: 309 additions & 0 deletions b/‎pkg/volume/util/volumepathhandler/volume_path_handler_test.go
Copy file name to clipboard
+309Lines changed: 309 additions & 0 deletions
@@ -0,0 +1,309 @@
+/*
+Copyright 2024 The Kubernetes Authors.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package volumepathhandler
+
+import (
+	"fmt"
+	"os"
+	"path/filepath"
+	"runtime"
+	"testing"
+)
+
+func TestMapDevice(t *testing.T) {
+	tempDir := os.TempDir()
+	testCases := []struct {
+		name        string
+		devicePath  string
+		mapPath     string
+		linkName    string
+		bindMount   bool
+		expectError bool
+	}{
+		{
+			name:        "valid symlink map",
+			devicePath:  "/dev/fakeDevice",
+			mapPath:     filepath.Join(tempDir, "test-map-valid"),
+			linkName:    "validLink",
+			bindMount:   false,
+			expectError: false,
+		},
+		{
+			name:        "empty device path",
+			devicePath:  "",
+			mapPath:     filepath.Join(tempDir, "test-map-empty-device"),
+			linkName:    "emptyDeviceLink",
+			bindMount:   false,
+			expectError: true,
+		},
+		{
+			name:        "empty map path",
+			devicePath:  "/dev/fakeDevice",
+			mapPath:     "",
+			linkName:    "emptyMapLink",
+			bindMount:   false,
+			expectError: true,
+		},
+	}
+
+	for _, tc := range testCases {
+		t.Run(tc.name, func(t *testing.T) {
+			if tc.mapPath != "" {
+				if err := os.MkdirAll(tc.mapPath, 0750); err != nil {
+					t.Fatalf("failed to create mapPath: %v", err)
+				}
+				defer func() {
+					if err := os.RemoveAll(tc.mapPath); err != nil {
+						t.Fatalf("failed to remove mapPath: %v", err)
+					}
+				}()
+			}
+			handler := VolumePathHandler{}
+			err := handler.MapDevice(tc.devicePath, tc.mapPath, tc.linkName, tc.bindMount)
+			if (err != nil) != tc.expectError {
+				t.Fatalf("expected error: %v, got: %v", tc.expectError, err)
+			}
+			if !tc.expectError {
+				linkPath := filepath.Join(tc.mapPath, tc.linkName)
+				if _, err := os.Lstat(linkPath); os.IsNotExist(err) {
+					t.Fatalf("expected symlink to exist, got error: %v", err)
+				}
+			}
+		})
+	}
+}
+
+func TestUnmapDevice(t *testing.T) {
+	tempDir := os.TempDir()
+	testCases := []struct {
+		name        string
+		mapPath     string
+		linkName    string
+		bindMount   bool
+		setup       func(string, string)
+		expectError bool
+	}{
+		{
+			name:      "valid symlink unmap",
+			mapPath:   filepath.Join(tempDir, "test-unmap-valid"),
+			linkName:  "validLink",
+			bindMount: false,
+			setup: func(mapPath, linkName string) {
+				if err := os.MkdirAll(mapPath, 0750); err != nil {
+					t.Fatalf("failed to create mapPath in setup: %v", err)
+				}
+				if err := os.Symlink("/dev/fakeDevice", filepath.Join(mapPath, linkName)); err != nil {
+					t.Fatalf("failed to create symlink in setup: %v", err)
+				}
+			},
+			expectError: false,
+		},
+		{
+			name:      "symlink does not exist",
+			mapPath:   filepath.Join(tempDir, "test-unmap-nonexistent"),
+			linkName:  "nonexistentLink",
+			bindMount: false,
+			setup: func(mapPath, linkName string) {
+				if err := os.MkdirAll(mapPath, 0750); err != nil {
+					t.Fatalf("failed to create mapPath in setup: %v", err)
+				}
+			},
+			expectError: false,
+		},
+	}
+
+	for _, tc := range testCases {
+		t.Run(tc.name, func(t *testing.T) {
+			handler := VolumePathHandler{}
+			if tc.setup != nil {
+				tc.setup(tc.mapPath, tc.linkName)
+			}
+			defer func() {
+				if err := os.RemoveAll(tc.mapPath); err != nil {
+					t.Fatalf("failed to remove mapPath: %v", err)
+				}
+			}()
+
+			err := handler.UnmapDevice(tc.mapPath, tc.linkName, tc.bindMount)
+			if (err != nil) != tc.expectError {
+				t.Fatalf("expected error: %v, got: %v", tc.expectError, err)
+			}
+			if !tc.expectError {
+				linkPath := filepath.Join(tc.mapPath, tc.linkName)
+				if _, err := os.Lstat(linkPath); !os.IsNotExist(err) {
+					t.Fatalf("expected symlink to be removed, got error: %v", err)
+				}
+			}
+		})
+	}
+}
+
+func TestRemoveMapPath(t *testing.T) {
+	tempDir := os.TempDir()
+	mapPath := filepath.Join(tempDir, "test-remove-map-path")
+
+	if err := os.MkdirAll(mapPath, 0755); err != nil {
+		t.Fatalf("failed to create map path: %v", err)
+	}
+
+	testCases := []struct {
+		name        string
+		mapPath     string
+		expectError bool
+	}{
+		{
+			name:        "Remove Existing Map Path",
+			mapPath:     mapPath,
+			expectError: false,
+		},
+		{
+			name:        "Remove Non-existing Map Path",
+			mapPath:     filepath.Join(tempDir, "non-existing-path"),
+			expectError: false,
+		},
+	}
+
+	for _, tc := range testCases {
+		t.Run(tc.name, func(t *testing.T) {
+			handler := VolumePathHandler{}
+			err := handler.RemoveMapPath(tc.mapPath)
+			if (err != nil) != tc.expectError {
+				t.Fatalf("expected error: %v, got: %v", tc.expectError, err)
+			}
+			if !tc.expectError {
+				if _, err := os.Stat(tc.mapPath); !os.IsNotExist(err) {
+					t.Fatalf("expected map path to be deleted, got error: %v", err)
+				}
+			}
+		})
+	}
+}
+
+func TestIsSymlinkExist(t *testing.T) {
+	tempDir := os.TempDir()
+	mapPath := filepath.Join(tempDir, "test-symlink")
+	linkName := "test-symlink-link"
+
+	if err := os.MkdirAll(mapPath, 0755); err != nil {
+		t.Fatalf("failed to create test directory: %v", err)
+	}
+
+	defer func() {
+		if err := os.RemoveAll(mapPath); err != nil {
+			t.Fatalf("failed to remove mapPath: %v", err)
+		}
+	}()
+
+	if err := os.Symlink("/dev/fakeDevice", filepath.Join(mapPath, linkName)); err != nil {
+		t.Fatalf("failed to create symlink: %v", err)
+	}
+
+	testCases := []struct {
+		name           string
+		mapPath        string
+		expectedExists bool
+	}{
+		{
+			name:           "Check Existing Symlink",
+			mapPath:        filepath.Join(mapPath, linkName),
+			expectedExists: true,
+		},
+		{
+			name:           "Check Non-existing Symlink",
+			mapPath:        filepath.Join(mapPath, "non-existing-symlink"),
+			expectedExists: false,
+		},
+	}
+
+	for _, tc := range testCases {
+		t.Run(tc.name, func(t *testing.T) {
+			exists, err := VolumePathHandler{}.IsSymlinkExist(tc.mapPath)
+			if err != nil {
+				t.Fatalf("expected no error, got: %v", err)
+			}
+			if exists != tc.expectedExists {
+				t.Fatalf("expected symlink existence to be: %v, got: %v", tc.expectedExists, exists)
+			}
+		})
+	}
+}
+
+func checkMountSupport() error {
+	// Check if the operating system supports mounting
+	if runtime.GOOS != "linux" {
+		return fmt.Errorf("mount is not supported on %s", runtime.GOOS)
+	}
+	return nil
+}
+
+func TestIsDeviceBindMountExist(t *testing.T) {
+	tempDir := os.TempDir()
+	mapPath := filepath.Join(tempDir, "test-bind-mount")
+	linkName := "test-bind-mount-link"
+	devicePath := "/dev/fakeDevice"
+
+	if err := os.MkdirAll(mapPath, 0755); err != nil {
+		t.Fatalf("failed to create test directory: %v", err)
+	}
+	defer func() {
+		if err := os.RemoveAll(mapPath); err != nil {
+			t.Fatalf("failed to remove mapPath: %v", err)
+		}
+	}()
+
+	if err := checkMountSupport(); err != nil {
+		t.Skipf("skipping test because mount is not supported: %v", err)
+	}
+
+	if _, err := os.Stat(devicePath); os.IsNotExist(err) {
+		t.Skipf("skipping test because the device %s does not exist: %v", devicePath, err)
+	}
+
+	err := VolumePathHandler{}.MapDevice(devicePath, mapPath, linkName, true)
+	if err != nil {
+		t.Fatalf("failed to create bind mount: %v", err)
+	}
+
+	testCases := []struct {
+		name           string
+		mapPath        string
+		expectedExists bool
+	}{
+		{
+			name:           "Check Existing Bind Mount",
+			mapPath:        filepath.Join(mapPath, linkName),
+			expectedExists: true,
+		},
+		{
+			name:           "Check Non-existing Bind Mount",
+			mapPath:        filepath.Join(mapPath, "non-existing-bind-mount"),
+			expectedExists: false,
+		},
+	}
+
+	for _, tc := range testCases {
+		t.Run(tc.name, func(t *testing.T) {
+			exists, err := VolumePathHandler{}.IsDeviceBindMountExist(tc.mapPath)
+			if err != nil {
+				t.Fatalf("expected no error, got: %v", err)
+			}
+			if exists != tc.expectedExists {
+				t.Fatalf("expected bind mount existence to be: %v, got: %v", tc.expectedExists, exists)
+			}
+		})
+	}
+}