feat: add ability to convert ContainerPort to u16 (#664)

Add `as_u16` method to `ContainerPort`
testcontainers · Jun 16, 2024 · c97f8be · c97f8be
1 parent 0afec13
commit c97f8be
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/testcontainers/src/core/ports.rs b/testcontainers/src/core/ports.rs
@@ -131,6 +131,15 @@ impl TryFrom<PortMap> for Ports {
     }
 }
 
+impl ContainerPort {
+    /// Returns the port number, regardless of the protocol.
+    pub fn as_u16(self) -> u16 {
+        match self {
+            ContainerPort::Tcp(port) | ContainerPort::Udp(port) | ContainerPort::Sctp(port) => port,
+        }
+    }
+}
+
 impl IntoContainerPort for u16 {
     fn tcp(self) -> ContainerPort {
         ContainerPort::Tcp(self)