hypergene/hypergene_test.go at master · pattack/hypergene · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
package hypergene_test

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

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

	"github.com/pattack/hypergene"
	"github.com/pattack/hypergene/exporters/dummy"
)

func TestExport(t *testing.T) {
	var hg hypergene.HyperGene

	wdir, err := os.MkdirTemp("", "hypergene-test-")
	require.NoError(t, err, "unable to create working dir")
	defer func() {
		_ = os.RemoveAll(wdir)
	}()

	schDir, err := os.MkdirTemp(wdir, "sch-")
	require.NoError(t, err, "unable to create scheme dir")

	// fill in scheme files

	err = os.MkdirAll(filepath.Join(schDir, "cmd", "pouyan"), 0o700)
	require.NoError(t, err, "unable to create scheme dir")

	require.NoError(t, os.WriteFile(
		filepath.Join(schDir, "cmd", "pouyan", "main.go.tpl"),
		[]byte(`package main

func main() {
  println("Hello World!")
}
`),
		0o700,
	))

	outDir, err := os.MkdirTemp(wdir, "out-")
	require.NoError(t, err, "unable to output dir")

	err = hg.Export(context.Background(), dummy.Exporter{}, schDir, outDir)
	require.NoError(t, err)

	dee := make([]string, 0, 10)
	outFiles := make(map[string]fs.DirEntry)
	err = filepath.WalkDir(outDir, func(path string, d fs.DirEntry, serr error) error {
		relPath, _ := filepath.Rel(outDir, path)
		dee = append(dee, relPath)
		outFiles[path] = d

		return nil
	})
	require.NoError(t, err)

	assert.ElementsMatch(t, dee, []string{
		".",
		"cmd",
		filepath.FromSlash("cmd/pouyan"),
		filepath.FromSlash("cmd/pouyan/main.go"),
	})

	bb, err := os.ReadFile(filepath.Join(outDir, filepath.FromSlash("cmd/pouyan/main.go")))
	require.NoError(t, err)
	assert.Equal(t, string(bb), `package main

func main() {
  println("Hello World!")
}
`)
}