package mounts

import (
	"os"
	"path/filepath"
	"reflect"
	"testing"
)

func TestParseProc(t *testing.T) {
	// Real /proc/mounts shape, including a pseudo fs and an octal-escaped space.
	data := `proc /proc proc rw,nosuid,nodev,noexec 0 0
/dev/sda1 / ext4 rw,relatime 0 0
/dev/sdb1 /mnt/my\040disk ext4 rw 0 0
`
	got := parseProc(data)
	want := []Mount{
		{Device: "proc", Mountpoint: "/proc", FSType: "proc", Options: "rw,nosuid,nodev,noexec"},
		{Device: "/dev/sda1", Mountpoint: "/", FSType: "ext4", Options: "rw,relatime"},
		{Device: "/dev/sdb1", Mountpoint: "/mnt/my disk", FSType: "ext4", Options: "rw"},
	}
	if !reflect.DeepEqual(got, want) {
		t.Errorf("parseProc:\n got %+v\nwant %+v", got, want)
	}
}

func TestUnescape(t *testing.T) {
	cases := map[string]string{
		`/mnt/my\040disk`: "/mnt/my disk",
		`/no/escapes`:     "/no/escapes",
		`tab\011here`:     "tab\there",
		`back\134slash`:   `back\slash`,
	}
	for in, want := range cases {
		if got := Unescape(in); got != want {
			t.Errorf("Unescape(%q) = %q, want %q", in, got, want)
		}
	}
}

func TestProc(t *testing.T) {
	dir := t.TempDir()
	f := filepath.Join(dir, "mounts")
	if err := os.WriteFile(f, []byte("/dev/sda1 / ext4 rw 0 0\n"), 0644); err != nil {
		t.Fatal(err)
	}
	old := procMounts
	procMounts = f
	defer func() { procMounts = old }()

	got, err := Proc()
	if err != nil {
		t.Fatal(err)
	}
	if len(got) != 1 || got[0].Mountpoint != "/" {
		t.Errorf("Proc() = %+v", got)
	}
}