add ability to change image tag

testcontainers · Aug 13, 2024 · f736fbb · f736fbb
1 parent 5489051
commit f736fbb
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/kafka/apache.rs b/src/kafka/apache.rs
@@ -41,6 +41,7 @@ const DEFAULT_BROKER_ID: usize = 1;
 pub struct Kafka {
     env_vars: HashMap<String, String>,
     image_name: String,
+    tag: String,
 }
 
 impl Default for Kafka {
@@ -88,6 +89,7 @@ impl Default for Kafka {
         Self {
             env_vars,
             image_name: KAFKA_NATIVE_IMAGE_NAME.to_string(),
+            tag: TAG.to_string(),
         }
     }
 }
@@ -99,6 +101,11 @@ impl Kafka {
 
         self
     }
+    /// Specify image tag
+    pub fn with_tag(mut self, tag: &str) -> Self {
+        self.tag = tag.to_string();
+        self
+    }
 }
 
 impl Image for Kafka {
@@ -107,7 +114,7 @@ impl Image for Kafka {
     }
 
     fn tag(&self) -> &str {
-        TAG
+        self.tag.as_str()
     }
 
     fn ready_conditions(&self) -> Vec<WaitFor> {