mirror of
https://github.com/containers/skopeo.git
synced 2025-10-21 19:03:44 +00:00
693de29e37
... as found by (golangci-lint run). Note: this does not add (golangci-lint run) to the Makefile to ensure the coding standard. (BTW golangci-lint currently fails on structcheck, which doesn't handle embedded structs, and that's a years-long known unfixed limitation.) Signed-off-by: Miloslav Trmač <mitr@redhat.com>
136 lines
3.0 KiB
Go
136 lines
3.0 KiB
Go
package main
|
||
|
||
import (
|
||
"fmt"
|
||
"io/ioutil"
|
||
"net/http"
|
||
"os"
|
||
"os/exec"
|
||
"path/filepath"
|
||
"time"
|
||
|
||
"gopkg.in/check.v1"
|
||
)
|
||
|
||
const (
|
||
binaryV2 = "registry-v2"
|
||
binaryV2Schema1 = "registry-v2-schema1"
|
||
)
|
||
|
||
type testRegistryV2 struct {
|
||
cmd *exec.Cmd
|
||
url string
|
||
dir string
|
||
username string
|
||
password string
|
||
email string
|
||
}
|
||
|
||
func setupRegistryV2At(c *check.C, url string, auth, schema1 bool) *testRegistryV2 {
|
||
reg, err := newTestRegistryV2At(c, url, auth, schema1)
|
||
c.Assert(err, check.IsNil)
|
||
|
||
// Wait for registry to be ready to serve requests.
|
||
for i := 0; i != 50; i++ {
|
||
if err = reg.Ping(); err == nil {
|
||
break
|
||
}
|
||
time.Sleep(100 * time.Millisecond)
|
||
}
|
||
|
||
if err != nil {
|
||
c.Fatal("Timeout waiting for test registry to become available")
|
||
}
|
||
return reg
|
||
}
|
||
|
||
func newTestRegistryV2At(c *check.C, url string, auth, schema1 bool) (*testRegistryV2, error) {
|
||
tmp, err := ioutil.TempDir("", "registry-test-")
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
template := `version: 0.1
|
||
loglevel: debug
|
||
storage:
|
||
filesystem:
|
||
rootdirectory: %s
|
||
delete:
|
||
enabled: true
|
||
http:
|
||
addr: %s
|
||
%s`
|
||
var (
|
||
htpasswd string
|
||
username string
|
||
password string
|
||
email string
|
||
)
|
||
if auth {
|
||
htpasswdPath := filepath.Join(tmp, "htpasswd")
|
||
userpasswd := "testuser:$2y$05$sBsSqk0OpSD1uTZkHXc4FeJ0Z70wLQdAX/82UiHuQOKbNbBrzs63m"
|
||
username = "testuser"
|
||
password = "testpassword"
|
||
email = "test@test.org"
|
||
if err := ioutil.WriteFile(htpasswdPath, []byte(userpasswd), os.FileMode(0644)); err != nil {
|
||
return nil, err
|
||
}
|
||
htpasswd = fmt.Sprintf(`auth:
|
||
htpasswd:
|
||
realm: basic-realm
|
||
path: %s
|
||
`, htpasswdPath)
|
||
}
|
||
confPath := filepath.Join(tmp, "config.yaml")
|
||
config, err := os.Create(confPath)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
if _, err := fmt.Fprintf(config, template, tmp, url, htpasswd); err != nil {
|
||
os.RemoveAll(tmp)
|
||
return nil, err
|
||
}
|
||
|
||
binary := binaryV2
|
||
if schema1 {
|
||
binary = binaryV2Schema1
|
||
}
|
||
|
||
cmd := exec.Command(binary, confPath)
|
||
consumeAndLogOutputs(c, fmt.Sprintf("registry-%s", url), cmd)
|
||
if err := cmd.Start(); err != nil {
|
||
os.RemoveAll(tmp)
|
||
if os.IsNotExist(err) {
|
||
c.Skip(err.Error())
|
||
}
|
||
return nil, err
|
||
}
|
||
return &testRegistryV2{
|
||
cmd: cmd,
|
||
url: url,
|
||
dir: tmp,
|
||
username: username,
|
||
password: password,
|
||
email: email,
|
||
}, nil
|
||
}
|
||
|
||
func (t *testRegistryV2) Ping() error {
|
||
// We always ping through HTTP for our test registry.
|
||
resp, err := http.Get(fmt.Sprintf("http://%s/v2/", t.url))
|
||
if err != nil {
|
||
return err
|
||
}
|
||
if resp.StatusCode != http.StatusOK && resp.StatusCode != http.StatusUnauthorized {
|
||
return fmt.Errorf("registry ping replied with an unexpected status code %d", resp.StatusCode)
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func (t *testRegistryV2) tearDown(c *check.C) {
|
||
// It’s undocumented what Kill() returns if the process has terminated,
|
||
// so we couldn’t check just for that. This is running in a container anyway…
|
||
_ = t.cmd.Process.Kill()
|
||
err := os.RemoveAll(t.dir)
|
||
c.Assert(err, check.IsNil)
|
||
}
|