mirror of
https://github.com/containers/podman.git
synced 2025-05-31 07:27:13 +08:00
Create framework for varlink endpoint integration tests
add the ability to write integration tests similar to our e2e tests for the varlink endpoints. Signed-off-by: baude <bbaude@redhat.com>
This commit is contained in:
baude
218
test/endpoint/endpoint.go
Normal file
218
test/endpoint/endpoint.go
Normal file
@ -0,0 +1,218 @@
|
||||
package endpoint
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"os"
|
||||
"os/exec"
|
||||
"strconv"
|
||||
"strings"
|
||||
"syscall"
|
||||
"time"
|
||||
|
||||
iopodman "github.com/containers/libpod/cmd/podman/varlink"
|
||||
"github.com/containers/libpod/pkg/rootless"
|
||||
. "github.com/onsi/ginkgo"
|
||||
"github.com/onsi/gomega/gexec"
|
||||
)
|
||||
|
||||
var (
|
||||
ARTIFACT_DIR = "/tmp/.artifacts"
|
||||
CGROUP_MANAGER = "systemd"
|
||||
defaultWaitTimeout = 90
|
||||
//RESTORE_IMAGES = []string{ALPINE, BB}
|
||||
INTEGRATION_ROOT string
|
||||
ImageCacheDir = "/tmp/podman/imagecachedir"
|
||||
VarlinkBinary = "/usr/bin/varlink"
|
||||
ALPINE = "docker.io/library/alpine:latest"
|
||||
infra = "k8s.gcr.io/pause:3.1"
|
||||
BB = "docker.io/library/busybox:latest"
|
||||
redis = "docker.io/library/redis:alpine"
|
||||
fedoraMinimal = "registry.fedoraproject.org/fedora-minimal:latest"
|
||||
)
|
||||
|
||||
type EndpointTestIntegration struct {
|
||||
ArtifactPath string
|
||||
CNIConfigDir string
|
||||
CgroupManager string
|
||||
ConmonBinary string
|
||||
CrioRoot string
|
||||
//Host HostOS
|
||||
ImageCacheDir string
|
||||
ImageCacheFS string
|
||||
OCIRuntime string
|
||||
PodmanBinary string
|
||||
RemoteTest bool
|
||||
RunRoot string
|
||||
SignaturePolicyPath string
|
||||
StorageOptions string
|
||||
TmpDir string
|
||||
Timings []string
|
||||
VarlinkBinary string
|
||||
VarlinkCommand *exec.Cmd
|
||||
VarlinkEndpoint string
|
||||
VarlinkSession *os.Process
|
||||
}
|
||||
|
||||
func (p *EndpointTestIntegration) StartVarlink() {
|
||||
p.startVarlink(false)
|
||||
}
|
||||
|
||||
func (p *EndpointTestIntegration) StartVarlinkWithCache() {
|
||||
p.startVarlink(true)
|
||||
}
|
||||
|
||||
func (p *EndpointTestIntegration) startVarlink(useImageCache bool) {
|
||||
var (
|
||||
counter int
|
||||
)
|
||||
if os.Geteuid() == 0 {
|
||||
os.MkdirAll("/run/podman", 0755)
|
||||
}
|
||||
varlinkEndpoint := p.VarlinkEndpoint
|
||||
//p.SetVarlinkAddress(p.VarlinkEndpoint)
|
||||
|
||||
args := []string{"varlink", "--timeout", "0", varlinkEndpoint}
|
||||
podmanOptions := getVarlinkOptions(p, args)
|
||||
if useImageCache {
|
||||
cacheOptions := []string{"--storage-opt", fmt.Sprintf("%s.imagestore=%s", p.ImageCacheFS, p.ImageCacheDir)}
|
||||
podmanOptions = append(cacheOptions, podmanOptions...)
|
||||
}
|
||||
command := exec.Command(p.PodmanBinary, podmanOptions...)
|
||||
fmt.Printf("Running: %s %s\n", p.PodmanBinary, strings.Join(podmanOptions, " "))
|
||||
command.Start()
|
||||
command.SysProcAttr = &syscall.SysProcAttr{Setpgid: true}
|
||||
p.VarlinkCommand = command
|
||||
p.VarlinkSession = command.Process
|
||||
for {
|
||||
if result := p.endpointReady(); result == 0 {
|
||||
break
|
||||
}
|
||||
fmt.Println("Waiting for varlink connection to become active", counter)
|
||||
time.Sleep(250 * time.Millisecond)
|
||||
counter++
|
||||
if counter > 40 {
|
||||
Fail("varlink endpoint never became ready")
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (p *EndpointTestIntegration) endpointReady() int {
|
||||
session := p.Varlink("GetVersion", "", false)
|
||||
return session.ExitCode()
|
||||
}
|
||||
|
||||
func (p *EndpointTestIntegration) StopVarlink() {
|
||||
var out bytes.Buffer
|
||||
var pids []int
|
||||
varlinkSession := p.VarlinkSession
|
||||
|
||||
if !rootless.IsRootless() {
|
||||
if err := varlinkSession.Kill(); err != nil {
|
||||
fmt.Fprintf(os.Stderr, "error on varlink stop-kill %q", err)
|
||||
}
|
||||
if _, err := varlinkSession.Wait(); err != nil {
|
||||
fmt.Fprintf(os.Stderr, "error on varlink stop-wait %q", err)
|
||||
}
|
||||
|
||||
} else {
|
||||
//p.ResetVarlinkAddress()
|
||||
parentPid := fmt.Sprintf("%d", p.VarlinkSession.Pid)
|
||||
pgrep := exec.Command("pgrep", "-P", parentPid)
|
||||
fmt.Printf("running: pgrep %s\n", parentPid)
|
||||
pgrep.Stdout = &out
|
||||
err := pgrep.Run()
|
||||
if err != nil {
|
||||
fmt.Fprint(os.Stderr, "unable to find varlink pid")
|
||||
}
|
||||
|
||||
for _, s := range strings.Split(out.String(), "\n") {
|
||||
if len(s) == 0 {
|
||||
continue
|
||||
}
|
||||
p, err := strconv.Atoi(s)
|
||||
if err != nil {
|
||||
fmt.Fprintf(os.Stderr, "unable to convert %s to int", s)
|
||||
}
|
||||
if p != 0 {
|
||||
pids = append(pids, p)
|
||||
}
|
||||
}
|
||||
|
||||
pids = append(pids, p.VarlinkSession.Pid)
|
||||
for _, pid := range pids {
|
||||
syscall.Kill(pid, syscall.SIGKILL)
|
||||
}
|
||||
}
|
||||
socket := strings.Split(p.VarlinkEndpoint, ":")[1]
|
||||
if err := os.Remove(socket); err != nil {
|
||||
fmt.Println(err)
|
||||
}
|
||||
}
|
||||
|
||||
type EndpointSession struct {
|
||||
*gexec.Session
|
||||
}
|
||||
|
||||
func getVarlinkOptions(p *EndpointTestIntegration, args []string) []string {
|
||||
podmanOptions := strings.Split(fmt.Sprintf("--root %s --runroot %s --runtime %s --conmon %s --cni-config-dir %s --cgroup-manager %s",
|
||||
p.CrioRoot, p.RunRoot, p.OCIRuntime, p.ConmonBinary, p.CNIConfigDir, p.CgroupManager), " ")
|
||||
if os.Getenv("HOOK_OPTION") != "" {
|
||||
podmanOptions = append(podmanOptions, os.Getenv("HOOK_OPTION"))
|
||||
}
|
||||
podmanOptions = append(podmanOptions, strings.Split(p.StorageOptions, " ")...)
|
||||
podmanOptions = append(podmanOptions, args...)
|
||||
return podmanOptions
|
||||
}
|
||||
|
||||
func (p *EndpointTestIntegration) Varlink(endpoint, message string, more bool) *EndpointSession {
|
||||
//call unix:/run/user/1000/podman/io.podman/io.podman.GetContainerStats '{"name": "foobar" }'
|
||||
var (
|
||||
command *exec.Cmd
|
||||
)
|
||||
|
||||
args := []string{"call"}
|
||||
if more {
|
||||
args = append(args, "-m")
|
||||
}
|
||||
args = append(args, []string{fmt.Sprintf("%s/io.podman.%s", p.VarlinkEndpoint, endpoint)}...)
|
||||
if len(message) > 0 {
|
||||
args = append(args, message)
|
||||
}
|
||||
command = exec.Command(p.VarlinkBinary, args...)
|
||||
session, err := gexec.Start(command, GinkgoWriter, GinkgoWriter)
|
||||
if err != nil {
|
||||
Fail(fmt.Sprintf("unable to run varlink command: %s\n%v", strings.Join(args, " "), err))
|
||||
}
|
||||
session.Wait(defaultWaitTimeout)
|
||||
return &EndpointSession{session}
|
||||
}
|
||||
|
||||
func (s *EndpointSession) OutputToString() string {
|
||||
fields := strings.Fields(fmt.Sprintf("%s", s.Out.Contents()))
|
||||
return strings.Join(fields, " ")
|
||||
}
|
||||
|
||||
func (s *EndpointSession) OutputToBytes() []byte {
|
||||
out := s.OutputToString()
|
||||
return []byte(out)
|
||||
}
|
||||
|
||||
func (s *EndpointSession) OutputToStringMap() map[string]string {
|
||||
var out map[string]string
|
||||
json.Unmarshal(s.OutputToBytes(), &out)
|
||||
return out
|
||||
}
|
||||
|
||||
func (s *EndpointSession) OutputToMapToInt() map[string]int {
|
||||
var out map[string]int
|
||||
json.Unmarshal(s.OutputToBytes(), &out)
|
||||
return out
|
||||
}
|
||||
|
||||
func (s *EndpointSession) OutputToMoreResponse() iopodman.MoreResponse {
|
||||
out := make(map[string]iopodman.MoreResponse)
|
||||
json.Unmarshal(s.OutputToBytes(), &out)
|
||||
return out["reply"]
|
||||
}
|
||||
Reference in New Issue
Block a user