// Copyright 2018 The gVisor 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 fs

import (
	"testing"
)

// TestSplitLast tests variants of path splitting.
func TestSplitLast(t *testing.T) {
	cases := []struct {
		path string
		dir  string
		file string
	}{
		{path: "/", dir: "/", file: "."},
		{path: "/.", dir: "/", file: "."},
		{path: "/./", dir: "/", file: "."},
		{path: "/./.", dir: "/.", file: "."},
		{path: "/././", dir: "/.", file: "."},
		{path: "/./..", dir: "/.", file: ".."},
		{path: "/./../", dir: "/.", file: ".."},
		{path: "/..", dir: "/", file: ".."},
		{path: "/../", dir: "/", file: ".."},
		{path: "/../.", dir: "/..", file: "."},
		{path: "/.././", dir: "/..", file: "."},
		{path: "/../..", dir: "/..", file: ".."},
		{path: "/../../", dir: "/..", file: ".."},

		{path: "", dir: ".", file: "."},
		{path: ".", dir: ".", file: "."},
		{path: "./", dir: ".", file: "."},
		{path: "./.", dir: ".", file: "."},
		{path: "././", dir: ".", file: "."},
		{path: "./..", dir: ".", file: ".."},
		{path: "./../", dir: ".", file: ".."},
		{path: "..", dir: ".", file: ".."},
		{path: "../", dir: ".", file: ".."},
		{path: "../.", dir: "..", file: "."},
		{path: ".././", dir: "..", file: "."},
		{path: "../..", dir: "..", file: ".."},
		{path: "../../", dir: "..", file: ".."},

		{path: "/foo", dir: "/", file: "foo"},
		{path: "/foo/", dir: "/", file: "foo"},
		{path: "/foo/.", dir: "/foo", file: "."},
		{path: "/foo/./", dir: "/foo", file: "."},
		{path: "/foo/./.", dir: "/foo/.", file: "."},
		{path: "/foo/./..", dir: "/foo/.", file: ".."},
		{path: "/foo/..", dir: "/foo", file: ".."},
		{path: "/foo/../", dir: "/foo", file: ".."},
		{path: "/foo/../.", dir: "/foo/..", file: "."},
		{path: "/foo/../..", dir: "/foo/..", file: ".."},

		{path: "/foo/bar", dir: "/foo", file: "bar"},
		{path: "/foo/bar/", dir: "/foo", file: "bar"},
		{path: "/foo/bar/.", dir: "/foo/bar", file: "."},
		{path: "/foo/bar/./", dir: "/foo/bar", file: "."},
		{path: "/foo/bar/./.", dir: "/foo/bar/.", file: "."},
		{path: "/foo/bar/./..", dir: "/foo/bar/.", file: ".."},
		{path: "/foo/bar/..", dir: "/foo/bar", file: ".."},
		{path: "/foo/bar/../", dir: "/foo/bar", file: ".."},
		{path: "/foo/bar/../.", dir: "/foo/bar/..", file: "."},
		{path: "/foo/bar/../..", dir: "/foo/bar/..", file: ".."},

		{path: "foo", dir: ".", file: "foo"},
		{path: "foo", dir: ".", file: "foo"},
		{path: "foo/", dir: ".", file: "foo"},
		{path: "foo/.", dir: "foo", file: "."},
		{path: "foo/./", dir: "foo", file: "."},
		{path: "foo/./.", dir: "foo/.", file: "."},
		{path: "foo/./..", dir: "foo/.", file: ".."},
		{path: "foo/..", dir: "foo", file: ".."},
		{path: "foo/../", dir: "foo", file: ".."},
		{path: "foo/../.", dir: "foo/..", file: "."},
		{path: "foo/../..", dir: "foo/..", file: ".."},
		{path: "foo/", dir: ".", file: "foo"},
		{path: "foo/.", dir: "foo", file: "."},

		{path: "foo/bar", dir: "foo", file: "bar"},
		{path: "foo/bar/", dir: "foo", file: "bar"},
		{path: "foo/bar/.", dir: "foo/bar", file: "."},
		{path: "foo/bar/./", dir: "foo/bar", file: "."},
		{path: "foo/bar/./.", dir: "foo/bar/.", file: "."},
		{path: "foo/bar/./..", dir: "foo/bar/.", file: ".."},
		{path: "foo/bar/..", dir: "foo/bar", file: ".."},
		{path: "foo/bar/../", dir: "foo/bar", file: ".."},
		{path: "foo/bar/../.", dir: "foo/bar/..", file: "."},
		{path: "foo/bar/../..", dir: "foo/bar/..", file: ".."},
		{path: "foo/bar/", dir: "foo", file: "bar"},
		{path: "foo/bar/.", dir: "foo/bar", file: "."},
	}

	for _, c := range cases {
		dir, file := SplitLast(c.path)
		if dir != c.dir || file != c.file {
			t.Errorf("SplitLast(%q) got (%q, %q), expected (%q, %q)", c.path, dir, file, c.dir, c.file)
		}
	}
}

