// SPDX-License-Identifier: GPL-3.0-or-later

package xdg

import "testing"

func TestBaseDirs(t *testing.T) {
	t.Setenv("HOME", "/home/u")
	t.Setenv("XDG_CONFIG_HOME", "")
	t.Setenv("XDG_STATE_HOME", "relative/ignored")
	t.Setenv("XDG_DATA_HOME", "/data/")
	if got := ConfigHome(); got != "/home/u/.config" {
		t.Errorf("ConfigHome() = %q", got)
	}
	if got := StateHome(); got != "/home/u/.local/state" {
		t.Errorf("StateHome() = %q, a relative value must be ignored", got)
	}
	if got := DataHome(); got != "/data" {
		t.Errorf("DataHome() = %q", got)
	}
}

// TestHomeTrailingSlash is item H: Home() must clean its result, or a
// trailing slash from $HOME breaks Abbrev's prefix check.
func TestHomeTrailingSlash(t *testing.T) {
	t.Setenv("HOME", "/home/u/")
	if got := Abbrev("/home/u/x"); got != "~/x" {
		t.Errorf("Abbrev(/home/u/x) = %q, want ~/x", got)
	}
	if got := Expand("~/x"); got != "/home/u/x" {
		t.Errorf("Expand(~/x) = %q, want /home/u/x", got)
	}
}

func TestExpandAbbrev(t *testing.T) {
	t.Setenv("HOME", "/home/u")
	expand := map[string]string{
		"~":        "/home/u",
		"~/d/x":    "/home/u/d/x",
		"~other/x": "~other/x",
		"/abs":     "/abs",
		"rel/x":    "rel/x",
	}
	for in, want := range expand {
		if got := Expand(in); got != want {
			t.Errorf("Expand(%q) = %q, want %q", in, got, want)
		}
	}
	abbrev := map[string]string{
		"/home/u":     "~",
		"/home/u/d/x": "~/d/x",
		"/home/ux/y":  "/home/ux/y",
		"/etc":        "/etc",
	}
	for in, want := range abbrev {
		if got := Abbrev(in); got != want {
			t.Errorf("Abbrev(%q) = %q, want %q", in, got, want)
		}
	}
}