Files
Netbox-Utilities/netbox_utilities/tests/test_svg_export.py
T
MrBlakeandClaude Sonnet 5 bfc4382403 feat: honor images-only/labels-only rack view mode in offline SVG export
The rack elevation page's "Bilder und Beschriftungen" dropdown only
toggles CSS visibility client-side after the browser already has the
SVG, so a freshly server-rendered export had no way to know which mode
was selected and always exported everything.

apply_rack_view_mode() replicates that filtering server-side by
removing the .label text elements (images-only) or the .device-image /
.device-image-label elements (labels-only) from the parsed SVG.
RackElevationOfflineSVGView now accepts a ?view= query param, and the
rack detail page panel syncs its download links to the page's own
"select.rack-view" dropdown so the exported file matches what's
currently shown.

Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com>
2026-09-17 15:30:38 +02:00

123 lines
4.2 KiB
Python

from django.test import SimpleTestCase
from netbox_utilities.svg_export import apply_rack_view_mode, strip_svg_hyperlinks
class StripSvgHyperlinksTest(SimpleTestCase):
def test_unwraps_anchor_elements_keeping_children(self):
svg = (
'<svg xmlns="http://www.w3.org/2000/svg">'
'<a xmlns:xlink="http://www.w3.org/1999/xlink" xlink:href="/dcim/devices/1/">'
'<rect width="10" height="10"/>'
"</a>"
"</svg>"
)
result = strip_svg_hyperlinks(svg)
self.assertNotIn("<a ", result)
self.assertNotIn("href", result)
self.assertIn("<rect", result)
def test_removes_href_on_non_image_non_anchor_elements(self):
svg = (
'<svg xmlns="http://www.w3.org/2000/svg" '
'xmlns:xlink="http://www.w3.org/1999/xlink">'
'<use xlink:href="#some-symbol" href="#some-symbol"/>'
"</svg>"
)
result = strip_svg_hyperlinks(svg)
self.assertNotIn("href", result)
self.assertIn("<use", result)
def test_leaves_image_href_untouched_without_a_loader(self):
svg = (
'<svg xmlns="http://www.w3.org/2000/svg" '
'xmlns:xlink="http://www.w3.org/1999/xlink">'
'<image xlink:href="/media/devicetype-images/example.png"/>'
"</svg>"
)
result = strip_svg_hyperlinks(svg)
self.assertIn("/media/devicetype-images/example.png", result)
def test_replaces_image_href_with_data_uri_from_loader(self):
svg = (
'<svg xmlns="http://www.w3.org/2000/svg" '
'xmlns:xlink="http://www.w3.org/1999/xlink">'
'<image xlink:href="/media/devicetype-images/example.png"/>'
"</svg>"
)
result = strip_svg_hyperlinks(svg, image_loader=lambda href: "data:image/png;base64,AAAA")
self.assertIn("data:image/png;base64,AAAA", result)
self.assertNotIn("/media/devicetype-images/example.png", result)
def test_drops_image_href_when_loader_cannot_resolve_it(self):
svg = (
'<svg xmlns="http://www.w3.org/2000/svg" '
'xmlns:xlink="http://www.w3.org/1999/xlink">'
'<image xlink:href="/media/devicetype-images/missing.png"/>'
"</svg>"
)
result = strip_svg_hyperlinks(svg, image_loader=lambda href: None)
self.assertNotIn("href", result)
self.assertIn("<image", result)
def test_preserves_sibling_order_around_unwrapped_anchor(self):
svg = (
'<svg xmlns="http://www.w3.org/2000/svg">'
'<rect id="before"/>'
'<a href="/dcim/devices/2/"><rect id="linked"/></a>'
'<rect id="after"/>'
"</svg>"
)
result = strip_svg_hyperlinks(svg)
self.assertLess(result.index('id="before"'), result.index('id="linked"'))
self.assertLess(result.index('id="linked"'), result.index('id="after"'))
def test_leaves_svg_without_links_unchanged_in_content(self):
svg = '<svg xmlns="http://www.w3.org/2000/svg"><rect width="5" height="5"/></svg>'
result = strip_svg_hyperlinks(svg)
self.assertIn("<rect", result)
self.assertNotIn("href", result)
class ApplyRackViewModeTest(SimpleTestCase):
SVG = (
'<svg xmlns="http://www.w3.org/2000/svg">'
'<text class="label">USV 8</text>'
'<image class="device-image"/>'
'<text class="device-image-label">USV 8</text>'
"</svg>"
)
def test_images_and_labels_leaves_svg_untouched(self):
self.assertEqual(apply_rack_view_mode(self.SVG, "images-and-labels"), self.SVG)
def test_images_only_removes_label_text(self):
result = apply_rack_view_mode(self.SVG, "images-only")
self.assertNotIn('class="label"', result)
self.assertIn('class="device-image"', result)
self.assertIn('class="device-image-label"', result)
def test_labels_only_removes_device_images(self):
result = apply_rack_view_mode(self.SVG, "labels-only")
self.assertIn('class="label"', result)
self.assertNotIn("device-image", result)
def test_unknown_mode_leaves_svg_untouched(self):
self.assertEqual(apply_rack_view_mode(self.SVG, "bogus"), self.SVG)