支持通过环境变量配置温控档位

tests/test_fan_controller.py

@@ -0,0 +1,54 @@
+import unittest
+
+from controller.client import FanController
+
+
+class FanControllerConfigTest(unittest.TestCase):
+    def make_controller(self, fan_speed_steps=None):
+        return FanController(
+            host='127.0.0.1',
+            username='root',
+            password='password',
+            fan_speed_steps=fan_speed_steps,
+        )
+
+    def test_default_fan_speed_steps_keep_existing_policy(self):
+        controller = self.make_controller()
+
+        self.assertEqual(controller.get_required_fan_speed(50), 20)
+        self.assertEqual(controller.get_required_fan_speed(55), 25)
+        self.assertEqual(controller.get_required_fan_speed(60), 30)
+        self.assertEqual(controller.get_required_fan_speed(65), 40)
+        self.assertEqual(controller.get_required_fan_speed(66), -1)
+        self.assertEqual(controller.get_required_fan_speed(0), -1)
+
+    def test_custom_fan_speed_steps_are_sorted_by_temperature(self):
+        controller = self.make_controller('70:50,45:15,55:25')
+
+        self.assertEqual(controller.get_required_fan_speed(45), 15)
+        self.assertEqual(controller.get_required_fan_speed(46), 25)
+        self.assertEqual(controller.get_required_fan_speed(55), 25)
+        self.assertEqual(controller.get_required_fan_speed(56), 50)
+        self.assertEqual(controller.get_required_fan_speed(70), 50)
+        self.assertEqual(controller.get_required_fan_speed(71), -1)
+
+    def test_invalid_fan_speed_steps_raise_clear_error(self):
+        invalid_values = [
+            '',
+            '50',
+            'abc:20',
+            '50:abc',
+            '-1:20',
+            '50:9',
+            '50:101',
+            '50:20,50:30',
+        ]
+
+        for invalid_value in invalid_values:
+            with self.subTest(invalid_value=invalid_value):
+                with self.assertRaises(ValueError):
+                    self.make_controller(invalid_value)
+
+
+if __name__ == '__main__':
+    unittest.main()