internal/test/txtar_test.go

// Copyright 2023 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package test

import (
	"io/fs"
	"path/filepath"
	"testing"

	"github.com/google/go-cmp/cmp"
	"golang.org/x/tools/txtar"
)

func TestWriteTxtar(t *testing.T) {
	tmp := t.TempDir()

	filename := filepath.Join(tmp, "example", "file.txtar")
	files := []txtar.File{
		{
			Name: "a.txt",
			Data: []byte("abcdefg\n"),
		},
		{
			Name: "b.txt",
			Data: []byte("hijklmnop\n"),
		},
	}
	comment := "Context about this archive"

	if err := WriteTxtar(filename, files, comment); err != nil {
		t.Fatal(err)
	}

	got, err := txtar.ParseFile(filename)
	if err != nil {
		t.Fatal(err)
	}

	want := &txtar.Archive{
		Comment: []byte(addBoilerplate(currentYear(), comment)),
		Files:   files,
	}

	if diff := cmp.Diff(want, got); diff != "" {
		t.Errorf("mismatch (-want, +got):\n%s", diff)
	}
}

func TestReadTxtar(t *testing.T) {
	archiveFilename := filepath.Join(t.TempDir(), "archive.txtar")
	fname, content := "dir/to/a/file.txt", []byte("some content\n")
	files := []txtar.File{
		{
			Name: fname,
			Data: content,
		},
	}
	if err := WriteTxtar(archiveFilename, files, ""); err != nil {
		t.Fatal(err)
	}

	fsys, err := ReadTxtarFS(archiveFilename)
	if err != nil {
		t.Fatal(err)
	}

	got, err := fs.ReadFile(fsys, fname)
	if err != nil {
		t.Error(err)
	}

	want := content
	if !cmp.Equal(got, want) {
		t.Errorf("fs.ReadFile = %s, want %s", got, want)
	}

	gotMatches, err := fs.Glob(fsys, "dir/*/*/*.txt")
	if err != nil {
		t.Errorf("fs.Glob = %v", err)
	}

	wantMatches := []string{fname}
	if !cmp.Equal(gotMatches, wantMatches) {
		t.Errorf("fs.Glob = %s, want %s", gotMatches, wantMatches)
	}
}