plugin_test.go

package main

import (
	"archive/zip"
	"os"
	"path/filepath"
	"testing"

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

func TestPlugin(t *testing.T) {

	t.Run("zip file ./test/*", func(t *testing.T) {
		tmpDir := t.TempDir()
		t.Logf("tmpDir: %v", tmpDir)
		p := Plugin{
			Input:  []string{"./test/*"},
			Output: filepath.Join(tmpDir, "test.zip"),
		}
		p.Exec()

		expectedFiles := []string{
			"test/a.txt",
			"test/b.js",
		}

		ensureContains(t, zipFiles(t, filepath.Join(tmpDir, "test.zip")), expectedFiles)

	})

	t.Run("zip file ./test", func(t *testing.T) {
		tmpDir := t.TempDir()
		t.Logf("tmpDir: %v", tmpDir)
		p := Plugin{
			Input:  []string{"./test"},
			Output: filepath.Join(tmpDir, "dot-test.zip"),
		}
		p.Exec()

		expectedFiles := []string{
			"test/a.txt",
			"test/b.js",
			"test/foo/a.js",
			"test/foo/b.txt",
			"test/foo/bar/bar.js",
		}

		ensureContains(t, zipFiles(t, filepath.Join(tmpDir, "dot-test.zip")), expectedFiles)

	})

	t.Run("zip file ./test/**/*", func(t *testing.T) {
		tmpDir := t.TempDir()
		t.Logf("tmpDir: %v", tmpDir)
		p := Plugin{
			Input:  []string{"./test/**/*"},
			Output: filepath.Join(tmpDir, "db-star-test.zip"),
		}
		p.Exec()

		expectedFiles := []string{
			"test/a.txt",
			"test/b.js",
			"test/foo/a.js",
			"test/foo/b.txt",
			"test/foo/bar/bar.js",
		}

		ensureContains(t, zipFiles(t, filepath.Join(tmpDir, "db-star-test.zip")), expectedFiles)

	})

	t.Run("zip file ./test/**/*.js", func(t *testing.T) {
		tmpDir := t.TempDir()
		t.Logf("tmpDir: %v", tmpDir)
		p := Plugin{
			Input:  []string{"./test/**/*.js"},
			Output: filepath.Join(tmpDir, "db-star-js-test.zip"),
		}
		p.Exec()

		expectedFiles := []string{
			"test/b.js",
			"test/foo/a.js",
			"test/foo/bar/bar.js",
		}
		ensureContains(t, zipFiles(t, filepath.Join(tmpDir, "db-star-js-test.zip")), expectedFiles)

	})

	t.Run("zip file ./test/**/*.txt", func(t *testing.T) {
		tmpDir := t.TempDir()
		t.Logf("tmpDir: %v", tmpDir)
		p := Plugin{
			Input:  []string{"./test/**/*.txt"},
			Output: filepath.Join(tmpDir, "db-star-txt-test.zip"),
		}
		p.Exec()

		expectedFiles := []string{
			"test/a.txt",
			"test/foo/b.txt",
		}

		ensureContains(t, zipFiles(t, filepath.Join(tmpDir, "db-star-txt-test.zip")), expectedFiles)

	})

}

// issue#6
func TestOutput(t *testing.T) {
	t.Run("zip file output dir", func(t *testing.T) {
		tmpDir := t.TempDir()
		t.Logf("tmpDir: %v", tmpDir)
		p := Plugin{
			Input:  []string{"test-output/output-9071a1942d0d334aa224a1370d98e18015782d6f"},
			Output: filepath.Join(tmpDir, "output-9071a1942d0d334aa224a1370d98e18015782d6f.zip"),
		}
		p.Exec()

		expectedFiles := []string{
			"test-output/output-9071a1942d0d334aa224a1370d98e18015782d6f/404.html",
			"test-output/output-9071a1942d0d334aa224a1370d98e18015782d6f/begriffe.html",
			"test-output/output-9071a1942d0d334aa224a1370d98e18015782d6f/datenschutz.html",
			"test-output/output-9071a1942d0d334aa224a1370d98e18015782d6f/feed_rss_created.xml",
			"test-output/output-9071a1942d0d334aa224a1370d98e18015782d6f/feed_rss_updated.xml",
			"test-output/output-9071a1942d0d334aa224a1370d98e18015782d6f/impressum.html",
			"test-output/output-9071a1942d0d334aa224a1370d98e18015782d6f/index.html",
			"test-output/output-9071a1942d0d334aa224a1370d98e18015782d6f/sitemap.xml",
			"test-output/output-9071a1942d0d334aa224a1370d98e18015782d6f/sitemap.xml.gz",
			"test-output/output-9071a1942d0d334aa224a1370d98e18015782d6f/rechtsformen/index.html",
			"test-output/output-9071a1942d0d334aa224a1370d98e18015782d6f/rechtsformen/test/1.html",
			"test-output/output-9071a1942d0d334aa224a1370d98e18015782d6f/rechtsformen/test/2.html",
			"test-output/output-9071a1942d0d334aa224a1370d98e18015782d6f/rechtsformen/test/3.html",
			"test-output/output-9071a1942d0d334aa224a1370d98e18015782d6f/rechtsformen/test/index.html",
			"test-output/output-9071a1942d0d334aa224a1370d98e18015782d6f/search/search_index.js",
			"test-output/output-9071a1942d0d334aa224a1370d98e18015782d6f/search/search_index.json",
		}

		ensureContains(t, zipFiles(t, filepath.Join(tmpDir, "output-9071a1942d0d334aa224a1370d98e18015782d6f.zip")), expectedFiles)

	})
}

func zipFiles(t *testing.T, path string) []string {
	t.Helper()
	f, err := os.Open(path)
	if err != nil {
		t.Fatal(err)
	}
	defer func() {
		if err := f.Close(); err != nil {
			t.Fatal(err)
		}
	}()
	info, err := f.Stat()
	assert.NoError(t, err)
	r, err := zip.NewReader(f, info.Size())
	assert.NoError(t, err)
	paths := make([]string, len(r.File))
	for i, zf := range r.File {
		paths[i] = zf.Name
	}
	return paths
}

func ensureContains(t *testing.T, fileList []string, expectedFiles []string) {
	for _, expectedFile := range expectedFiles {
		assert.Contains(t, fileList, expectedFile)
	}
}

func TestIsDir(t *testing.T) {

	t.Run("It's a folder", func(t *testing.T) {
		assert.True(t, IsDir("./test/"))
		assert.True(t, IsDir("test/"))
		assert.True(t, IsDir("test"))

	})

	t.Run("It's not a folder", func(t *testing.T) {
		assert.False(t, IsDir("./main.go"))
		assert.False(t, IsDir("main.go"))
		assert.False(t, IsDir("aaa"))
	})

}

func TestGetFilePaths(t *testing.T) {
	t.Run("glob match", func(t *testing.T) {

		assert.Equal(t, []string{"test/a.txt"}, getFilePaths("./test/a.txt"))

		assert.Equal(t, []string{"test/a.txt", "test/b.js"}, getFilePaths("./test/*"))

		assert.Equal(t,
			[]string{
				"test/a.txt",
				"test/b.js",
				"test/foo/a.js",
				"test/foo/b.txt",
				"test/foo/bar/bar.js"},
			getFilePaths("./test"),
		)

		assert.Equal(t,
			[]string{
				"test/a.txt",
				"test/b.js",
				"test/foo/a.js",
				"test/foo/b.txt",
				"test/foo/bar/bar.js"},
			getFilePaths("./test/**"),
		)

		assert.Equal(t,
			[]string{
				"test/a.txt",
				"test/b.js",
				"test/foo/a.js",
				"test/foo/b.txt",
				"test/foo/bar/bar.js",
			},
			getFilePaths("./test/**/*"),
		)

		assert.Equal(t,
			[]string{
				"test/a.txt",
			},
			getFilePaths("./test/*.txt"),
		)

		assert.Equal(t,
			[]string{
				"test/a.txt",
				"test/foo/b.txt",
			},
			getFilePaths("./test/**/*.txt"),
		)

		assert.Equal(t,
			[]string{
				"test/b.js",
			},
			getFilePaths("./test/*.js"),
		)

		assert.Equal(t,
			[]string{
				"test/b.js",
				"test/foo/a.js",
				"test/foo/bar/bar.js",
			},
			getFilePaths("./test/**/*.js"),
		)

	})
}