feat: take IntoIterator instead of Vec for added flexibility

testcontainers · Aug 30, 2024 · 7c85132 · 7c85132
1 parent 172073f
commit 7c85132
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/testcontainers/src/core/image/image_ext.rs b/testcontainers/src/core/image/image_ext.rs
@@ -80,10 +80,10 @@ pub trait ImageExt<I: Image> {
     fn with_privileged(self, privileged: bool) -> ContainerRequest<I>;
 
     /// Adds the capabilities to the container
-    fn with_cap_add(self, capabilities: Vec<String>) -> ContainerRequest<I>;
+    fn with_cap_add(self, capabilities: impl IntoIterator<Item = String>) -> ContainerRequest<I>;
 
     /// Drops the capabilities from the container's capabilities
-    fn with_cap_drop(self, capabilities: Vec<String>) -> ContainerRequest<I>;
+    fn with_cap_drop(self, capabilities: impl IntoIterator<Item = String>) -> ContainerRequest<I>;
 
     /// cgroup namespace mode for the container. Possible values are:
     /// - [`CgroupnsMode::Private`]: the container runs in its own private cgroup namespace
@@ -212,7 +212,7 @@ impl<RI: Into<ContainerRequest<I>>, I: Image> ImageExt<I> for RI {
         }
     }
 
-    fn with_cap_add(self, capabilities: Vec<String>) -> ContainerRequest<I> {
+    fn with_cap_add(self, capabilities: impl IntoIterator<Item = String>) -> ContainerRequest<I> {
         let mut container_req = self.into();
         container_req
             .cap_add
@@ -222,7 +222,7 @@ impl<RI: Into<ContainerRequest<I>>, I: Image> ImageExt<I> for RI {
         container_req
     }
 
-    fn with_cap_drop(self, capabilities: Vec<String>) -> ContainerRequest<I> {
+    fn with_cap_drop(self, capabilities: impl IntoIterator<Item = String>) -> ContainerRequest<I> {
         let mut container_req = self.into();
         container_req
             .cap_drop