From 8163ecefb13aae225d8f1d2cd321a9781144272a Mon Sep 17 00:00:00 2001
From: Jorik Cronenberg <jcronenberg@suse.de>
Date: Mon, 29 Jan 2024 15:34:36 +0100
Subject: [PATCH] Add wireless migration tests

---
 rust/migrate-wicked/src/wireless.rs | 103 ++++++++++++++++++++++++++++
 1 file changed, 103 insertions(+)

diff --git a/rust/migrate-wicked/src/wireless.rs b/rust/migrate-wicked/src/wireless.rs
index 6315c54e00..9f13ee9671 100644
--- a/rust/migrate-wicked/src/wireless.rs
+++ b/rust/migrate-wicked/src/wireless.rs
@@ -171,3 +171,106 @@ impl TryFrom<&Network> for model::ConnectionConfig {
         Ok(model::ConnectionConfig::Wireless(config))
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use crate::interface::*;
+    use crate::MIGRATION_SETTINGS;
+
+    #[allow(dead_code)]
+    fn setup_default_migration_settings() {
+        let _ = MIGRATION_SETTINGS.set(crate::MigrationSettings {
+            continue_migration: false,
+            dry_run: false,
+            activate_connections: true,
+        });
+    }
+
+    #[test]
+    fn test_wireless_bands() {
+        let mut wireless_interface = Interface {
+            wireless: Some(Wireless {
+                networks: Some(vec![Network {
+                    channel: Some(0),
+                    essid: "testssid".to_string(),
+                    scan_ssid: false,
+                    mode: WickedWirelessMode::AP,
+                    wpa_psk: None,
+                    key_management: vec!["wpa-psk".to_string()],
+                    access_point: None,
+                    wep: None,
+                }]),
+                ap_scan: 0,
+            }),
+            ..Default::default()
+        };
+        let ifc = wireless_interface.to_connection();
+        assert!(ifc.is_ok());
+        let ifc = &ifc.unwrap().connections[0];
+        let model::ConnectionConfig::Wireless(wireless) = &ifc.config else {
+            panic!()
+        };
+        assert_eq!(wireless.band, Some("bg".try_into().unwrap()));
+
+        wireless_interface
+            .wireless
+            .as_mut()
+            .unwrap()
+            .networks
+            .as_mut()
+            .unwrap()[0]
+            .channel = Some(32);
+        let ifc = wireless_interface.to_connection();
+        assert!(ifc.is_ok());
+        let ifc = &ifc.unwrap().connections[0];
+        let model::ConnectionConfig::Wireless(wireless) = &ifc.config else {
+            panic!()
+        };
+        assert_eq!(wireless.band, Some("a".try_into().unwrap()));
+    }
+
+    #[test]
+    fn test_wireless_migration() {
+        let wireless_interface = Interface {
+            wireless: Some(Wireless {
+                networks: Some(vec![Network {
+                    essid: "testssid".to_string(),
+                    scan_ssid: true,
+                    mode: WickedWirelessMode::Infrastructure,
+                    wpa_psk: Some(WpaPsk {
+                        passphrase: "testpassword".to_string(),
+                    }),
+                    key_management: vec!["wpa-psk".to_string()],
+                    channel: Some(14),
+                    access_point: Some("12:34:56:78:9A:BC".to_string()),
+                    wep: Some(Wep {
+                        auth_algo: "open".to_string(),
+                        default_key: 1,
+                        key: vec!["01020304ff".to_string(), "s:hello".to_string()],
+                    }),
+                }]),
+                ap_scan: 0,
+            }),
+            ..Default::default()
+        };
+        let ifc = wireless_interface.to_connection();
+        assert!(ifc.is_ok());
+        let ifc = &ifc.unwrap().connections[0];
+        let model::ConnectionConfig::Wireless(wireless) = &ifc.config else {
+            panic!()
+        };
+        assert_eq!(wireless.ssid, SSID("testssid".as_bytes().to_vec()));
+        assert_eq!(wireless.mode, model::WirelessMode::Infra);
+        assert_eq!(wireless.password, Some("testpassword".to_string()));
+        assert_eq!(wireless.security, model::SecurityProtocol::WPA2);
+        assert_eq!(wireless.bssid, Some(MacAddr6::from_str("12:34:56:78:9A:BC").unwrap()));
+        assert_eq!(wireless.wep_security, Some(WEPSecurity {
+            auth_alg: WEPAuthAlg::Open,
+            wep_key_type: WEPKeyType::Key,
+            keys: vec!["01020304ff".to_string(), "hello".to_string()],
+            wep_key_index: 1,
+        }));
+        assert_eq!(wireless.band, Some("bg".try_into().unwrap()));
+    }
+}