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
119
120
121
122
123
124
125
126
127
128
129
130
|
/*
Copyright The ORAS Authors.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
package utils
import (
"fmt"
"os"
"os/exec"
"path/filepath"
. "github.com/onsi/ginkgo/v2"
"github.com/onsi/gomega"
"github.com/onsi/gomega/gexec"
"oras.land/oras-go/v2/registry"
)
// ORASPath points to the to-be-tested oras binary.
var ORASPath string
// Host points to the registry service where E2E specs will be run against.
var Host string
// FallbackHost points to the registry service where fallback E2E specs will be run against.
var FallbackHost string
// ZOTHost points to the zot service where E2E specs will be run against.
var ZOTHost string
func init() {
Host = os.Getenv(RegHostKey)
if Host == "" {
Host = "localhost:5000"
fmt.Fprintf(os.Stderr, "cannot find host name in %s, using %s instead\n", RegHostKey, Host)
}
ref := registry.Reference{
Registry: Host,
}
if err := ref.ValidateRegistry(); err != nil {
panic(err)
}
FallbackHost = os.Getenv(FallbackRegHostKey)
if FallbackHost == "" {
FallbackHost = "localhost:6000"
fmt.Fprintf(os.Stderr, "cannot find fallback host name in %s, using %s instead\n", FallbackRegHostKey, FallbackHost)
}
ref.Registry = FallbackHost
if err := ref.ValidateRegistry(); err != nil {
panic(err)
}
ZOTHost = os.Getenv(ZOTHostKey)
if ZOTHost == "" {
ZOTHost = "localhost:7000"
fmt.Fprintf(os.Stderr, "cannot find zot host name in %s, using %s instead\n", ZOTHostKey, ZOTHost)
}
ref.Registry = ZOTHost
if err := ref.ValidateRegistry(); err != nil {
panic(err)
}
// setup test data
pwd, err := os.Getwd()
if err != nil {
panic(err)
}
// to simplify debugging via `go test`, TestDataRoot cannot be passed via ginkgo argument or env var
TestDataRoot = filepath.Join(pwd, "..", "..", "testdata")
if fi, err := os.Stat(TestDataRoot); err != nil || !fi.IsDir() {
panic(fmt.Errorf("filed to find test data in %q", TestDataRoot))
}
BeforeSuite(func() {
ORASPath = os.Getenv("ORAS_PATH")
var covDumpPath string
if covDumpPath = os.Getenv("GOCOVERDIR"); covDumpPath != "" {
fmt.Printf("Coverage file dump path: %q\n", covDumpPath)
if ORASPath != "" {
fmt.Printf("Pre-built oras in %q will be ignored\n", ORASPath)
ORASPath = ""
}
// confirm the existence of dump folder
err := os.MkdirAll(covDumpPath, 0700)
gomega.Expect(err).NotTo(gomega.HaveOccurred())
}
if filepath.IsAbs(ORASPath) {
fmt.Printf("Testing based on pre-built binary locates in %q\n", ORASPath)
} else if workspacePath := os.Getenv("GITHUB_WORKSPACE"); ORASPath != "" && workspacePath != "" {
// add workspacePath as prefix, both path env should not be empty
ORASPath = filepath.Join(workspacePath, ORASPath)
ORASPath, err = filepath.Abs(ORASPath)
gomega.Expect(err).NotTo(gomega.HaveOccurred())
fmt.Printf("Testing based on pre-built binary locates in %q\n", ORASPath)
} else {
// fallback to native build to facilitate local debugging
buildArgs := []string{}
if covDumpPath != "" {
fmt.Printf("coverage instrumenting is enabled\n")
buildArgs = append(buildArgs, "-coverpkg", "oras.land/oras/cmd/oras/...,oras.land/oras/internal/...")
}
ORASPath, err = gexec.Build("oras.land/oras/cmd/oras", buildArgs...)
gomega.Expect(err).NotTo(gomega.HaveOccurred())
DeferCleanup(gexec.CleanupBuildArtifacts)
fmt.Printf("Testing based on temp binary locates in %q\n", ORASPath)
}
// Login
cmd := exec.Command(ORASPath, "login", Host, "-u", Username, "-p", Password)
gomega.Expect(cmd.Run()).ShouldNot(gomega.HaveOccurred())
cmd = exec.Command(ORASPath, "login", FallbackHost, "-u", Username, "-p", Password)
gomega.Expect(cmd.Run()).ShouldNot(gomega.HaveOccurred())
cmd = exec.Command(ORASPath, "login", ZOTHost, "-u", Username, "-p", Password)
gomega.Expect(cmd.Run()).ShouldNot(gomega.HaveOccurred())
})
}
|
