From 96a9c27fc08dc1b4a1803f846e19d401868de95f Mon Sep 17 00:00:00 2001
From: Jacob Kiers <jacob@jacobkiers.net>
Date: Tue, 2 Aug 2022 23:17:59 +0200
Subject: [PATCH] Use Message struct in write_to_test_path()

Signed-off-by: Jacob Kiers <jacob@jacobkiers.net>
---
 bin/src/main.rs | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/bin/src/main.rs b/bin/src/main.rs
index d934be2..954da71 100644
--- a/bin/src/main.rs
+++ b/bin/src/main.rs
@@ -103,15 +103,18 @@ fn process_html(input: &str) -> Result<String, sanitize_html::errors::SanitizeEr
     //Ok(input.to_owned())
 }
 
-fn write_to_test_path(uid: &str, message: &[u8]) {
-    let test_path: PathBuf = [Path::new("tests/data"), Path::new(&format!("{}.eml", &uid))]
-        .iter()
-        .collect();
+fn write_to_test_path(msg: &Message) {
+    let test_path: PathBuf = [
+        Path::new("tests/data"),
+        Path::new(&format!("{}.eml", &msg.get_uid())),
+    ]
+    .iter()
+    .collect();
 
     let _ = OpenOptions::new()
         .write(true)
         .create(true)
         .open(test_path)
         .expect("Could not open file fir writing")
-        .write_all(&message);
+        .write_all(&msg.data);
 }