From 7399132d5d7535059f1a0b58a9daeebe950e5bda Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jo=C3=ABlle=20van=20Essen?=
 <39169351+JoelleJS@users.noreply.github.com>
Date: Wed, 1 Jan 2025 23:02:59 +0100
Subject: [PATCH] Fix test in zluda_dump (#316)

---
 zluda_dump/src/log.rs | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/zluda_dump/src/log.rs b/zluda_dump/src/log.rs
index 484364c..311e646 100644
--- a/zluda_dump/src/log.rs
+++ b/zluda_dump/src/log.rs
@@ -515,8 +515,10 @@ mod os {
 mod tests {
     use std::{borrow::Cow, cell::RefCell, io, rc::Rc, str};
 
+    use cuda_types::cuda::CUresultConsts;
+
     use super::{FunctionLogger, LogEntry, WriteTrailingZeroAware};
-    use crate::{log::WriteBuffer, CUresult};
+    use crate::{log::{CudaFunctionName, WriteBuffer}, CUresult};
 
     struct FailOnNthWrite {
         fail_on: usize,
@@ -574,12 +576,13 @@ mod tests {
         write_buffer.unprefixed_buffer = Some(Vec::new());
         let mut log_queue = Vec::new();
         let mut func_logger = FunctionLogger {
-            result: Some(CUresult::CUDA_SUCCESS),
-            name: Cow::Borrowed("cuInit"),
+            result: Some(CUresult::SUCCESS),
+            name: CudaFunctionName::Normal("cuInit"),
             infallible_emitter: &mut infallible_emitter,
             fallible_emitter: &mut fallible_emitter,
             write_buffer: &mut write_buffer,
             log_queue: &mut log_queue,
+            arguments_writer: None,
         };
 
         func_logger.log(LogEntry::IoError(io::Error::from_raw_os_error(1)));
@@ -592,7 +595,7 @@ mod tests {
         let result_str = str::from_utf8(&*result).unwrap();
         let result_lines = result_str.lines().collect::<Vec<_>>();
         assert_eq!(result_lines.len(), 5);
-        assert_eq!(result_lines[0], "cuInit(...) -> 0x0");
+        assert_eq!(result_lines[0], "cuInit(...) -> CUDA_SUCCESS");
         assert!(result_lines[1].starts_with("    "));
         assert!(result_lines[2].starts_with("    "));
         assert!(result_lines[3].starts_with("    "));