Skip to content

Commit 84a6196

Browse files
committed
Add additional tests
1 parent 831a83f commit 84a6196

1 file changed

Lines changed: 14 additions & 1 deletion

File tree

tests/test_waf_v2_builder.py

Lines changed: 14 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ def test_waf_v2_ip_rule(waf_builder):
3535
waf_builder.add_ip_rule(
3636
name="TestIp",
3737
priority=1,
38-
addresses={"IPV4": ["17.0.0.0.1/24"]},
38+
addresses={"IPV4": ["17.0.0.0.1/24"], "IPV6": ["::"]},
3939
allow=True,
4040
count_only=False,
4141
cloud_watch_metrics_enabled=False,
@@ -46,6 +46,19 @@ def test_waf_v2_ip_rule(waf_builder):
4646
assert waf.rules[0].name == "TestIpRule"
4747

4848

49+
def test_waf_v2_empty_ip_rule(waf_builder):
50+
waf_builder.add_ip_rule(
51+
name="TestIp",
52+
priority=1,
53+
addresses={},
54+
allow=True,
55+
count_only=False,
56+
cloud_watch_metrics_enabled=False,
57+
)
58+
waf = waf_builder.build()
59+
assert len(waf.rules) == 1
60+
61+
4962
def test_waf_v2_managed_rule(waf_builder):
5063
waf_builder.add_managed_rule(
5164
name="TestManaged",

0 commit comments

Comments
 (0)