Added Testing for SSID Parsing

tests/test_ssid_parsing.py

@@ -0,0 +1,87 @@
+import unittest
+from dataclasses import dataclass
+from typing import Self
+
+
+@dataclass
+class MockWifiNetwork:
+	bssid: str
+	frequency: str
+	signal_level: str
+	flags: str
+	ssid: str
+
+	@classmethod
+	def from_wpa(cls, results: str) -> list[Self]:
+		entries = []
+
+		for line in results.splitlines():
+			line = line.strip()
+			if not line:
+				continue
+
+			if '\t' in line:
+				parts = line.split('\t')
+			else:
+				parts = line.split(maxsplit=4)
+
+			if len(parts) < 5:
+				continue
+
+			wifi = cls(bssid=parts[0], frequency=parts[1], signal_level=parts[2], flags=parts[3], ssid=parts[4])
+			entries.append(wifi)
+
+		return entries
+
+
+class TestWifiNetworkParser(unittest.TestCase):
+	def test_standard_tab_separated_no_spaces(self):
+		"""Test classic tab-separated output with an SSID containing no spaces."""
+		raw_data = '00:11:22:33:44:55\t2412\t-41\t[WPA2-PSK-CCMP][ESS]\tMyNetwork'
+		results = MockWifiNetwork.from_wpa(raw_data)
+
+		self.assertEqual(len(results), 1)
+		self.assertEqual(results[0].ssid, 'MyNetwork')
+		self.assertEqual(results[0].bssid, '00:11:22:33:44:55')
+
+	def test_tab_separated_with_spaces_in_ssid(self):
+		"""Test tab-separated output where the SSID contains spaces (the core bug)."""
+		raw_data = '00:11:22:33:44:55\t2412\t-41\t[WPA2-PSK-CCMP][ESS]\tMy Home Wi-Fi Network'
+		results = MockWifiNetwork.from_wpa(raw_data)
+
+		self.assertEqual(len(results), 1)
+		self.assertEqual(results[0].ssid, 'My Home Wi-Fi Network')
+
+	def test_fallback_space_separated_with_spaces_in_ssid(self):
+		"""Test space-separated fallback behavior when tabs are missing entirely."""
+		raw_data = '00:11:22:33:44:55 2412 -41 [WPA2-PSK-CCMP][ESS] My Home Wi-Fi Network'
+		results = MockWifiNetwork.from_wpa(raw_data)
+
+		self.assertEqual(len(results), 1)
+		self.assertEqual(results[0].ssid, 'My Home Wi-Fi Network')
+
+	def test_malformed_short_line_ignored(self):
+		"""Test that lines with fewer than 5 columns are skipped and do not throw IndexError."""
+		raw_data = '00:11:22:33:44:55\t2412\t-41'
+		results = MockWifiNetwork.from_wpa(raw_data)
+
+		self.assertEqual(len(results), 0)
+
+	def test_extra_columns_handled_gracefully(self):
+		"""Test that output with more than 5 columns (e.g. newer wpa_cli formats) doesn't crash."""
+		raw_data = '00:11:22:33:44:55\t2412\t-41\t[WPA2-PSK-CCMP][ESS]\tMyNetwork\textra_col_1\textra_col_2'
+		results = MockWifiNetwork.from_wpa(raw_data)
+
+		self.assertEqual(len(results), 1)
+		self.assertEqual(results[0].ssid, 'MyNetwork')
+
+	def test_empty_input_and_whitespace_lines(self):
+		"""Test that empty lines, headers, or purely whitespace rows are skipped cleanly."""
+		raw_data = '\n   \n\n00:11:22:33:44:55\t2412\t-41\t[WPA2-PSK-CCMP][ESS]\tMyNetwork\n\n'
+		results = MockWifiNetwork.from_wpa(raw_data)
+
+		self.assertEqual(len(results), 1)
+
+
+if __name__ == '__main__':
+	unittest.main()