85 lines
2.7 KiB
Python
85 lines
2.7 KiB
Python
|
import unittest
|
||
|
|
from unittest.mock import patch
|
||
|
|
|
||
|
|
from sn_basis.functions.qgiscore_wrapper import (
|
||
|
|
QgsLayoutItemLegend,
|
||
|
|
QgsLayoutItemMap,
|
||
|
|
QgsPrintLayout,
|
||
|
|
QgsProject,
|
||
|
|
)
|
||
|
|
from sn_basis.modules.print_layout import PrintLayout
|
||
|
|
|
||
|
|
|
||
|
|
class _FakeExtent:
|
||
|
|
def isNull(self) -> bool:
|
||
|
|
return False
|
||
|
|
|
||
|
|
|
||
|
|
class TestPrintLayout(unittest.TestCase):
|
||
|
|
|
||
|
|
def test_create_legend_item_links_map_and_enables_map_filtering(self):
|
||
|
|
project = QgsProject.instance()
|
||
|
|
print_layout = PrintLayout(project=project)
|
||
|
|
layout = QgsPrintLayout(project)
|
||
|
|
linked_map = QgsLayoutItemMap(layout)
|
||
|
|
|
||
|
|
legend = print_layout._create_legend_item(
|
||
|
|
layout,
|
||
|
|
"legende",
|
||
|
|
linked_map,
|
||
|
|
10.0,
|
||
|
|
20.0,
|
||
|
|
width_mm=50.0,
|
||
|
|
height_mm=30.0,
|
||
|
|
)
|
||
|
|
|
||
|
|
self.assertIs(legend.linked_map, linked_map)
|
||
|
|
self.assertTrue(legend.auto_update_model)
|
||
|
|
self.assertTrue(legend.legend_filter_by_map_enabled)
|
||
|
|
|
||
|
|
def test_create_legend_item_does_not_copy_project_layers_into_legend_model(self):
|
||
|
|
project = QgsProject.instance()
|
||
|
|
print_layout = PrintLayout(project=project)
|
||
|
|
layout = QgsPrintLayout(project)
|
||
|
|
linked_map = QgsLayoutItemMap(layout)
|
||
|
|
|
||
|
|
legend = print_layout._create_legend_item(
|
||
|
|
layout,
|
||
|
|
"legende",
|
||
|
|
linked_map,
|
||
|
|
10.0,
|
||
|
|
20.0,
|
||
|
|
)
|
||
|
|
|
||
|
|
root_group = legend.model().rootGroup()
|
||
|
|
self.assertEqual(root_group.layers, [])
|
||
|
|
|
||
|
|
@patch("sn_basis.modules.print_layout.open_layout_designer")
|
||
|
|
def test_create_single_page_layout_keeps_legend_linked_to_hauptkarte_and_theme(self, _mock_open_designer):
|
||
|
|
project = QgsProject.instance()
|
||
|
|
print_layout = PrintLayout(project=project)
|
||
|
|
|
||
|
|
layout = print_layout.create_single_page_layout(
|
||
|
|
name="Testlayout",
|
||
|
|
page_width_mm=420.0,
|
||
|
|
page_height_mm=297.0,
|
||
|
|
map_width_mm=200.0,
|
||
|
|
map_height_mm=150.0,
|
||
|
|
extent=_FakeExtent(),
|
||
|
|
plotmassstab=5000.0,
|
||
|
|
thema="Thema A",
|
||
|
|
)
|
||
|
|
|
||
|
|
hauptkarte = next(item for item in layout.items if isinstance(item, QgsLayoutItemMap))
|
||
|
|
legende = next(item for item in layout.items if isinstance(item, QgsLayoutItemLegend))
|
||
|
|
|
||
|
|
self.assertTrue(hauptkarte.follow_visibility_preset)
|
||
|
|
self.assertEqual(hauptkarte.follow_visibility_preset_name, "Thema A")
|
||
|
|
self.assertTrue(hauptkarte.keep_layer_set)
|
||
|
|
self.assertIs(legende.linked_map, hauptkarte)
|
||
|
|
self.assertTrue(legende.auto_update_model)
|
||
|
|
self.assertTrue(legende.legend_filter_by_map_enabled)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
unittest.main()
|