// TestSplitFirst tests variants of path splitting.
func TestSplitFirst(t *testing.T) {
	cases := []struct {
		path      string
		first     string
		remainder string
	}{
		{path: "/", first: "/", remainder: ""},
		{path: "/.", first: "/", remainder: "."},
		{path: "///.", first: "/", remainder: "//."},
		{path: "/.///", first: "/", remainder: "."},
		{path: "/./.", first: "/", remainder: "./."},
		{path: "/././", first: "/", remainder: "./."},
		{path: "/./..", first: "/", remainder: "./.."},
		{path: "/./../", first: "/", remainder: "./.."},
		{path: "/..", first: "/", remainder: ".."},
		{path: "/../", first: "/", remainder: ".."},
		{path: "/../.", first: "/", remainder: "../."},
		{path: "/.././", first: "/", remainder: "../."},
		{path: "/../..", first: "/", remainder: "../.."},
		{path: "/../../", first: "/", remainder: "../.."},

		{path: "", first: ".", remainder: ""},
		{path: ".", first: ".", remainder: ""},
		{path: "./", first: ".", remainder: ""},
		{path: ".///", first: ".", remainder: ""},
		{path: "./.", first: ".", remainder: "."},
		{path: "././", first: ".", remainder: "."},
		{path: "./..", first: ".", remainder: ".."},
		{path: "./../", first: ".", remainder: ".."},
		{path: "..", first: "..", remainder: ""},
		{path: "../", first: "..", remainder: ""},
		{path: "../.", first: "..", remainder: "."},
		{path: ".././", first: "..", remainder: "."},
		{path: "../..", first: "..", remainder: ".."},
		{path: "../../", first: "..", remainder: ".."},

		{path: "/foo", first: "/", remainder: "foo"},
		{path: "/foo/", first: "/", remainder: "foo"},
		{path: "/foo///", first: "/", remainder: "foo"},
		{path: "/foo/.", first: "/", remainder: "foo/."},
		{path: "/foo/./", first: "/", remainder: "foo/."},
		{path: "/foo/./.", first: "/", remainder: "foo/./."},
		{path: "/foo/./..", first: "/", remainder: "foo/./.."},
		{path: "/foo/..", first: "/", remainder: "foo/.."},
		{path: "/foo/../", first: "/", remainder: "foo/.."},
		{path: "/foo/../.", first: "/", remainder: "foo/../."},
		{path: "/foo/../..", first: "/", remainder: "foo/../.."},

		{path: "/foo/bar", first: "/", remainder: "foo/bar"},
		{path: "///foo/bar", first: "/", remainder: "//foo/bar"},
		{path: "/foo///bar", first: "/", remainder: "foo///bar"},
		{path: "/foo/bar/.", first: "/", remainder: "foo/bar/."},
		{path: "/foo/bar/./", first: "/", remainder: "foo/bar/."},
		{path: "/foo/bar/./.", first: "/", remainder: "foo/bar/./."},
		{path: "/foo/bar/./..", first: "/", remainder: "foo/bar/./.."},
		{path: "/foo/bar/..", first: "/", remainder: "foo/bar/.."},
		{path: "/foo/bar/../", first: "/", remainder: "foo/bar/.."},
		{path: "/foo/bar/../.", first: "/", remainder: "foo/bar/../."},
		{path: "/foo/bar/../..", first: "/", remainder: "foo/bar/../.."},

		{path: "foo", first: "foo", remainder: ""},
		{path: "foo", first: "foo", remainder: ""},
		{path: "foo/", first: "foo", remainder: ""},
		{path: "foo///", first: "foo", remainder: ""},
		{path: "foo/.", first: "foo", remainder: "."},
		{path: "foo/./", first: "foo", remainder: "."},
		{path: "foo/./.", first: "foo", remainder: "./."},
		{path: "foo/./..", first: "foo", remainder: "./.."},
		{path: "foo/..", first: "foo", remainder: ".."},
		{path: "foo/../", first: "foo", remainder: ".."},
		{path: "foo/../.", first: "foo", remainder: "../."},
		{path: "foo/../..", first: "foo", remainder: "../.."},
		{path: "foo/", first: "foo", remainder: ""},
		{path: "foo/.", first: "foo", remainder: "."},

		{path: "foo/bar", first: "foo", remainder: "bar"},
		{path: "foo///bar", first: "foo", remainder: "bar"},
		{path: "foo/bar/", first: "foo", remainder: "bar"},
		{path: "foo/bar/.", first: "foo", remainder: "bar/."},
		{path: "foo/bar/./", first: "foo", remainder: "bar/."},
		{path: "foo/bar/./.", first: "foo", remainder: "bar/./."},
		{path: "foo/bar/./..", first: "foo", remainder: "bar/./.."},
		{path: "foo/bar/..", first: "foo", remainder: "bar/.."},
		{path: "foo/bar/../", first: "foo", remainder: "bar/.."},
		{path: "foo/bar/../.", first: "foo", remainder: "bar/../."},
		{path: "foo/bar/../..", first: "foo", remainder: "bar/../.."},
		{path: "foo/bar/", first: "foo", remainder: "bar"},
		{path: "foo/bar/.", first: "foo", remainder: "bar/."},
	}

	for _, c := range cases {
		first, remainder := SplitFirst(c.path)
		if first != c.first || remainder != c.remainder {
			t.Errorf("SplitFirst(%q) got (%q, %q), expected (%q, %q)", c.path, first, remainder, c.first, c.remainder)
		}
	}
}

