From bc970a960327fb22cc008edd74227bc89b250db1 Mon Sep 17 00:00:00 2001
From: AmauryBrodu <60550980+AmauryBrodu@users.noreply.github.com>
Date: Sat, 18 Mar 2023 12:38:13 +0100
Subject: [PATCH] disk test

---
 src/simulator/disk.rs | 39 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)

diff --git a/src/simulator/disk.rs b/src/simulator/disk.rs
index 17e7d0f..70c5b32 100644
--- a/src/simulator/disk.rs
+++ b/src/simulator/disk.rs
@@ -95,3 +95,42 @@ impl Disk {
         Ok(())
     }
 }
+
+
+#[cfg(test)]
+mod test {
+
+    use crate::Disk;
+
+    #[test]
+    fn test_init_disk() {
+        let disk = Disk::init_disk();
+    }
+
+    #[test]
+    fn test_read_write_disk() {
+        let mut disk = Disk::init_disk();
+
+        let mut data = Vec::new();
+        data.push(0 as u8);
+        data.push(0 as u8);
+        data.push(0 as u8);
+        data.push(0 as u8);
+
+        let mut data1 = Vec::new();
+        data1.push(1 as u8);
+        data1.push(1 as u8);
+        data1.push(1 as u8);
+        data1.push(1 as u8);
+
+        let mut data2: Vec<u8> = Vec::new();
+        
+        Disk::write_request(&mut disk, 0, &mut data);
+        Disk::write_request(&mut disk, 1, &mut data1);
+        Disk::read_request(&mut disk, 1, &mut data2);
+        
+        assert_eq!(data1, data2);
+        assert_ne!(data, data1);
+        assert_ne!(data, data2);
+    }
+}