From 6736cdd431a7eb23f261d9ddc2051395e7b133aa Mon Sep 17 00:00:00 2001
From: John Doty <john@d0ty.me>
Date: Thu, 8 Aug 2024 10:26:36 -0700
Subject: [PATCH] Additional tests for configuration

---
 src/client/config.rs |  5 ++++
 src/client/ui.rs     | 54 ++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 59 insertions(+)

diff --git a/src/client/config.rs b/src/client/config.rs
index c7f0bea..3d57e73 100644
--- a/src/client/config.rs
+++ b/src/client/config.rs
@@ -20,6 +20,11 @@ impl ServerConfig {
         ServerConfig { auto: true, ports: HashMap::new() }
     }
 
+    #[cfg(test)]
+    pub fn insert(&mut self, port: u16, config: PortConfig) {
+        self.ports.insert(port, config);
+    }
+
     pub fn contains_key(&self, port: u16) -> bool {
         self.ports.contains_key(&port)
     }
diff --git a/src/client/ui.rs b/src/client/ui.rs
index 24a4def..50daac3 100644
--- a/src/client/ui.rs
+++ b/src/client/ui.rs
@@ -674,6 +674,8 @@ fn centered_rect(width_chars: u16, height_chars: u16, r: Rect) -> Rect {
 
 #[cfg(test)]
 mod tests {
+    use crate::client::config::PortConfig;
+
     use super::*;
     use assert_matches::assert_matches;
 
@@ -938,4 +940,56 @@ mod tests {
         assert_eq!(centered.width, 10);
         assert_eq!(centered.height, 10);
     }
+
+    #[test]
+    fn port_config_description_respected() {
+        let (sender, receiver) = mpsc::channel(64);
+        let mut config = ServerConfig::default();
+        config.insert(
+            8080,
+            PortConfig {
+                enabled: true,
+                description: Some("override".to_string()),
+            },
+        );
+
+        let mut ui = UI::new(receiver, config);
+
+        // There are ports...
+        ui.handle_internal_event(Some(UIEvent::Ports(vec![PortDesc {
+            port: 8080,
+            desc: "my-service".to_string(),
+        }])));
+
+        let description = ui.ports.get(&8080).unwrap().desc.as_ref().unwrap();
+        assert_eq!(description.desc, "override");
+
+        drop(sender);
+    }
+
+    #[test]
+    fn port_config_enabled_respected() {
+        let (sender, receiver) = mpsc::channel(64);
+        let mut config = ServerConfig::default();
+        config.insert(
+            8080,
+            PortConfig {
+                enabled: false,
+                description: Some("override".to_string()),
+            },
+        );
+
+        let mut ui = UI::new(receiver, config);
+
+        // There are ports...
+        ui.handle_internal_event(Some(UIEvent::Ports(vec![PortDesc {
+            port: 8080,
+            desc: "my-service".to_string(),
+        }])));
+
+        let state = ui.ports.get(&8080).unwrap().state();
+        assert_eq!(state, State::Disabled);
+
+        drop(sender);
+    }
 }