object: Add tests

1 files changed, 125 insertions, 0 deletions

diff --git a/object/tree_helpers_test.go b/object/tree_helpers_test.go
new file mode 100644
index 00000000..ce7160d6
--- /dev/null
+++ b/object/tree_helpers_test.go
@@ -0,0 +1,125 @@
+package object_test
+
+import (
+	"bytes"
+	"fmt"
+	"strings"
+	"testing"
+
+	"codeberg.org/lindenii/furgit/internal/testgit"
+	"codeberg.org/lindenii/furgit/object"
+)
+
+func mktreeTypeFromMode(t *testing.T, mode object.FileMode) string {
+	t.Helper()
+	switch mode {
+	case object.FileModeDir:
+		return "tree"
+	case object.FileModeRegular, object.FileModeExecutable, object.FileModeSymlink:
+		return "blob"
+	case object.FileModeGitlink:
+		return "commit"
+	default:
+		t.Fatalf("unsupported file mode: %o", mode)
+		return ""
+	}
+}
+
+func buildGitMktreeInput(entries []object.TreeEntry) string {
+	var b strings.Builder
+	for _, e := range entries {
+		fmt.Fprintf(&b, "%o %s %s\t%s\n", e.Mode, mktreeTypeFromModeNoTB(e.Mode), e.ID.String(), e.Name)
+	}
+	return b.String()
+}
+
+func mktreeTypeFromModeNoTB(mode object.FileMode) string {
+	switch mode {
+	case object.FileModeDir:
+		return "tree"
+	case object.FileModeRegular, object.FileModeExecutable, object.FileModeSymlink:
+		return "blob"
+	case object.FileModeGitlink:
+		return "commit"
+	default:
+		return ""
+	}
+}
+
+func gitLsTreeNames(out []byte) [][]byte {
+	if len(out) == 0 {
+		return nil
+	}
+	parts := bytes.Split(out, []byte{0})
+	if len(parts) > 0 && len(parts[len(parts)-1]) == 0 {
+		parts = parts[:len(parts)-1]
+	}
+	names := make([][]byte, 0, len(parts))
+	for _, name := range parts {
+		names = append(names, append([]byte(nil), name...))
+	}
+	return names
+}
+
+func adversarialRootEntries(t *testing.T, repo *testgit.TestRepo) []object.TreeEntry {
+	t.Helper()
+
+	blobA := repo.HashObject(t, "blob", []byte("blob-A\n"))
+	blobB := repo.HashObject(t, "blob", []byte("blob-B\n"))
+	blobC := repo.HashObject(t, "blob", []byte("blob-C\n"))
+
+	subDirA := repo.Mktree(t,
+		fmt.Sprintf("100644 blob %s\tnested-a.txt\n100755 blob %s\trun-a.sh\n", blobA.String(), blobB.String()))
+	subDirB := repo.Mktree(t,
+		fmt.Sprintf("100644 blob %s\tnested-b.txt\n100644 blob %s\tz-last\n", blobB.String(), blobC.String()))
+	subDirC := repo.Mktree(t,
+		fmt.Sprintf("120000 blob %s\tlink-c\n100644 blob %s\tchild\n", blobC.String(), blobA.String()))
+	subDirD := repo.Mktree(t,
+		fmt.Sprintf("100644 blob %s\tleaf\n", blobA.String()))
+
+	return []object.TreeEntry{
+		{Mode: object.FileModeRegular, Name: []byte("z"), ID: blobA},
+		{Mode: object.FileModeRegular, Name: []byte("A"), ID: blobB},
+		{Mode: object.FileModeRegular, Name: []byte("aa"), ID: blobC},
+		{Mode: object.FileModeRegular, Name: []byte("a0"), ID: blobA},
+		{Mode: object.FileModeRegular, Name: []byte("a-"), ID: blobB},
+		{Mode: object.FileModeRegular, Name: []byte("a."), ID: blobC},
+		{Mode: object.FileModeRegular, Name: []byte("a_"), ID: blobA},
+		{Mode: object.FileModeRegular, Name: []byte("a~"), ID: blobB},
+		{Mode: object.FileModeRegular, Name: []byte("Z"), ID: blobC},
+		{Mode: object.FileModeRegular, Name: []byte("0"), ID: blobA},
+		{Mode: object.FileModeRegular, Name: []byte("9"), ID: blobB},
+		{Mode: object.FileModeRegular, Name: []byte("00"), ID: blobC},
+		{Mode: object.FileModeRegular, Name: []byte("这是一些非 ASCII 的字符"), ID: blobC},
+		{Mode: object.FileModeRegular, Name: []byte("𲰼是新进入 Unicode 的字符"), ID: blobC},
+		{Mode: object.FileModeRegular, Name: []byte("Emoji 👀"), ID: blobC},
+		{Mode: object.FileModeRegular, Name: []byte("_"), ID: blobA},
+		{Mode: object.FileModeRegular, Name: []byte("-dash"), ID: blobB},
+		{Mode: object.FileModeRegular, Name: []byte("dot.file"), ID: blobC},
+		{Mode: object.FileModeRegular, Name: []byte(".hidden"), ID: blobA},
+		{Mode: object.FileModeRegular, Name: []byte("CAPS"), ID: blobB},
+		{Mode: object.FileModeRegular, Name: []byte("caps"), ID: blobC},
+		{Mode: object.FileModeRegular, Name: []byte("mixCase"), ID: blobA},
+		{Mode: object.FileModeRegular, Name: []byte("name with space"), ID: blobB},
+		{Mode: object.FileModeRegular, Name: []byte("name-with-dash"), ID: blobC},
+		{Mode: object.FileModeRegular, Name: []byte("name.with.dot"), ID: blobA},
+		{Mode: object.FileModeRegular, Name: []byte("name_with_underscore"), ID: blobB},
+		{Mode: object.FileModeRegular, Name: []byte("tilde~name"), ID: blobC},
+		{Mode: object.FileModeRegular, Name: []byte("brace{name}"), ID: blobA},
+		{Mode: object.FileModeRegular, Name: []byte("plus+name"), ID: blobB},
+		{Mode: object.FileModeRegular, Name: []byte("equal=name"), ID: blobC},
+		{Mode: object.FileModeRegular, Name: []byte("at@name"), ID: blobA},
+		{Mode: object.FileModeRegular, Name: []byte("percent%name"), ID: blobB},
+		{Mode: object.FileModeRegular, Name: []byte("caret^name"), ID: blobC},
+		{Mode: object.FileModeRegular, Name: []byte("comma,name"), ID: blobA},
+		{Mode: object.FileModeRegular, Name: []byte("semi;name"), ID: blobB},
+		{Mode: object.FileModeRegular, Name: []byte("paren(name)"), ID: blobC},
+		{Mode: object.FileModeRegular, Name: []byte("bracket[name]"), ID: blobA},
+		{Mode: object.FileModeExecutable, Name: []byte("exec.sh"), ID: blobB},
+		{Mode: object.FileModeSymlink, Name: []byte("sym.link"), ID: blobC},
+		{Mode: object.FileModeDir, Name: []byte("dir"), ID: subDirA},
+		{Mode: object.FileModeDir, Name: []byte("dir0"), ID: subDirB},
+		{Mode: object.FileModeDir, Name: []byte("dir.space"), ID: subDirC},
+		{Mode: object.FileModeDir, Name: []byte("x"), ID: subDirD},
+	}
+}