Function TestValidateFileAccess

cmd/gosqlx/cmd/integration_test.go:226–267

TestValidateFileAccess tests the file access validation

(t *testing.T)

// TestValidateFileAccess tests the file access validation
func TestValidateFileAccess(t *testing.T) {
	tmpDir := t.TempDir()

	// Test 1: Valid file
	t.Run("ValidFile", func(t *testing.T) {
		testFile := filepath.Join(tmpDir, "valid.sql")
		err := os.WriteFile(testFile, []byte("SELECT 1"), 0644)
		if err != nil {
			t.Fatalf("Failed to create test file: %v", err)
		}

		err = ValidateFileAccess(testFile)
		if err != nil {
			t.Errorf("Expected success for valid file, got error: %v", err)
		}
	})

	// Test 2: Directory
	t.Run("Directory", func(t *testing.T) {
		err := ValidateFileAccess(tmpDir)
		if err == nil {
			t.Error("Expected error for directory, got success")
		}
		// Accept either "directory" or "not a regular file" error messages
		if !strings.Contains(err.Error(), "directory") && !strings.Contains(err.Error(), "not a regular file") {
			t.Errorf("Expected directory or regular file error message, got: %v", err)
		}
	})

	// Test 3: Nonexistent file
	t.Run("NonexistentFile", func(t *testing.T) {
		nonexistent := filepath.Join(tmpDir, "nonexistent.sql")
		err := ValidateFileAccess(nonexistent)
		if err == nil {
			t.Error("Expected error for nonexistent file, got success")
		}
		// Accept either "cannot access file" or "invalid file path" error messages
		if !strings.Contains(err.Error(), "cannot access file") && !strings.Contains(err.Error(), "invalid file path") {
			t.Errorf("Expected access or path error message, got: %v", err)
		}
	})
}

// TestMemoryManagement tests that AST objects are properly released
func TestMemoryManagement(t *testing.T) {