First unit test

tests/__init__.py

@@ -0,0 +1,46 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+import logging
+import unittest
+import unittest.mock
+
+import hddfancontrol
+
+
+class TestDrive(unittest.TestCase):
+
+  def setUp(self):
+    with unittest.mock.patch("hddfancontrol.stat") as stat_mock, \
+         unittest.mock.patch("hddfancontrol.subprocess") as subprocess_mock:
+      stat_mock.stat.S_IFBLK.return_value = True
+      subprocess_mock.check_call.side_effect = subprocess_mock.CalledProcessError(0, "")
+      self.drive = hddfancontrol.Drive("/dev/sda", "/dummy", None)
+
+  def test_getState(self):
+    with unittest.mock.patch("hddfancontrol.subprocess") as subprocess_mock:
+      subprocess_mock.check_output.return_value = "\n/dev/sda:\n drive state is:  active/idle\n"
+      self.assertEqual(self.drive.getState(), hddfancontrol.Drive.DriveState.ACTIVE_IDLE)
+    with unittest.mock.patch("hddfancontrol.subprocess") as subprocess_mock:
+      subprocess_mock.check_output.return_value = "\n/dev/sda:\n drive state is:  standby\n"
+      self.assertEqual(self.drive.getState(), hddfancontrol.Drive.DriveState.STANDBY)
+    with unittest.mock.patch("hddfancontrol.subprocess") as subprocess_mock:
+      subprocess_mock.check_output.return_value = "\n/dev/sda:\n drive state is:  sleeping\n"
+      self.assertEqual(self.drive.getState(), hddfancontrol.Drive.DriveState.SLEEPING)
+    with unittest.mock.patch("hddfancontrol.subprocess") as subprocess_mock:
+      subprocess_mock.check_output.return_value = "/dev/sda: No such file or directory\n"
+      with self.assertRaises(Exception):
+        self.drive.getState()
+    with unittest.mock.patch("hddfancontrol.subprocess") as subprocess_mock:
+      subprocess_mock.check_output.side_effect = subprocess_mock.CalledProcessError(0, "")
+      with self.assertRaises(Exception):
+        self.drive.getState()
+
+
+
+if __name__ == "__main__":
+  # disable logging
+  logging.basicConfig(level=logging.CRITICAL + 1)
+
+  # run tests
+  unittest.main()