# Copyright (c) 2010-2022 openpyxl
# test
import pytest
from openpyxl.tests.helper import compare_xml
# package
from openpyxl import Workbook
from openpyxl.xml.functions import tostring
@pytest.fixture
def Unicode_Workbook():
wb = Workbook()
ws = wb.active
ws.title = u"D\xfcsseldorf Sheet"
return wb
@pytest.fixture
def WorkbookWriter():
from .._writer import WorkbookWriter
return WorkbookWriter
class TestWorkbookWriter:
def test_write_auto_filter(self, datadir, WorkbookWriter):
datadir.chdir()
wb = Workbook()
ws = wb.active
ws['F42'].value = 'hello'
ws.auto_filter.ref = 'A1:F1'
writer = WorkbookWriter(wb)
xml = writer.write()
with open('workbook_auto_filter.xml') as expected:
diff = compare_xml(xml, expected.read())
assert diff is None, diff
def test_write_hidden_worksheet(self, WorkbookWriter):
wb = Workbook()
ws = wb.active
ws.sheet_state = ws.SHEETSTATE_HIDDEN
wb.create_sheet()
writer = WorkbookWriter(wb)
xml = writer.write()
expected = """
"""
diff = compare_xml(xml, expected)
assert diff is None, diff
def test_write_workbook(self, datadir, WorkbookWriter):
datadir.chdir()
wb = Workbook()
writer = WorkbookWriter(wb)
xml = writer.write()
assert len(writer.rels) == 1
with open('workbook.xml') as expected:
diff = compare_xml(xml, expected.read())
assert diff is None, diff
def test_write_workbook_code_name(self, WorkbookWriter):
wb = Workbook()
wb.code_name = u'MyWB'
writer = WorkbookWriter(wb)
xml = writer.write()
expected = """
"""
diff = compare_xml(xml, expected)
assert diff is None, diff
def test_print_area(self, Unicode_Workbook, WorkbookWriter):
wb = Unicode_Workbook
ws = wb.active
ws.print_area = 'A1:D4'
writer = WorkbookWriter(wb)
xml = writer.write()
expected = """
'Düsseldorf Sheet'!$A$1:$D$4
"""
diff = compare_xml(xml, expected)
assert diff is None, diff
def test_print_titles(self, Unicode_Workbook, WorkbookWriter):
wb = Unicode_Workbook
ws = wb.active
ws.print_title_rows = '1:5'
writer = WorkbookWriter(wb)
xml = writer.write()
expected = """
'Düsseldorf Sheet'!1:5
"""
diff = compare_xml(xml, expected)
assert diff is None, diff
def test_print_autofilter(self, Unicode_Workbook, WorkbookWriter):
wb = Unicode_Workbook
ws = wb.active
ws.auto_filter.ref = "A1:A10"
ws.auto_filter.add_filter_column(0, ["Kiwi", "Apple", "Mango"])
writer = WorkbookWriter(wb)
xml = writer.write()
expected = """
'Düsseldorf Sheet'!$A$1:$A$10
"""
diff = compare_xml(xml, expected)
assert diff is None, diff
def test_write_workbook_protection(self, datadir, WorkbookWriter):
from ...workbook.protection import WorkbookProtection
datadir.chdir()
wb = Workbook()
wb.security = WorkbookProtection(lockStructure=True)
wb.security.set_workbook_password('ABCD', already_hashed=True)
writer = WorkbookWriter(wb)
xml = writer.write()
with open('workbook_protection.xml') as expected:
diff = compare_xml(xml, expected.read())
assert diff is None, diff
def test_write_hidden_single_worksheet():
wb = Workbook()
ws = wb.active
ws.sheet_state = "hidden"
from .._writer import get_active_sheet
with pytest.raises(IndexError):
get_active_sheet(wb)
@pytest.mark.parametrize("vba, filename",
[
(None, 'workbook.xml.rels',),
(True, 'workbook_vba.xml.rels'),
]
)
def test_write_workbook_rels(datadir, vba, filename, WorkbookWriter):
datadir.chdir()
wb = Workbook()
wb.vba_archive = vba
writer = WorkbookWriter(wb)
xml = writer.write_rels()
with open(filename) as expected:
diff = compare_xml(xml, expected.read())
assert diff is None, diff
def test_write_root_rels(WorkbookWriter):
wb = Workbook()
writer = WorkbookWriter(wb)
xml = writer.write_root_rels()
expected = """
"""
diff = compare_xml(xml, expected)
assert diff is None, diff