重构

toolbox/tests/test_excel_services.py

@@ -0,0 +1,79 @@
+#  Copyright (c) 2025  Jeffrey Hsu - JITToolBox
+#  #
+#  This program is free software: you can redistribute it and/or modify
+#  it under the terms of the GNU General Public License as published by
+#  the Free Software Foundation, either version 3 of the License, or
+#  (at your option) any later version.
+#  #
+#  This program is distributed in the hope that it will be useful,
+#  but WITHOUT ANY WARRANTY; without even the implied warranty of
+#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#  GNU General Public License for more details.
+#  #
+#  You should have received a copy of the GNU General Public License
+#  along with this program.  If not, see <https://www.gnu.org/licenses/>.
+import shutil
+from pathlib import Path
+
+import pytest
+from openpyxl.workbook import Workbook
+
+from toolbox.services.excel_service import ExcelService, AchievementExcelService
+from toolbox.tests import TEST_FILE_PATH
+
+SAVE_TEMP_FILE = False
+
+
+class TestExcelService:
+    es = ExcelService(TEST_FILE_PATH / 'test_excel_services_01.xlsx')
+    es.open(data_only=True)
+
+    def test_open(self):
+        assert isinstance(self.es._workbook, Workbook)
+        assert self.es._sheet is None
+
+    def test_open_failed(self):
+        with pytest.raises(FileNotFoundError):
+            ExcelService(TEST_FILE_PATH / 'non_existent_file.xlsx').open()
+
+    def test_active_sheet(self):
+        self.es.active_sheet('Sheet1')
+        assert self.es._sheet.title == 'Sheet1'
+        self.es.active_sheet('Sheet2')
+        assert self.es._sheet.title == 'Sheet2'
+
+    def test_cur_active_sheet(self):
+        self.es.active_sheet('Sheet1')
+        assert self.es.cur_active_sheet.title == 'Sheet1'
+        self.es.active_sheet('Sheet2')
+        assert self.es.cur_active_sheet.title == 'Sheet2'
+
+    def test_save_and_close(self):
+        temp_excel_file = TEST_FILE_PATH / 'test_excel_services_01_temp.xlsx'
+        shutil.copy(self.es._file_path, temp_excel_file)
+
+        self.es.active_sheet('Sheet1').cur_active_sheet['A1'] = 'Modified'
+        self.es.save().close()
+
+        es2 = ExcelService(temp_excel_file).open().active_sheet('Sheet1')
+        assert es2.cur_active_sheet['A1'].value == 'Modified'
+        es2.close()
+
+        if not SAVE_TEMP_FILE:
+            Path(TEST_FILE_PATH / 'test_excel_services_01_temp.xlsx').unlink()
+
+
+class TestAchievementExcelService:
+    aes = AchievementExcelService(TEST_FILE_PATH / 'test_achievement_excel_service_01.xlsm')
+    aes.load_config(TEST_FILE_PATH / 'test_achievement.default.excel_01.json')
+
+    def test_read_class_info(self):
+        cis = self.aes.read_class_info()
+        assert len(cis) == 2
+        assert cis[0].full_name == '22工程管理（1）（2）'
+        assert cis[1].full_name == '22工程管理（1）（2）'
+
+        assert cis[0].class_name == '22工程管理（1）'
+        assert cis[0].class_number == 34
+        assert cis[1].class_name == '22工程管理（2）'
+        assert cis[1].class_number == 36