diff --git a/tests/test_excel_store.py b/tests/test_excel_store.py index a32ddff..18b55d4 100644 --- a/tests/test_excel_store.py +++ b/tests/test_excel_store.py @@ -143,7 +143,8 @@ class TestExcelStoreBase: # Check header formatting header_cell = excel_store.contents_sheet.cell(row=1, column=1) assert header_cell.font.bold is True - assert header_cell.fill.start_color.rgb == "FF366092" + # RGB color may have different prefix (00 or FF), check the actual color part + assert header_cell.fill.start_color.rgb[-6:] == "366092" def test_empty_sheets_removed(self, excel_store): """Test that empty sheets are removed on flush""" diff --git a/tests/test_store_factory.py b/tests/test_store_factory.py index 13ac4fc..ada123b 100644 --- a/tests/test_store_factory.py +++ b/tests/test_store_factory.py @@ -51,9 +51,9 @@ class TestXhsStoreFactory: assert isinstance(store, XhsMongoStoreImplement) @patch('config.SAVE_DATA_OPTION', 'excel') - @patch('var.crawler_type_var.get', return_value='search') - def test_create_excel_store(self, mock_crawler_type): + def test_create_excel_store(self): """Test creating Excel store""" + # ContextVar cannot be mocked, so we test with actual value store = XhsStoreFactory.create_store() assert isinstance(store, XhsExcelStoreImplement)