61 lines
1.6 KiB
Go
61 lines
1.6 KiB
Go
|
package testutils
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"os"
|
||
|
|
"os/exec"
|
||
|
|
"path/filepath"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Output directory for testing and benchmark artifacts
|
||
|
|
func GetTestOutDir() string {
|
||
|
|
out, _ := exec.Command("git", "rev-parse", "--show-toplevel").CombinedOutput()
|
||
|
|
repoRoot := string(out)
|
||
|
|
prefix := filepath.Join(strings.TrimSpace(repoRoot), "output")
|
||
|
|
return prefix
|
||
|
|
}
|
||
|
|
|
||
|
|
var (
|
||
|
|
ArchivesDir = filepath.Join("test-artifacts", "archives")
|
||
|
|
BundlesRoot = filepath.Join("test-artifacts", "oci-bundles")
|
||
|
|
OutputDirFormat = filepath.Join("test-artifacts", "runs", "%s")
|
||
|
|
RefOciSpecsPath = filepath.Join(BundlesRoot, "config.json")
|
||
|
|
StateDir = "/run/containerd-bench-test"
|
||
|
|
)
|
||
|
|
|
||
|
|
// untarRootfs untars the given `source` tarPath into `destination/rootfs`
|
||
|
|
func untarRootfs(source string, destination string) error {
|
||
|
|
rootfs := filepath.Join(destination, "rootfs")
|
||
|
|
|
||
|
|
if err := os.MkdirAll(rootfs, 0755); err != nil {
|
||
|
|
fmt.Println("untarRootfs os.MkdirAll failed with err %v", err)
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
tar := exec.Command("tar", "-C", rootfs, "-xf", source)
|
||
|
|
return tar.Run()
|
||
|
|
}
|
||
|
|
|
||
|
|
func GenerateReferenceSpecs(destination string) error {
|
||
|
|
if _, err := os.Stat(filepath.Join(destination, "config.json")); err == nil {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
specs := exec.Command("runc", "spec")
|
||
|
|
specs.Dir = destination
|
||
|
|
return specs.Run()
|
||
|
|
}
|
||
|
|
|
||
|
|
func CreateBundle(source, name string) error {
|
||
|
|
bundlePath := filepath.Join(BundlesRoot, name)
|
||
|
|
|
||
|
|
if err := untarRootfs(filepath.Join(ArchivesDir, source+".tar"), bundlePath); err != nil {
|
||
|
|
return fmt.Errorf("Failed to untar %s.tar: %v", source, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func CreateBusyboxBundle(name string) error {
|
||
|
|
return CreateBundle("busybox", name)
|
||
|
|
}
|