// TestIsSubpath tests the IsSubpath method.
func TestIsSubpath(t *testing.T) {
	tcs := []struct {
		// Two absolute paths.
		pathA string
		pathB string

		// Whether pathA is a subpath of pathB.
		wantIsSubpath bool

		// Relative path from pathA to pathB. Only checked if
		// wantIsSubpath is true.
		wantRelpath string
	}{
		{
			pathA:         "/foo/bar/baz",
			pathB:         "/foo",
			wantIsSubpath: true,
			wantRelpath:   "bar/baz",
		},
		{
			pathA:         "/foo",
			pathB:         "/foo/bar/baz",
			wantIsSubpath: false,
		},
		{
			pathA:         "/foo",
			pathB:         "/foo",
			wantIsSubpath: false,
		},
		{
			pathA:         "/foobar",
			pathB:         "/foo",
			wantIsSubpath: false,
		},
		{
			pathA:         "/foo",
			pathB:         "/foobar",
			wantIsSubpath: false,
		},
		{
			pathA:         "/foo",
			pathB:         "/foobar",
			wantIsSubpath: false,
		},
		{
			pathA:         "/",
			pathB:         "/foo",
			wantIsSubpath: false,
		},
		{
			pathA:         "/foo",
			pathB:         "/",
			wantIsSubpath: true,
			wantRelpath:   "foo",
		},
		{
			pathA:         "/foo/bar/../bar",
			pathB:         "/foo",
			wantIsSubpath: true,
			wantRelpath:   "bar",
		},
		{
			pathA:         "/foo/bar",
			pathB:         "/foo/../foo",
			wantIsSubpath: true,
			wantRelpath:   "bar",
		},
	}

	for _, tc := range tcs {
		gotRelpath, gotIsSubpath := IsSubpath(tc.pathA, tc.pathB)
		if gotRelpath != tc.wantRelpath || gotIsSubpath != tc.wantIsSubpath {
			t.Errorf("IsSubpath(%q, %q) got %q %t, want %q %t", tc.pathA, tc.pathB, gotRelpath, gotIsSubpath, tc.wantRelpath, tc.wantIsSubpath)
		}
	}
}