From 162bd41ba3de8b24ce8746c8cff37d68445aca88 Mon Sep 17 00:00:00 2001
From: Rutherther <rutherther@proton.me>
Date: Sun, 31 Dec 2023 19:58:09 +0100
Subject: [PATCH] tests(mcu): count correctly, excluding the address

---
 mcu_tests/slave_counter/src/main.rs |  5 +----
 mcu_tests/slave_regs/src/main.rs    | 17 ++++++++---------
 2 files changed, 9 insertions(+), 13 deletions(-)

diff --git a/mcu_tests/slave_counter/src/main.rs b/mcu_tests/slave_counter/src/main.rs
index ed5af7a..c11ac80 100644
--- a/mcu_tests/slave_counter/src/main.rs
+++ b/mcu_tests/slave_counter/src/main.rs
@@ -39,10 +39,7 @@ fn main() -> ! {
     loop {
         let res = i2c.read(ADDRESS, &mut buffer);
         if res.is_ok() {
-            for data in buffer {
-                hprintln!("Got: {data}");
-                dbg!(data);
-            }
+            dbg!(buffer);
         }
         else {
             hprintln!("Got an error when trying to read :(");
diff --git a/mcu_tests/slave_regs/src/main.rs b/mcu_tests/slave_regs/src/main.rs
index 36af609..be84f49 100644
--- a/mcu_tests/slave_regs/src/main.rs
+++ b/mcu_tests/slave_regs/src/main.rs
@@ -34,31 +34,30 @@ fn main() -> ! {
 
     let mut delay = Delay::new(cp.SYST, &clocks);
 
-    let mut buffer: [u8; 5] = [0; 5];
+    let mut buffer: [u8; 20] = [0; 20];
     let mut count: u8 = 0;
     const ADDRESS: u8 = 0b1110101;
     fn write_on(buffer: &mut [u8], address: u8, count: &mut u8) {
-        for data in buffer.iter_mut() {
+        for data in buffer[1..=5].iter_mut() {
             *data = *count;
             *count = (*count).wrapping_add(1);
         }
         buffer[0] = address;
     }
 
-    let mut rx_buffer: [u8; 20] = [0; 20];
     loop {
-        write_on(&mut buffer, 0, &mut count);
-        if i2c.write(ADDRESS, &mut buffer).is_err() {
+        write_on(&mut buffer[0..=5], 0, &mut count);
+        if i2c.write(ADDRESS, &mut buffer[0..=5]).is_err() {
             hprintln!("There was an error during write");
         }
-        write_on(&mut buffer, 10, &mut count);
-        if i2c.write(ADDRESS, &mut buffer).is_err() {
+        write_on(&mut buffer[0..=5], 10, &mut count);
+        if i2c.write(ADDRESS, &mut buffer[0..=5]).is_err() {
             hprintln!("There was an error during write");
         }
 
-        let res = i2c.write_read(ADDRESS, &[0], &mut rx_buffer);
+        let res = i2c.write_read(ADDRESS, &[0], &mut buffer);
         if res.is_ok() {
-            dbg!(rx_buffer);
+            dbg!(buffer);
         }
         else {
             hprintln!("Got an error when trying to read :(");
-- 
2.48.1