env_testing.go

 1package execenv
 2
 3import (
 4	"bytes"
 5	"fmt"
 6	"testing"
 7
 8	"github.com/stretchr/testify/require"
 9
10	"github.com/MichaelMure/git-bug/cache"
11	"github.com/MichaelMure/git-bug/repository"
12)
13
14type TestOut struct {
15	*bytes.Buffer
16}
17
18func (te *TestOut) Printf(format string, a ...interface{}) {
19	_, _ = fmt.Fprintf(te.Buffer, format, a...)
20}
21
22func (te *TestOut) Print(a ...interface{}) {
23	_, _ = fmt.Fprint(te.Buffer, a...)
24}
25
26func (te *TestOut) Println(a ...interface{}) {
27	_, _ = fmt.Fprintln(te.Buffer, a...)
28}
29
30func NewTestEnv(t *testing.T) *Env {
31	t.Helper()
32
33	repo := repository.CreateGoGitTestRepo(t, false)
34
35	buf := new(bytes.Buffer)
36
37	backend, err := cache.NewRepoCacheNoEvents(repo)
38	require.NoError(t, err)
39
40	t.Cleanup(func() {
41		backend.Close()
42	})
43
44	return &Env{
45		Repo:    repo,
46		Backend: backend,
47		Out:     &TestOut{buf},
48		Err:     &TestOut{buf},
49	}
50}