From fda064a6f33cfd313752e81677d81bde47bf3172 Mon Sep 17 00:00:00 2001
From: m1ngsama <contact@m1ng.space>
Date: Thu, 26 Dec 2024 13:00:00 +0800
Subject: [PATCH] Add unit tests for core functionality

---
 test_tracker.py | 45 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 45 insertions(+)
 create mode 100644 test_tracker.py

diff --git a/test_tracker.py b/test_tracker.py
new file mode 100644
index 0000000..33a8c05
--- /dev/null
+++ b/test_tracker.py
@@ -0,0 +1,45 @@
+"""
+Unit tests for tracker functionality
+"""
+
+import unittest
+from unittest.mock import Mock, patch
+from tracker import SystemTracker
+
+
+class TestSystemTracker(unittest.TestCase):
+    def setUp(self):
+        self.tracker = SystemTracker()
+
+    @patch('psutil.cpu_percent')
+    def test_get_cpu_usage(self, mock_cpu):
+        mock_cpu.return_value = 50.0
+        result = self.tracker.get_cpu_usage()
+        self.assertEqual(result, 50.0)
+
+    @patch('psutil.virtual_memory')
+    def test_get_memory_info(self, mock_mem):
+        mock_mem.return_value = Mock(
+            total=8589934592,
+            available=4294967296,
+            percent=50.0,
+            used=4294967296
+        )
+        result = self.tracker.get_memory_info()
+        self.assertEqual(result['percent'], 50.0)
+        self.assertEqual(result['total'], 8589934592)
+
+    @patch('psutil.disk_usage')
+    def test_get_disk_usage(self, mock_disk):
+        mock_disk.return_value = Mock(
+            total=1000000000000,
+            used=500000000000,
+            free=500000000000,
+            percent=50.0
+        )
+        result = self.tracker.get_disk_usage()
+        self.assertEqual(result['percent'], 50.0)
+
+
+if __name__ == '__main__':
+    unittest.main()