forked from containers/podman
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request containers#3824 from baude/varlinkendpointtest
Create framework for varlink endpoint integration tests
- Loading branch information
Showing
12 changed files
with
718 additions
and
1 deletion.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
|
|
@@ -22,3 +22,4 @@ __pycache__ | |
| test/e2e/e2e.coverprofile | ||
| /podman*zip | ||
| podman*.tar.gz | ||
| .idea* | ||
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,22 @@ | ||
| package endpoint | ||
|
|
||
| import "encoding/json" | ||
|
|
||
| var ( | ||
| STORAGE_FS = "vfs" | ||
| STORAGE_OPTIONS = "--storage-driver vfs" | ||
| ROOTLESS_STORAGE_FS = "vfs" | ||
| ROOTLESS_STORAGE_OPTIONS = "--storage-driver vfs" | ||
| CACHE_IMAGES = []string{ALPINE, BB, fedoraMinimal, nginx, redis, registry, infra, labels} | ||
| nginx = "quay.io/libpod/alpine_nginx:latest" | ||
| BB_GLIBC = "docker.io/library/busybox:glibc" | ||
| registry = "docker.io/library/registry:2" | ||
| labels = "quay.io/libpod/alpine_labels:latest" | ||
| ) | ||
|
|
||
| func makeNameMessage(name string) string { | ||
| n := make(map[string]string) | ||
| n["name"] = name | ||
| b, _ := json.Marshal(n) | ||
| return string(b) | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -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"] | ||
| } |
Oops, something went wrong.