mirror of https://github.com/containers/podman.git
add go-bindings for `hack/podman-registry`
Signed-off-by: Valentin Rothberg <rothberg@redhat.com>
This commit is contained in:
parent
973d624418
commit
dfe0579a2f
|
|
@ -0,0 +1,98 @@
|
|||
package registry
|
||||
|
||||
import (
|
||||
"strings"
|
||||
|
||||
"github.com/containers/libpod/utils"
|
||||
"github.com/pkg/errors"
|
||||
"github.com/sirupsen/logrus"
|
||||
)
|
||||
|
||||
const (
|
||||
imageKey = "PODMAN_REGISTRY_IMAGE"
|
||||
userKey = "PODMAN_REGISTRY_USER"
|
||||
passKey = "PODMAN_REGISTRY_PASS"
|
||||
portKey = "PODMAN_REGISTRY_PORT"
|
||||
)
|
||||
|
||||
var binary = "podman-registry"
|
||||
|
||||
// Registry is locally running registry.
|
||||
type Registry struct {
|
||||
// Image - container image of the registry.
|
||||
Image string
|
||||
// User - the user to authenticate against the registry.
|
||||
User string
|
||||
// Password - the accompanying password for the user.
|
||||
Password string
|
||||
// Port - the port the registry is listening to on the host.
|
||||
Port string
|
||||
// running indicates if the registry is running.
|
||||
running bool
|
||||
}
|
||||
|
||||
// Start a new registry and return it along with it's image, user, password, and port.
|
||||
func Start() (*Registry, error) {
|
||||
// Start a registry.
|
||||
out, err := utils.ExecCmd(binary, "start")
|
||||
if err != nil {
|
||||
return nil, errors.Wrapf(err, "error running %q: %s", binary, out)
|
||||
}
|
||||
|
||||
// Parse the output.
|
||||
registry := Registry{}
|
||||
for _, s := range strings.Split(out, "\n") {
|
||||
if s == "" {
|
||||
continue
|
||||
}
|
||||
spl := strings.Split(s, "=")
|
||||
if len(spl) != 2 {
|
||||
return nil, errors.Errorf("unexpected output format %q: want 'PODMAN_...=...'", s)
|
||||
}
|
||||
key := spl[0]
|
||||
val := strings.TrimSuffix(strings.TrimPrefix(spl[1], "\""), "\"")
|
||||
switch key {
|
||||
case imageKey:
|
||||
registry.Image = val
|
||||
case userKey:
|
||||
registry.User = val
|
||||
case passKey:
|
||||
registry.Password = val
|
||||
case portKey:
|
||||
registry.Port = val
|
||||
default:
|
||||
logrus.Errorf("unexpected podman-registry output: %q", s)
|
||||
}
|
||||
}
|
||||
|
||||
// Extra sanity check.
|
||||
if registry.Image == "" {
|
||||
return nil, errors.Errorf("unexpected output %q: %q missing", out, imageKey)
|
||||
}
|
||||
if registry.User == "" {
|
||||
return nil, errors.Errorf("unexpected output %q: %q missing", out, userKey)
|
||||
}
|
||||
if registry.Password == "" {
|
||||
return nil, errors.Errorf("unexpected output %q: %q missing", out, passKey)
|
||||
}
|
||||
if registry.Port == "" {
|
||||
return nil, errors.Errorf("unexpected output %q: %q missing", out, portKey)
|
||||
}
|
||||
|
||||
registry.running = true
|
||||
|
||||
return ®istry, nil
|
||||
}
|
||||
|
||||
// Stop the registry.
|
||||
func (r *Registry) Stop() error {
|
||||
// Stop a registry.
|
||||
if !r.running {
|
||||
return nil
|
||||
}
|
||||
if _, err := utils.ExecCmd(binary, "-P", r.Port, "stop"); err != nil {
|
||||
return errors.Wrapf(err, "error stopping registry (%v) with %q", *r, binary)
|
||||
}
|
||||
r.running = false
|
||||
return nil
|
||||
}
|
||||
|
|
@ -0,0 +1,40 @@
|
|||
package registry
|
||||
|
||||
import (
|
||||
"testing"
|
||||
|
||||
"github.com/hashicorp/go-multierror"
|
||||
"github.com/stretchr/testify/assert"
|
||||
"github.com/stretchr/testify/require"
|
||||
)
|
||||
|
||||
func TestStartAndStopMultipleRegistries(t *testing.T) {
|
||||
binary = "../podman-registry"
|
||||
|
||||
registries := []*Registry{}
|
||||
|
||||
// Start registries.
|
||||
var errors *multierror.Error
|
||||
for i := 0; i < 3; i++ {
|
||||
reg, err := Start()
|
||||
if err != nil {
|
||||
errors = multierror.Append(errors, err)
|
||||
continue
|
||||
}
|
||||
assert.True(t, len(reg.Image) > 0)
|
||||
assert.True(t, len(reg.User) > 0)
|
||||
assert.True(t, len(reg.Password) > 0)
|
||||
assert.True(t, len(reg.Port) > 0)
|
||||
registries = append(registries, reg)
|
||||
}
|
||||
|
||||
// Stop registries.
|
||||
for _, reg := range registries {
|
||||
// Make sure we can stop it properly.
|
||||
errors = multierror.Append(errors, reg.Stop())
|
||||
// Stopping an already stopped registry is fine as well.
|
||||
errors = multierror.Append(errors, reg.Stop())
|
||||
}
|
||||
|
||||
require.NoError(t, errors.ErrorOrNil())
|
||||
}
|
||||
Loading…
Reference in New Issue