github.com/goern/docker@v1.9.0-rc1/api/client/ps/custom_test.go (about)

     1  package ps
     2  
     3  import (
     4  	"reflect"
     5  	"strings"
     6  	"testing"
     7  	"time"
     8  
     9  	"github.com/docker/docker/api/types"
    10  	"github.com/docker/docker/pkg/stringid"
    11  )
    12  
    13  func TestContainerPsContext(t *testing.T) {
    14  	containerID := stringid.GenerateRandomID()
    15  	unix := time.Now().Unix()
    16  
    17  	var ctx containerContext
    18  	cases := []struct {
    19  		container types.Container
    20  		trunc     bool
    21  		expValue  string
    22  		expHeader string
    23  		call      func() string
    24  	}{
    25  		{types.Container{ID: containerID}, true, stringid.TruncateID(containerID), idHeader, ctx.ID},
    26  		{types.Container{ID: containerID}, false, containerID, idHeader, ctx.ID},
    27  		{types.Container{Names: []string{"/foobar_baz"}}, true, "foobar_baz", namesHeader, ctx.Names},
    28  		{types.Container{Image: "ubuntu"}, true, "ubuntu", imageHeader, ctx.Image},
    29  		{types.Container{Image: "verylongimagename"}, true, "verylongimagename", imageHeader, ctx.Image},
    30  		{types.Container{Image: "verylongimagename"}, false, "verylongimagename", imageHeader, ctx.Image},
    31  		{types.Container{
    32  			Image:   "a5a665ff33eced1e0803148700880edab4",
    33  			ImageID: "a5a665ff33eced1e0803148700880edab4269067ed77e27737a708d0d293fbf5",
    34  		},
    35  			true,
    36  			"a5a665ff33ec",
    37  			imageHeader,
    38  			ctx.Image,
    39  		},
    40  		{types.Container{
    41  			Image:   "a5a665ff33eced1e0803148700880edab4",
    42  			ImageID: "a5a665ff33eced1e0803148700880edab4269067ed77e27737a708d0d293fbf5",
    43  		},
    44  			false,
    45  			"a5a665ff33eced1e0803148700880edab4",
    46  			imageHeader,
    47  			ctx.Image,
    48  		},
    49  		{types.Container{Image: ""}, true, "<no image>", imageHeader, ctx.Image},
    50  		{types.Container{Command: "sh -c 'ls -la'"}, true, `"sh -c 'ls -la'"`, commandHeader, ctx.Command},
    51  		{types.Container{Created: unix}, true, time.Unix(unix, 0).String(), createdAtHeader, ctx.CreatedAt},
    52  		{types.Container{Ports: []types.Port{{PrivatePort: 8080, PublicPort: 8080, Type: "tcp"}}}, true, "8080/tcp", portsHeader, ctx.Ports},
    53  		{types.Container{Status: "RUNNING"}, true, "RUNNING", statusHeader, ctx.Status},
    54  		{types.Container{SizeRw: 10}, true, "10 B", sizeHeader, ctx.Size},
    55  		{types.Container{SizeRw: 10, SizeRootFs: 20}, true, "10 B (virtual 20 B)", sizeHeader, ctx.Size},
    56  		{types.Container{}, true, "", labelsHeader, ctx.Labels},
    57  		{types.Container{Labels: map[string]string{"cpu": "6", "storage": "ssd"}}, true, "cpu=6,storage=ssd", labelsHeader, ctx.Labels},
    58  		{types.Container{Created: unix}, true, "Less than a second", runningForHeader, ctx.RunningFor},
    59  	}
    60  
    61  	for _, c := range cases {
    62  		ctx = containerContext{c: c.container, trunc: c.trunc}
    63  		v := c.call()
    64  		if strings.Contains(v, ",") {
    65  			// comma-separated values means probably a map input, which won't
    66  			// be guaranteed to have the same order as our expected value
    67  			// We'll create maps and use reflect.DeepEquals to check instead:
    68  			entriesMap := make(map[string]string)
    69  			expMap := make(map[string]string)
    70  			entries := strings.Split(v, ",")
    71  			expectedEntries := strings.Split(c.expValue, ",")
    72  			for _, entry := range entries {
    73  				keyval := strings.Split(entry, "=")
    74  				entriesMap[keyval[0]] = keyval[1]
    75  			}
    76  			for _, expected := range expectedEntries {
    77  				keyval := strings.Split(expected, "=")
    78  				expMap[keyval[0]] = keyval[1]
    79  			}
    80  			if !reflect.DeepEqual(expMap, entriesMap) {
    81  				t.Fatalf("Expected entries: %v, got: %v", c.expValue, v)
    82  			}
    83  		} else if v != c.expValue {
    84  			t.Fatalf("Expected %s, was %s\n", c.expValue, v)
    85  		}
    86  
    87  		h := ctx.fullHeader()
    88  		if h != c.expHeader {
    89  			t.Fatalf("Expected %s, was %s\n", c.expHeader, h)
    90  		}
    91  	}
    92  
    93  	c1 := types.Container{Labels: map[string]string{"com.docker.swarm.swarm-id": "33", "com.docker.swarm.node_name": "ubuntu"}}
    94  	ctx = containerContext{c: c1, trunc: true}
    95  
    96  	sid := ctx.Label("com.docker.swarm.swarm-id")
    97  	node := ctx.Label("com.docker.swarm.node_name")
    98  	if sid != "33" {
    99  		t.Fatalf("Expected 33, was %s\n", sid)
   100  	}
   101  
   102  	if node != "ubuntu" {
   103  		t.Fatalf("Expected ubuntu, was %s\n", node)
   104  	}
   105  
   106  	h := ctx.fullHeader()
   107  	if h != "SWARM ID\tNODE NAME" {
   108  		t.Fatalf("Expected %s, was %s\n", "SWARM ID\tNODE NAME", h)
   109  
   110  	}
   111  
   112  	c2 := types.Container{}
   113  	ctx = containerContext{c: c2, trunc: true}
   114  
   115  	label := ctx.Label("anything.really")
   116  	if label != "" {
   117  		t.Fatalf("Expected an empty string, was %s", label)
   118  	}
   119  
   120  	ctx = containerContext{c: c2, trunc: true}
   121  	fullHeader := ctx.fullHeader()
   122  	if fullHeader != "" {
   123  		t.Fatalf("Expected fullHeader to be empty, was %s", fullHeader)
   124  	}
   125  
   126  }