def but_it_replaces_the_xlsx_blob_when_the_part_exists(
self, xlsx_part_prop_, xlsx_part_
):
xlsx_part_prop_.return_value = xlsx_part_
chart_data = ChartWorkbook(None, None)
chart_data.update_from_xlsx_blob(b"xlsx-blob")
assert chart_data.xlsx_part.blob == b"xlsx-blob"
def it_can_change_the_chart_xlsx_part(
self, chart_part_, xlsx_part_, chartSpace_cxml, expected_cxml
):
chart_part_.relate_to.return_value = "rId42"
chart_data = ChartWorkbook(element(chartSpace_cxml), chart_part_)
chart_data.xlsx_part = xlsx_part_
chart_part_.relate_to.assert_called_once_with(xlsx_part_, RT.PACKAGE)
assert chart_data._chartSpace.xml == xml(expected_cxml)
def it_adds_an_xlsx_part_on_update_if_needed(
self, request, chart_part_, package_, xlsx_part_, xlsx_part_prop_
):
EmbeddedXlsxPart_ = class_mock(request, "pptx.parts.chart.EmbeddedXlsxPart")
EmbeddedXlsxPart_.new.return_value = xlsx_part_
chart_part_.package = package_
xlsx_part_prop_.return_value = None
chart_data = ChartWorkbook(element("c:chartSpace"), chart_part_)
chart_data.update_from_xlsx_blob(b"xlsx-blob")
EmbeddedXlsxPart_.new.assert_called_once_with(b"xlsx-blob", package_)
xlsx_part_prop_.assert_called_with(xlsx_part_)
def add_part_fixture(self, request, chart_part_, xlsx_blob_,
EmbeddedXlsxPart_, package_, xlsx_part_,
xlsx_part_prop_):
chartSpace_cxml = 'c:chartSpace'
chart_data = ChartWorkbook(element(chartSpace_cxml), chart_part_)
xlsx_part_prop_.return_value = None
return (chart_data, xlsx_blob_, EmbeddedXlsxPart_, package_,
xlsx_part_prop_, xlsx_part_)
def it_can_get_the_chart_xlsx_part(self, chart_part_, xlsx_part_):
chart_part_.related_part.return_value = xlsx_part_
chart_workbook = ChartWorkbook(
element("c:chartSpace/c:externalData{r:id=rId42}"), chart_part_
)
xlsx_part = chart_workbook.xlsx_part
chart_part_.related_part.assert_called_once_with("rId42")
assert xlsx_part is xlsx_part_
def xlsx_part_set_fixture(self, request, chart_part_, xlsx_part_):
chartSpace_cxml, expected_cxml = request.param
chart_data = ChartWorkbook(element(chartSpace_cxml), chart_part_)
expected_xml = xml(expected_cxml)
return chart_data, xlsx_part_, expected_xml
def xlsx_part_get_fixture(self, request, chart_part_, xlsx_part_):
chartSpace_cxml, xlsx_part_rId = request.param
chart_data = ChartWorkbook(element(chartSpace_cxml), chart_part_)
expected_object = xlsx_part_ if xlsx_part_rId else None
return chart_data, expected_object
def update_blob_fixture(self, request, xlsx_blob_, xlsx_part_prop_):
chart_data = ChartWorkbook(None, None)
return chart_data, xlsx_blob_
def but_it_returns_None_when_the_chart_has_no_xlsx_part(self):
chart_workbook = ChartWorkbook(element("c:chartSpace"), None)
assert chart_workbook.xlsx_part is None