Skip to content

Commit ef89309

Browse files
Fix failing alias tests (#165)
* Fix alias test imports * Fix XML multiline formating * Fix FirewallAlias counter type in tests * Format fw alias unit tests
1 parent 2705319 commit ef89309

File tree

1 file changed

+22
-31
lines changed

1 file changed

+22
-31
lines changed

tests/unit/plugins/module_utils/test_firewall_alias_utils.py

+22-31
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@
2828
)
2929
from ansible_collections.puzzle.opnsense.plugins.module_utils.xml_utils import (
3030
elements_equal,
31+
etree_to_dict,
3132
)
3233

3334
# Test version map for OPNsense versions and modules
@@ -240,11 +241,9 @@
240241
<interface/>
241242
<counters>0</counters>
242243
<updatefreq/>
243-
<content>
244-
CF
245-
DZ
246-
AG
247-
</content>
244+
<content>CF
245+
DZ
246+
AG</content>
248247
<description>geoip_test</description>
249248
</alias>
250249
<alias uuid="7e776c20-5658-45fb-924d-9fd833eae142">
@@ -447,7 +446,7 @@ def test_firewall_alias_to_etree():
447446
type=FirewallAliasType.HOSTS.value,
448447
proto=None,
449448
interface=None,
450-
counters="0",
449+
counters=False,
451450
updatefreq=None,
452451
content="10.0.0.1",
453452
description="host_test",
@@ -459,8 +458,7 @@ def test_firewall_alias_to_etree():
459458
orig_alias: Element = test_etree_opnsense.find("OPNsense/Firewall/Alias/aliases")[0]
460459

461460
assert elements_equal(test_element, orig_alias), (
462-
f"{xml_utils.etree_to_dict(test_element)}\n"
463-
f"{xml_utils.etree_to_dict(orig_alias)}"
461+
f"{etree_to_dict(test_element)}\n" f"{etree_to_dict(orig_alias)}"
464462
)
465463

466464

@@ -476,7 +474,7 @@ def test_firewall_alias_to_etree_with_content():
476474
type=FirewallAliasType.GEOIP,
477475
proto=IPProtocol.IPv4,
478476
interface=None,
479-
counters="0",
477+
counters=False,
480478
updatefreq=None,
481479
content=["CF", "DZ", "AG"],
482480
description="geoip_test",
@@ -488,8 +486,7 @@ def test_firewall_alias_to_etree_with_content():
488486
orig_alias: Element = test_etree_opnsense.find("OPNsense/Firewall/Alias/aliases")[5]
489487

490488
assert elements_equal(test_element, orig_alias), (
491-
f"{xml_utils.etree_to_dict(test_element)}\n"
492-
f"{xml_utils.etree_to_dict(orig_alias)}"
489+
f"{etree_to_dict(test_element)}\n" f"{etree_to_dict(orig_alias)}"
493490
)
494491

495492

@@ -505,7 +502,7 @@ def test_firewall_alias_to_etree_with_updatefreq():
505502
type=FirewallAliasType.URLTABLES.value,
506503
proto=None,
507504
interface=None,
508-
counters="0",
505+
counters=False,
509506
updatefreq="2",
510507
content="www.puzzle.ch",
511508
description="url_table_test",
@@ -517,8 +514,7 @@ def test_firewall_alias_to_etree_with_updatefreq():
517514
orig_alias: Element = test_etree_opnsense.find("OPNsense/Firewall/Alias/aliases")[4]
518515

519516
assert elements_equal(test_element, orig_alias), (
520-
f"{xml_utils.etree_to_dict(test_element)}\n"
521-
f"{xml_utils.etree_to_dict(orig_alias)}"
517+
f"{etree_to_dict(test_element)}\n" f"{etree_to_dict(orig_alias)}"
522518
)
523519

524520

@@ -534,7 +530,7 @@ def test_firewall_alias_to_etree_with_bgpasn_none():
534530
type=FirewallAliasType.BGPASN.value,
535531
proto=None,
536532
interface=None,
537-
counters="0",
533+
counters=False,
538534
categories=None,
539535
content="123456",
540536
description="bgp_test_none",
@@ -548,8 +544,7 @@ def test_firewall_alias_to_etree_with_bgpasn_none():
548544
]
549545

