From b7b86bda2dea8e0cd6b381ede64d1850f3a5e393 Mon Sep 17 00:00:00 2001
From: baude <bbaude@redhat.com>
Date: Wed, 7 Aug 2019 09:09:47 -0500
Subject: [PATCH] fix port early return

when listing multiple ports on a container with podman port, an early
return was limiting results.

Fixes: #3747

Signed-off-by: baude <bbaude@redhat.com>
---
 cmd/podman/port.go    | 11 +++++++----
 test/e2e/port_test.go | 38 ++++++++++++++++++++++++++++++++++++++
 2 files changed, 45 insertions(+), 4 deletions(-)

diff --git a/cmd/podman/port.go b/cmd/podman/port.go
index 5753c8e56cb9..4e1f9642c96d 100644
--- a/cmd/podman/port.go
+++ b/cmd/podman/port.go
@@ -48,8 +48,8 @@ func init() {
 
 func portCmd(c *cliconfig.PortValues) error {
 	var (
-		userProto, containerName string
-		userPort                 int
+		userProto string
+		userPort  int
 	)
 	args := c.InputArgs
 
@@ -106,6 +106,7 @@ func portCmd(c *cliconfig.PortValues) error {
 		if err != nil {
 			return err
 		}
+		var found bool
 		// Iterate mappings
 		for _, v := range portmappings {
 			hostIP := v.HostIP
@@ -125,12 +126,14 @@ func portCmd(c *cliconfig.PortValues) error {
 			if v.ContainerPort == int32(userPort) {
 				if userProto == "" || userProto == v.Protocol {
 					fmt.Printf("%s:%d\n", hostIP, v.HostPort)
+					found = true
 					break
 				}
-			} else {
-				return errors.Errorf("No public port '%d' published for %s", userPort, containerName)
 			}
 		}
+		if !found && port != "" {
+			return errors.Errorf("failed to find published port '%d'", userPort)
+		}
 	}
 
 	return nil
diff --git a/test/e2e/port_test.go b/test/e2e/port_test.go
index 26c5fd7d05fe..b15d8e13338b 100644
--- a/test/e2e/port_test.go
+++ b/test/e2e/port_test.go
@@ -105,4 +105,42 @@ var _ = Describe("Podman port", func() {
 		result.WaitWithDefaultTimeout()
 		Expect(result.ExitCode()).To(Equal(0))
 	})
+
+	It("podman port nginx by name", func() {
+		session, cid := podmanTest.RunNginxWithHealthCheck("portcheck")
+		Expect(session.ExitCode()).To(Equal(0))
+
+		if err := podmanTest.RunHealthCheck(cid); err != nil {
+			Fail(err.Error())
+		}
+
+		result := podmanTest.Podman([]string{"port", "portcheck"})
+		result.WaitWithDefaultTimeout()
+		Expect(result.ExitCode()).To(Equal(0))
+		result.LineInOuputStartsWith("80/tcp -> 0.0.0.0:")
+	})
+
+	It("podman port multiple ports", func() {
+		// Acquire and release locks
+		lock1 := GetPortLock("5000")
+		defer lock1.Unlock()
+		lock2 := GetPortLock("5001")
+		defer lock2.Unlock()
+
+		setup := podmanTest.Podman([]string{"run", "-dt", "-p", "5000:5000", "-p", "5001:5001", ALPINE, "top"})
+		setup.WaitWithDefaultTimeout()
+		Expect(setup.ExitCode()).To(BeZero())
+
+		// Check that the first port was honored
+		result1 := podmanTest.Podman([]string{"port", "-l", "5000"})
+		result1.WaitWithDefaultTimeout()
+		Expect(result1.ExitCode()).To(BeZero())
+		Expect(result1.LineInOuputStartsWith("0.0.0.0:5000"))
+
+		// Check that the second port was honored
+		result2 := podmanTest.Podman([]string{"port", "-l", "5001"})
+		result2.WaitWithDefaultTimeout()
+		Expect(result2.ExitCode()).To(BeZero())
+		Expect(result2.LineInOuputStartsWith("0.0.0.0:5001"))
+	})
 })