Function assertPortRange

integration-cli/docker_cli_port_test.go:193–236

(ctx context.Context, id string, expectedTCP, expectedUDP []int)

}

func assertPortRange(ctx context.Context, id string, expectedTCP, expectedUDP []int) error {
	apiClient := testEnv.APIClient()
	res, err := apiClient.ContainerInspect(ctx, id, client.ContainerInspectOptions{})
	if err != nil {
		return err
	}

	var validTCP, validUDP bool
	for port, binding := range res.Container.NetworkSettings.Ports {
		if port.Proto() == "tcp" && len(expectedTCP) == 0 {
			continue
		}
		if port.Proto() == "udp" && len(expectedTCP) == 0 {
			continue
		}

		for _, b := range binding {
			port, err := strconv.Atoi(b.HostPort)
			if err != nil {
				return err
			}

			if len(expectedTCP) > 0 {
				if port < expectedTCP[0] || port > expectedTCP[1] {
					return fmt.Errorf("tcp port (%d) not in range expected range %d-%d", port, expectedTCP[0], expectedTCP[1])
				}
				validTCP = true
			}
			if len(expectedUDP) > 0 {
				if port < expectedUDP[0] || port > expectedUDP[1] {
					return fmt.Errorf("udp port (%d) not in range expected range %d-%d", port, expectedUDP[0], expectedUDP[1])
				}
				validUDP = true
			}
		}
	}
	if !validTCP {
		return errors.New("tcp port not found")
	}
	if !validUDP {
		return errors.New("udp port not found")
	}
	return nil
}

func stopRemoveContainer(id string, t *testing.T) {
	cli.DockerCmd(t, "rm", "-f", id)