Add integration test support for volume mounts

libcnb-test/src/container_config.rs

@@ -1,4 +1,5 @@
 use std::collections::{HashMap, HashSet};
+use std::path::{Path, PathBuf};
 
 /// Config used when starting a container.
 ///
@@ -31,6 +32,7 @@ pub struct ContainerConfig {
     pub(crate) command: Option<Vec<String>>,
     pub(crate) env: HashMap<String, String>,
     pub(crate) exposed_ports: HashSet<u16>,
+    pub(crate) volumes: HashMap<PathBuf, PathBuf>,
 }
 
 impl ContainerConfig {
@@ -169,6 +171,37 @@ impl ContainerConfig {
         self
     }
 
+    /// Mounts a named volume `source` into the container `destination`. Useful for integration
+    /// tests that depend on persistent storage shared between container executions.
+    ///
+    /// See: [Docker CLI, Mount Volume](https://docs.docker.com/reference/cli/docker/container/run/#volume)
+    ///
+    /// # Example
+    /// ```no_run
+    /// use libcnb_test::{BuildConfig, ContainerConfig, TestRunner};
+    /// use std::path::PathBuf;
+    ///
+    /// TestRunner::default().build(
+    ///     BuildConfig::new("heroku/builder:22", "tests/fixtures/app"),
+    ///     |context| {
+    ///         // ...
+    ///         context.start_container(
+    ///             ContainerConfig::new().volume(PathBuf::from("/shared/cache"), PathBuf::from("/workspace/cache")),
+    ///             |container| {
+    ///                 // ...
+    ///             },
+    ///         );
+    ///     },
+    /// );
+    /// ```
+    pub fn volume<P: AsRef<Path>>(&mut self, source: P, destination: P) -> &mut Self {
+        self.volumes.insert(
+            source.as_ref().to_path_buf(),
+            destination.as_ref().to_path_buf(),
+        );
+        self
+    }
+
     /// Adds or updates multiple environment variable mappings for the container.
     ///
     /// # Example

libcnb-test/src/docker.rs

@@ -1,4 +1,5 @@
 use std::collections::{BTreeMap, BTreeSet};
+use std::path::PathBuf;
 use std::process::Command;
 
 /// Represents a `docker run` command.
@@ -13,6 +14,7 @@ pub(crate) struct DockerRunCommand {
     image_name: String,
     platform: Option<String>,
     remove: bool,
+    volumes: BTreeMap<PathBuf, PathBuf>,
 }
 
 impl DockerRunCommand {
@@ -27,6 +29,7 @@ impl DockerRunCommand {
             image_name: image_name.into(),
             platform: None,
             remove: false,
+            volumes: BTreeMap::new(),
         }
     }
 
@@ -67,6 +70,15 @@ impl DockerRunCommand {
         self.remove = remove;
         self
     }
+
+    pub(crate) fn volume(
+        &mut self,
+        source: impl Into<PathBuf>,
+        destination: impl Into<PathBuf>,
+    ) -> &mut Self {
+        self.volumes.insert(source.into(), destination.into());
+        self
+    }
 }
 
 impl From<DockerRunCommand> for Command {
@@ -98,6 +110,17 @@ impl From<DockerRunCommand> for Command {
             command.args(["--publish", &format!("127.0.0.1::{port}")]);
         }
 
+        for (source, destination) in &docker_run_command.volumes {
+            command.args([
+                "--volume",
+                &format!(
+                    "{}:{}",
+                    source.to_string_lossy(),
+                    destination.to_string_lossy()
+                ),
+            ]);
+        }
+
         command.arg(docker_run_command.image_name);
 
         if let Some(container_command) = docker_run_command.command {
@@ -315,6 +338,8 @@ mod tests {
         docker_run_command.expose_port(55555);
         docker_run_command.platform("linux/amd64");
         docker_run_command.remove(true);
+        docker_run_command.volume(PathBuf::from("./test-cache"), PathBuf::from("/cache"));
+        docker_run_command.volume(PathBuf::from("foo"), PathBuf::from("/bar"));
 
         let command: Command = docker_run_command.clone().into();
         assert_eq!(
@@ -337,6 +362,10 @@ mod tests {
                 "127.0.0.1::12345",
                 "--publish",
                 "127.0.0.1::55555",
+                "--volume",
+                "./test-cache:/cache",
+                "--volume",
+                "foo:/bar",
                 "my-image",
                 "echo",
                 "hello",

libcnb-test/src/test_context.rs

@@ -108,6 +108,10 @@ impl<'a> TestContext<'a> {
             docker_run_command.env(key, value);
         });
 
+        config.volumes.iter().for_each(|(source, destination)| {
+            docker_run_command.volume(source, destination);
+        });
+
         config.exposed_ports.iter().for_each(|port| {
             docker_run_command.expose_port(*port);
         });