550546
assert elements_equal(test_element, orig_alias), (
551-
f"{xml_utils.etree_to_dict(test_element)}\n"
552-
f"{xml_utils.etree_to_dict(orig_alias)}"
547+
f"{etree_to_dict(test_element)}\n" f"{etree_to_dict(orig_alias)}"
553548
)
554549

555550

@@ -565,7 +560,7 @@ def test_firewall_alias_to_etree_with_bgpasn_ipv4():
565560
type=FirewallAliasType.BGPASN.value,
566561
proto="IPv4",
567562
interface=None,
568-
counters="0",
563+
counters=False,
569564
categories=None,
570565
content="123456",
571566
description="bgp_test_ipv4",
@@ -579,8 +574,7 @@ def test_firewall_alias_to_etree_with_bgpasn_ipv4():
579574
]
580575

581576
assert elements_equal(test_element, orig_alias), (
582-
f"{xml_utils.etree_to_dict(test_element)}\n"
583-
f"{xml_utils.etree_to_dict(orig_alias)}"
577+
f"{etree_to_dict(test_element)}\n" f"{etree_to_dict(orig_alias)}"
584578
)
585579

586580

@@ -596,7 +590,7 @@ def test_firewall_alias_to_etree_with_bgpasn_ipv6():
596590
type=FirewallAliasType.BGPASN.value,
597591
proto="IPv6",
598592
interface=None,
599-
counters="0",
593+
counters=False,
600594
categories=None,
601595
content="123456",
602596
description="bgp_test_ipv6",
@@ -610,8 +604,7 @@ def test_firewall_alias_to_etree_with_bgpasn_ipv6():
610604
]
611605

612606
assert elements_equal(test_element, orig_alias), (
613-
f"{xml_utils.etree_to_dict(test_element)}\n"
614-
f"{xml_utils.etree_to_dict(orig_alias)}"
607+
f"{etree_to_dict(test_element)}\n" f"{etree_to_dict(orig_alias)}"
615608
)
616609

617610

@@ -627,7 +620,7 @@ def test_firewall_alias_to_etree_with_bgpasn_ipv4_ipv6():
627620
type=FirewallAliasType.BGPASN.value,
628621
proto="IPv4,IPv6",
629622
interface=None,
630-
counters="0",
623+
counters=False,
631624
categories=None,
632625
content="123456",
633626
description="bgp_test_ipv4_ipv6",
@@ -641,8 +634,7 @@ def test_firewall_alias_to_etree_with_bgpasn_ipv4_ipv6():
641634
]
642635

643636
assert elements_equal(test_element, orig_alias), (
644-
f"{xml_utils.etree_to_dict(test_element)}\n"
645-
f"{xml_utils.etree_to_dict(orig_alias)}"
637+
f"{etree_to_dict(test_element)}\n" f"{etree_to_dict(orig_alias)}"
646638
)
647639

648640

@@ -1317,9 +1309,9 @@ def test_firewall_alias_from_ansible_module_params_with_content_type_bgpasn_vali
13171309
12
13181310
]
13191311

1320-
assert elements_equal(test_element, orig_alias), (
1321-
f"{xml_utils.etree_to_dict(test_element)}\n"
1322-
f"{xml_utils.etree_to_dict(orig_alias)}"
1312+
# description has changed
1313+
assert not elements_equal(test_element, orig_alias), (
1314+
f"{etree_to_dict(test_element)}\n" f"{etree_to_dict(orig_alias)}"
13231315
)
13241316

13251317

@@ -1376,8 +1368,7 @@ def test_firewall_alias_from_ansible_module_params_with_content_type_bgpasn_list
13761368
]
13771369

13781370
assert elements_equal(test_element, orig_alias), (
1379-
f"{xml_utils.etree_to_dict(test_element)}\n"
1380-
f"{xml_utils.etree_to_dict(orig_alias)}"
1371+
f"{etree_to_dict(test_element)}\n" f"{etree_to_dict(orig_alias)}"
13811372
)
13821373

13831374

0 commit comments

Comments
 (0)