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
|
package commands_test
import (
"bytes"
"context"
"flag"
"io/ioutil"
"os"
"testing"
"github.com/MichaelMure/git-bug/commands"
"github.com/MichaelMure/git-bug/repository"
"github.com/spf13/cobra"
"github.com/stretchr/testify/require"
)
var update = flag.Bool("update", false, "pass -update to the test runner to update golden files")
type testEnv struct {
cwd string
repo *repository.GoGitRepo
cmd *cobra.Command
out *bytes.Buffer
}
func newTestEnv(t *testing.T) *testEnv {
t.Helper()
cwd, err := ioutil.TempDir("", "")
require.NoError(t, err)
t.Cleanup(func() {
require.NoError(t, os.RemoveAll(cwd))
})
repo, err := repository.InitGoGitRepo(cwd, commands.GitBugNamespace)
require.NoError(t, err)
t.Cleanup(func() {
require.NoError(t, repo.Close())
})
out := new(bytes.Buffer)
cmd := commands.NewRootCommand()
cmd.SetArgs([]string{})
cmd.SetErr(out)
cmd.SetOut(out)
return &testEnv{
cwd: cwd,
repo: repo,
cmd: cmd,
out: out,
}
}
func (e *testEnv) Execute(t *testing.T) {
t.Helper()
ctx := context.WithValue(context.Background(), "cwd", e.cwd)
require.NoError(t, e.cmd.ExecuteContext(ctx))
}
|