1 files changed, 235 insertions, 0 deletions
diff --git a/server/inspect_test.go b/server/inspect_test.go
new file mode 100644
index 000000000..7be46c4ed
--- /dev/null
+++ b/server/inspect_test.go
@@ -0,0 +1,235 @@
+package server
+
+import (
+	"testing"
+	"time"
+
+	"k8s.io/kubernetes/pkg/kubelet/apis/cri/v1alpha1/runtime"
+
+	"github.com/containernetworking/plugins/pkg/ns"
+	"github.com/kubernetes-incubator/cri-o/libkpod"
+	"github.com/kubernetes-incubator/cri-o/libkpod/sandbox"
+	"github.com/kubernetes-incubator/cri-o/oci"
+	specs "github.com/opencontainers/runtime-spec/specs-go"
+)
+
+func TestGetInfo(t *testing.T) {
+	c := libkpod.DefaultConfig()
+	c.RootConfig.Storage = "afoobarstorage"
+	c.RootConfig.Root = "afoobarroot"
+	c.RuntimeConfig.CgroupManager = "systemd"
+	apiConfig := APIConfig{}
+	s := &Server{
+		config: Config{*c, apiConfig},
+	}
+	ci := s.getInfo()
+	if ci.CgroupDriver != "systemd" {
+		t.Fatalf("expected 'systemd', got %q", ci.CgroupDriver)
+	}
+	if ci.StorageDriver != "afoobarstorage" {
+		t.Fatalf("expected 'afoobarstorage', got %q", ci.StorageDriver)
+	}
+	if ci.StorageRoot != "afoobarroot" {
+		t.Fatalf("expected 'afoobarroot', got %q", ci.StorageRoot)
+	}
+}
+
+type mockNetNS struct {
+}
+
+func (ns mockNetNS) Close() error {
+	return nil
+}
+func (ns mockNetNS) Fd() uintptr {
+	ptr := new(uintptr)
+	return *ptr
+}
+func (ns mockNetNS) Do(toRun func(ns.NetNS) error) error {
+	return nil
+}
+func (ns mockNetNS) Set() error {
+	return nil
+}
+func (ns mockNetNS) Path() string {
+	return ""
+}
+
+func TestGetContainerInfo(t *testing.T) {
+	s := &Server{}
+	created := time.Now()
+	labels := map[string]string{
+		"io.kubernetes.container.name": "POD",
+		"io.kubernetes.test2":          "value2",
+		"io.kubernetes.test3":          "value3",
+	}
+	annotations := map[string]string{
+		"io.kubernetes.test":  "value",
+		"io.kubernetes.test1": "value1",
+	}
+	getContainerFunc := func(id string) *oci.Container {
+		container, err := oci.NewContainer("testid", "testname", "", "/container/logs", mockNetNS{}, labels, annotations, annotations, "imageName", "imageName", "imageRef", &runtime.ContainerMetadata{}, "testsandboxid", false, false, false, false, false, "/root/for/container", created, "SIGKILL")
+		if err != nil {
+			t.Fatal(err)
+		}
+		container.SetMountPoint("/var/foo/container")
+		cstate := &oci.ContainerState{}
+		cstate.State = specs.State{
+			Pid: 42,
+		}
+		cstate.Created = created
+		container.SetState(cstate)
+		return container
+	}
+	getInfraContainerFunc := func(id string) *oci.Container {
+		return nil
+	}
+	getSandboxFunc := func(id string) *sandbox.Sandbox {
+		s := &sandbox.Sandbox{}
+		s.AddIP("1.1.1.42")
+		return s
+	}
+	ci, err := s.getContainerInfo("", getContainerFunc, getInfraContainerFunc, getSandboxFunc)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if ci.CreatedTime != created.UnixNano() {
+		t.Fatalf("expected same created time %d, got %d", created.UnixNano(), ci.CreatedTime)
+	}
+	if ci.Pid != 42 {
+		t.Fatalf("expected pid 42, got %v", ci.Pid)
+	}
+	if ci.Name != "testname" {
+		t.Fatalf("expected name testname, got %s", ci.Name)
+	}
+	if ci.Image != "imageName" {
+		t.Fatalf("expected image name imageName, got %s", ci.Image)
+	}
+	if ci.Root != "/var/foo/container" {
+		t.Fatalf("expected root to be /var/foo/container, got %s", ci.Root)
+	}
+	if ci.LogPath != "/container/logs" {
+		t.Fatalf("expected log path to be /containers/logs, got %s", ci.LogPath)
+	}
+	if ci.Sandbox != "testsandboxid" {
+		t.Fatalf("expected sandbox to be testsandboxid, got %s", ci.Sandbox)
+	}
+	if ci.IP != "1.1.1.42" {
+		t.Fatalf("expected ip 1.1.1.42, got %s", ci.IP)
+	}
+	if len(ci.Annotations) == 0 {
+		t.Fatal("annotations are empty")
+	}
+	if len(ci.Labels) == 0 {
+		t.Fatal("labels are empty")
+	}
+	if len(ci.Annotations) != len(annotations) {
+		t.Fatalf("container info annotations len (%d) isn't the same as original annotations len (%d)", len(ci.Annotations), len(annotations))
+	}
+	if len(ci.Labels) != len(labels) {
+		t.Fatalf("container info labels len (%d) isn't the same as original labels len (%d)", len(ci.Labels), len(labels))
+	}
+	var found bool
+	for k, v := range annotations {
+		found = false
+		for key, value := range ci.Annotations {
+			if k == key && v == value {
+				found = true
+				break
+			}
+		}
+		if !found {
+			t.Fatalf("key %s with value %v wasn't in container info annotations", k, v)
+		}
+	}
+	for k, v := range labels {
+		found = false
+		for key, value := range ci.Labels {
+			if k == key && v == value {
+				found = true
+				break
+			}
+		}
+		if !found {
+			t.Fatalf("key %s with value %v wasn't in container info labels", k, v)
+		}
+	}
+}
+
+func TestGetContainerInfoCtrNotFound(t *testing.T) {
+	s := &Server{}
+	getContainerFunc := func(id string) *oci.Container {
+		return nil
+	}
+	getInfraContainerFunc := func(id string) *oci.Container {
+		return nil
+	}
+	getSandboxFunc := func(id string) *sandbox.Sandbox {
+		return nil
+	}
+	_, err := s.getContainerInfo("", getContainerFunc, getInfraContainerFunc, getSandboxFunc)
+	if err == nil {
+		t.Fatal("expected an error but got nothing")
+	}
+	if err != errCtrNotFound {
+		t.Fatalf("expected errCtrNotFound error, got %v", err)
+	}
+}
+
+func TestGetContainerInfoCtrStateNil(t *testing.T) {
+	s := &Server{}
+	created := time.Now()
+	labels := map[string]string{}
+	annotations := map[string]string{}
+	getContainerFunc := func(id string) *oci.Container {
+		container, err := oci.NewContainer("testid", "testname", "", "/container/logs", mockNetNS{}, labels, annotations, annotations, "imageName", "imageName", "imageRef", &runtime.ContainerMetadata{}, "testsandboxid", false, false, false, false, false, "/root/for/container", created, "SIGKILL")
+		if err != nil {
+			t.Fatal(err)
+		}
+		container.SetMountPoint("/var/foo/container")
+		container.SetState(nil)
+		return container
+	}
+	getInfraContainerFunc := func(id string) *oci.Container {
+		return nil
+	}
+	getSandboxFunc := func(id string) *sandbox.Sandbox {
+		s := &sandbox.Sandbox{}
+		s.AddIP("1.1.1.42")
+		return s
+	}
+	_, err := s.getContainerInfo("", getContainerFunc, getInfraContainerFunc, getSandboxFunc)
+	if err == nil {
+		t.Fatal("expected an error but got nothing")
+	}
+	if err != errCtrStateNil {
+		t.Fatalf("expected errCtrStateNil error, got %v", err)
+	}
+}
+
+func TestGetContainerInfoSandboxNotFound(t *testing.T) {
+	s := &Server{}
+	created := time.Now()
+	labels := map[string]string{}
+	annotations := map[string]string{}
+	getContainerFunc := func(id string) *oci.Container {
+		container, err := oci.NewContainer("testid", "testname", "", "/container/logs", mockNetNS{}, labels, annotations, annotations, "imageName", "imageName", "imageRef", &runtime.ContainerMetadata{}, "testsandboxid", false, false, false, false, false, "/root/for/container", created, "SIGKILL")
+		if err != nil {
+			t.Fatal(err)
+		}
+		container.SetMountPoint("/var/foo/container")
+		return container
+	}
+	getInfraContainerFunc := func(id string) *oci.Container {
+		return nil
+	}
+	getSandboxFunc := func(id string) *sandbox.Sandbox {
+		return nil
+	}
+	_, err := s.getContainerInfo("", getContainerFunc, getInfraContainerFunc, getSandboxFunc)
+	if err == nil {
+		t.Fatal("expected an error but got nothing")
+	}
+	if err != errSandboxNotFound {
+		t.Fatalf("expected errSandboxNotFound error, got %v", err)
+	}
+}