From 221cbd254af8c79d3c7dda7e0d7d7178677ed084 Mon Sep 17 00:00:00 2001
From: Daniel Sanchez Quiros <sanchez.quiros.daniel@gmail.com>
Date: Thu, 7 Nov 2024 12:03:07 +0700
Subject: [PATCH] Fix tests

---
 .../netrunner/src/streaming/settings_subscriber.rs   | 12 ++++--------
 1 file changed, 4 insertions(+), 8 deletions(-)

diff --git a/simlib/netrunner/src/streaming/settings_subscriber.rs b/simlib/netrunner/src/streaming/settings_subscriber.rs
index 2d3aa07..59105ce 100644
--- a/simlib/netrunner/src/streaming/settings_subscriber.rs
+++ b/simlib/netrunner/src/streaming/settings_subscriber.rs
@@ -101,9 +101,10 @@ where
 
 #[cfg(test)]
 mod tests {
+    use std::collections::HashSet;
     use std::{collections::HashMap, time::Duration};
 
-    use consensus_engine::View;
+    type View = usize;
 
     use crate::{
         network::{
@@ -123,7 +124,7 @@ mod tests {
     use super::*;
     #[derive(Debug, Clone, Serialize)]
     struct SettingsRecord {
-        states: HashMap<NodeId, View>,
+        states: HashSet<NodeId>,
     }
 
     impl<S, T: Serialize> TryFrom<&SimulationState<S, T>> for SettingsRecord {
@@ -131,12 +132,7 @@ mod tests {
 
         fn try_from(value: &SimulationState<S, T>) -> Result<Self, Self::Error> {
             Ok(Self {
-                states: value
-                    .nodes
-                    .read()
-                    .iter()
-                    .map(|node| (node.id(), node.current_view()))
-                    .collect(),
+                states: value.nodes.read().iter().map(|node| node.id()).collect(),
             })
         }
     }