mirror of
https://github.com/kopia/kopia.git
synced 2026-05-17 03:04:31 -04:00
e456f78fa2
- test rclone after canceling starting context - execute rclone with non-cancelling context - create initial SFTP connection with non-canceling context - nit: use context.WithoutCancel instead of Background - Fixes #5039 - Ref #4972
85 lines
2.1 KiB
Go
85 lines
2.1 KiB
Go
package testutil
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"net/url"
|
|
"os"
|
|
"os/exec"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/kopia/kopia/internal/testlogging"
|
|
)
|
|
|
|
// RunDockerAndGetOutputOrSkip runs Docker and returns the output as a string.
|
|
func RunDockerAndGetOutputOrSkip(tb testing.TB, args ...string) string {
|
|
tb.Helper()
|
|
|
|
return runDockerAndGetOutputOrSkip(testlogging.Context(tb), tb, args...)
|
|
}
|
|
|
|
func runDockerAndGetOutputOrSkip(ctx context.Context, tb testing.TB, args ...string) string {
|
|
tb.Helper()
|
|
tb.Logf("running docker %v", args)
|
|
|
|
var stderr bytes.Buffer
|
|
|
|
c := exec.CommandContext(ctx, "docker", args...)
|
|
c.Stderr = &stderr
|
|
|
|
out, err := c.Output()
|
|
if err != nil {
|
|
// skip or fail hard when running in CI environment.
|
|
if os.Getenv("CI") == "" {
|
|
tb.Skipf("unable to run docker: %v %s (stderr %s)", err, out, stderr.String())
|
|
} else {
|
|
tb.Fatalf("unable to run docker: %v %s (stderr %s)", err, out, stderr.String())
|
|
}
|
|
}
|
|
|
|
return strings.TrimSpace(string(out))
|
|
}
|
|
|
|
// RunContainerAndKillOnCloseOrSkip runs "docker run" and ensures that resulting container is killed
|
|
// on exit. Returns containerID.
|
|
func RunContainerAndKillOnCloseOrSkip(t *testing.T, args ...string) string {
|
|
t.Helper()
|
|
|
|
containerID := RunDockerAndGetOutputOrSkip(t, args...)
|
|
|
|
t.Cleanup(func() {
|
|
// t.Context() is canceled by the time cleanup executes, so it cannot be used here
|
|
runDockerAndGetOutputOrSkip(context.WithoutCancel(t.Context()), t, "kill", containerID)
|
|
})
|
|
|
|
return containerID
|
|
}
|
|
|
|
// GetContainerMappedPortAddress returns <host>:<port> that can be used to connect to
|
|
// a given container and private port.
|
|
func GetContainerMappedPortAddress(t *testing.T, containerID, privatePort string) string {
|
|
t.Helper()
|
|
|
|
portMapping := RunDockerAndGetOutputOrSkip(t, "port", containerID, privatePort)
|
|
|
|
p := strings.LastIndex(portMapping, ":")
|
|
if p < 0 {
|
|
t.Fatalf("invalid port mapping: %v", portMapping)
|
|
}
|
|
|
|
colonPort := portMapping[p:]
|
|
|
|
dockerhost := os.Getenv("DOCKER_HOST")
|
|
if dockerhost == "" {
|
|
return "localhost" + colonPort
|
|
}
|
|
|
|
u, err := url.Parse(dockerhost)
|
|
if err != nil {
|
|
t.Fatalf("unable to parse DOCKER_HOST: %v", err)
|
|
}
|
|
|
|
return u.Hostname() + colonPort
|
|
}
|