mirror of
https://github.com/NanmiCoder/MediaCrawler.git
synced 2026-06-09 11:27:26 +08:00
fix: Update tests to handle openpyxl color format and ContextVar
- Fix header color assertion to check only RGB values (not alpha channel) - Remove ContextVar mock as it cannot be patched in Python 3.11+ - All 17 tests now passing successfully
This commit is contained in:
@@ -143,7 +143,8 @@ class TestExcelStoreBase:
|
|||||||
# Check header formatting
|
# Check header formatting
|
||||||
header_cell = excel_store.contents_sheet.cell(row=1, column=1)
|
header_cell = excel_store.contents_sheet.cell(row=1, column=1)
|
||||||
assert header_cell.font.bold is True
|
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):
|
def test_empty_sheets_removed(self, excel_store):
|
||||||
"""Test that empty sheets are removed on flush"""
|
"""Test that empty sheets are removed on flush"""
|
||||||
|
|||||||
@@ -51,9 +51,9 @@ class TestXhsStoreFactory:
|
|||||||
assert isinstance(store, XhsMongoStoreImplement)
|
assert isinstance(store, XhsMongoStoreImplement)
|
||||||
|
|
||||||
@patch('config.SAVE_DATA_OPTION', 'excel')
|
@patch('config.SAVE_DATA_OPTION', 'excel')
|
||||||
@patch('var.crawler_type_var.get', return_value='search')
|
def test_create_excel_store(self):
|
||||||
def test_create_excel_store(self, mock_crawler_type):
|
|
||||||
"""Test creating Excel store"""
|
"""Test creating Excel store"""
|
||||||
|
# ContextVar cannot be mocked, so we test with actual value
|
||||||
store = XhsStoreFactory.create_store()
|
store = XhsStoreFactory.create_store()
|
||||||
assert isinstance(store, XhsExcelStoreImplement)
|
assert isinstance(store, XhsExcelStoreImplement)
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user