-
Notifications
You must be signed in to change notification settings - Fork 10
/
util_e2e.go
118 lines (112 loc) · 2.94 KB
/
util_e2e.go
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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
package e2etest
import (
"context"
"fmt"
"io"
"net/http"
"os"
"os/exec"
"path/filepath"
"testing"
"ariga.io/atlas-go-sdk/atlasexec"
)
const testFixtureDir = "testdata"
func runTestWithVersions(t *testing.T, versions []string, fixtureName string, cb func(t *testing.T, ver *atlasexec.Version, wd *atlasexec.WorkingDir, tf *atlasexec.Client)) {
if os.Getenv("ATLASEXEC_E2ETEST") == "" {
t.Skip("ATLASEXEC_E2ETEST not set")
}
t.Helper()
alreadyRunVersions := map[string]bool{}
for _, av := range versions {
t.Run(fmt.Sprintf("%s-%s", fixtureName, av), func(t *testing.T) {
if alreadyRunVersions[av] {
t.Skipf("already run version %q", av)
}
alreadyRunVersions[av] = true
var execPath string
if localBinPath := os.Getenv("ATLASEXEC_E2ETEST_ATLAS_PATH"); localBinPath != "" {
execPath = localBinPath
} else {
execPath = downloadAtlas(t, av)
}
c, err := atlasexec.NewClient("", execPath)
if err != nil {
t.Fatal(err)
}
// TODO: Check that the version is the same as the one we expect.
runningVersion, err := c.Version(context.Background())
if err != nil {
t.Fatalf("unable to determine running version (expected %q): %s", av, err)
}
wd, err := atlasexec.NewWorkingDir()
if err != nil {
t.Fatal(err)
}
defer wd.Close()
if fixtureName != "" {
err := wd.CopyFS("", os.DirFS(filepath.Join(testFixtureDir, fixtureName)))
if err != nil {
t.Fatalf("error copying config file into test dir: %s", err)
}
}
err = c.WithWorkDir(wd.Path(), func(c *atlasexec.Client) (err error) {
defer func() {
if r := recover(); r != nil {
var ok bool
if err, ok = r.(error); !ok {
err = fmt.Errorf("run test failure: %v", r)
}
}
}()
cb(t, runningVersion, wd, c)
return nil
})
if err != nil {
t.Fatal(err)
}
})
}
}
func downloadAtlas(t *testing.T, version string) string {
t.Helper()
c := http.DefaultClient
req, err := http.NewRequest(http.MethodGet, "https://atlasgo.sh?test=1", nil)
if err != nil {
t.Fatal(err)
}
req.Header.Set("User-Agent", "AtlasExec/Integration-Test")
res, err := c.Do(req)
if err != nil {
t.Fatal(err)
}
defer res.Body.Close()
if res.StatusCode != http.StatusOK {
t.Fatalf("unexpected status code: %d", res.StatusCode)
}
path := filepath.Join(t.TempDir(), "installer.sh")
f, err := os.OpenFile(path, os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0777)
if err != nil {
t.Fatal(err)
}
if _, err = io.Copy(f, res.Body); err != nil {
f.Close()
t.Fatal(err)
} else if err = f.Close(); err != nil {
t.Fatal(err)
}
atlasBin := filepath.Join(t.TempDir(), "atlas")
cmd := exec.Command(path,
"--user-agent", "AtlasExec/Integration-Test",
"--output", atlasBin,
"--no-install",
)
cmd.Env = append(os.Environ(), fmt.Sprintf("ATLAS_VERSION=%s", version))
if testing.Verbose() {
cmd.Stdout = os.Stdout
cmd.Stderr = os.Stderr
}
if err = cmd.Run(); err != nil {
t.Fatal(err)
}
return atlasBin
}