add one arg example to test

Signed-off-by: yeya24 <[email protected]>

Author: yeya24

test/cli_container_port_test.go

@@ -1,8 +1,11 @@
 package main
 
 import (
+	"strings"
+
 	"github.com/alibaba/pouch/test/command"
 	"github.com/alibaba/pouch/test/environment"
+
 	"github.com/go-check/check"
 	"github.com/gotestyourself/gotestyourself/icmd"
 )
@@ -21,7 +24,7 @@ func (suite *PouchContainerPortSuite) SetUpSuite(c *check.C) {
 }
 
 // Test pouch port
-func (suite *PouchContainerCopySuite) TestPouchPort(c *check.C) {
+func (suite *PouchContainerPortSuite) TestPouchPort(c *check.C) {
 	testcase1 := "8000:8000"
 	testcase2 := "10000:10000/udp"
 	testcase3 := "127.0.0.1:5000:5000"
@@ -44,7 +47,30 @@ func (suite *PouchContainerCopySuite) TestPouchPort(c *check.C) {
 		name, "8000").Assert(c, icmd.Success)
 	c.Assert(ret.Stdout(), check.Equals, "0.0.0.0:8000\n")
 
+	// Test for only one arg.
 	ret = command.PouchRun("port",
 		name, "10000/udp").Assert(c, icmd.Success)
 	c.Assert(ret.Stdout(), check.Equals, "0.0.0.0:10000\n")
+
+	portBindingMap := map[string]string{
+		"10000/udp": "0.0.0.0:10000",
+		"5000/tcp":  "127.0.0.1:5000",
+		"8000/tcp":  "0.0.0.0:8000",
+	}
+	ret = command.PouchRun("port", name).Assert(c, icmd.Success)
+	lines := strings.Split(ret.Stdout(), "\n")
+	for _, l := range lines {
+		outputs := strings.Split(l, "->")
+		// filter out the last line
+		if len(outputs) < 2 {
+			continue
+		}
+		port := strings.TrimSpace(outputs[0])
+		actualPortBinding := strings.TrimSpace(outputs[1])
+		if expected, exist := portBindingMap[port]; exist {
+			c.Assert(expected, check.Equals, actualPortBinding)
+		} else {
+			c.Errorf("Port %s not exists for container %s", port, name)
+		}
+	}
 }

test/environment/env.go

@@ -14,7 +14,7 @@ import (
 
 var (
 	// PouchBinary is default binary
-	PouchBinary = "/usr/local/bin/pouch"
+	PouchBinary = "/home/yeya24/hub/gowork/alibaba/src/github.com/alibaba/pouch/cli/pouch"
 
 	// PouchdAddress is default pouchd address
 	PouchdAddress = "unix:///var/run/pouchd.sock"