moth/pkg/transpile/mothball_test.go

package transpile

import (
	"archive/zip"
	"io/ioutil"
	"os"
	"path"
	"runtime"
	"testing"

	"github.com/spf13/afero"
	"github.com/spf13/afero/zipfs"
)

func TestMothballsMemFs(t *testing.T) {
	static := NewFsCategory(newTestFs(), "cat1")
	if _, err := Mothball(static); err != nil {
		t.Error(err)
	}
}

func TestMothballsOsFs(t *testing.T) {
	_, testfn, _, _ := runtime.Caller(0)
	os.Chdir(path.Dir(testfn))

	fs := NewRecursiveBasePathFs(afero.NewOsFs(), "testdata")
	static := NewFsCategory(fs, "static")
	mb, err := Mothball(static)
	if err != nil {
		t.Error(err)
		return
	}

	mbr, err := zip.NewReader(mb, int64(mb.Len()))
	if err != nil {
		t.Error(err)
	}
	zfs := zipfs.New(mbr)

	if f, err := zfs.Open("puzzles.txt"); err != nil {
		t.Error(err)
	} else {
		defer f.Close()
		if buf, err := ioutil.ReadAll(f); err != nil {
			t.Error(err)
		} else if string(buf) != "" {
			t.Error("Bad puzzles.txt", string(buf))
		}
	}
}