package rootfs

import (
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestTree(t *testing.T) {
	tests := []struct {
		name       string
		paths      []string
		matchTrue  []string
		matchFalse []string
	}{
		{
			name:       "empty sequence matches nothing",
			paths:      []string{},
			matchFalse: []string{"a", "b"},
		},
		{
			name:      "directory",
			paths:     []string{"a/b"},
			matchTrue: []string{"a/b/"},
		},
		{
			name:       "a few sequences",
			paths:      []string{"a", "b", "c/b/a"},
			matchTrue:  []string{"a", "a/b/c", "c/b/a", "c/b/a/d"},
			matchFalse: []string{"c/d", "c", "c/b"},
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			tree := newTree(tt.paths...)

			for _, path := range tt.matchTrue {
				t.Run(path, func(t *testing.T) {
					assert.True(t, tree.HasPrefix(path),
						"expected %s to be a prefix of %s", path, tree)
				})
			}

			for _, path := range tt.matchFalse {
				t.Run(path, func(t *testing.T) {
					assert.False(t, tree.HasPrefix(path),
						"expected %s to not be a prefix of %s", path, tree)
				})
			}
		})
	}
}

func TestTreeMerge(t *testing.T) {
	tree1 := newTree("bin/ar", "var/cache/apt")
	tree2 := newTree("bin/ar", "bin/busybox", "usr/share/doc")
	tree1.Merge(tree2)
	assert.Equal(t, "./bin/ar:./bin/busybox:./usr/share/doc:./var/cache/apt", tree1.String())
	assert.Equal(t, "./bin/ar:./bin/busybox:./usr/share/doc", tree2.String())
}

func TestString(t *testing.T) {
	tests := []struct {
		name    string
		paths   []string
		wantStr string
	}{
		{
			name:    "empty",
			paths:   []string{},
			wantStr: "<empty>",
		},
		{
			name:    "simple path",
			paths:   []string{"a/b/c"},
			wantStr: "./a/b/c",
		},
		{
			name:    "duplicate paths",
			paths:   []string{"a/a", "a//a"},
			wantStr: "./a/a",
		},
		{
			name:    "a few sequences",
			paths:   []string{"bin/ar", "bin/busybox", "var/cache/apt/archives"},
			wantStr: "./bin/ar:./bin/busybox:./var/cache/apt/archives",
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			tree := newTree(tt.paths...)
			assert.Equal(t, tt.wantStr, tree.String())
		})
	}
}