@@ -725,17 +725,60 @@ def test_invalid_application_security_groups(self):
725
725
namespace = SimpleNamespace (
726
726
** {
727
727
"asg_ids" : "invalid" ,
728
+ "allowed_host_ports" : ["80/tcp" , "443/tcp" , "8080-8090/tcp" , "53/udp" ],
728
729
}
729
730
)
730
731
with self .assertRaises (InvalidArgumentValueError ):
731
732
validators .validate_application_security_groups (
732
733
namespace
733
734
)
734
735
736
+ def test_application_security_groups_without_allowed_host_ports (self ):
737
+ asg_ids = [
738
+ "/subscriptions/00000000-0000-0000-0000-000000000000/resourceGroups/rg1/providers/Microsoft.Network/applicationSecurityGroups/asg1" ,
739
+ ]
740
+ namespace = SimpleNamespace (
741
+ ** {
742
+ "asg_ids" : asg_ids ,
743
+ "allowed_host_ports" : [],
744
+ }
745
+ )
746
+ with self .assertRaises (ArgumentUsageError ):
747
+ validators .validate_application_security_groups (
748
+ namespace
749
+ )
750
+
751
+ def test_nodepool_application_security_groups_without_allowed_host_ports (self ):
752
+ asg_ids = [
753
+ "/subscriptions/00000000-0000-0000-0000-000000000000/resourceGroups/rg1/providers/Microsoft.Network/applicationSecurityGroups/asg1" ,
754
+ ]
755
+ namespace = SimpleNamespace (
756
+ ** {
757
+ "nodepool_asg_ids" : asg_ids ,
758
+ "nodepool_allowed_host_ports" : [],
759
+ }
760
+ )
761
+ with self .assertRaises (ArgumentUsageError ):
762
+ validators .validate_application_security_groups (
763
+ namespace
764
+ )
765
+
735
766
def test_empty_application_security_groups (self ):
736
767
namespace = SimpleNamespace (
737
768
** {
738
769
"asg_ids" : "" ,
770
+ "allowed_host_ports" : [],
771
+ }
772
+ )
773
+ validators .validate_application_security_groups (
774
+ namespace
775
+ )
776
+
777
+ def test_empty_nodepool_application_security_groups (self ):
778
+ namespace = SimpleNamespace (
779
+ ** {
780
+ "nodepool_asg_ids" : "" ,
781
+ "nodepool_allowed_host_ports" : [],
739
782
}
740
783
)
741
784
validators .validate_application_security_groups (
@@ -750,6 +793,22 @@ def test_multiple_application_security_groups(self):
750
793
namespace = SimpleNamespace (
751
794
** {
752
795
"asg_ids" : asg_ids ,
796
+ "allowed_host_ports" : ["80/tcp" , "443/tcp" , "8080-8090/tcp" , "53/udp" ],
797
+ }
798
+ )
799
+ validators .validate_application_security_groups (
800
+ namespace
801
+ )
802
+
803
+ def test_multiple_nodepool_application_security_groups (self ):
804
+ asg_ids = [
805
+ "/subscriptions/00000000-0000-0000-0000-000000000000/resourceGroups/rg1/providers/Microsoft.Network/applicationSecurityGroups/asg1" ,
806
+ "/subscriptions/00000000-0000-0000-0000-000000000000/resourceGroups/rg2/providers/Microsoft.Network/applicationSecurityGroups/asg2" ,
807
+ ]
808
+ namespace = SimpleNamespace (
809
+ ** {
810
+ "nodepool_asg_ids" : asg_ids ,
811
+ "nodepool_allowed_host_ports" : ["80/tcp" , "443/tcp" , "8080-8090/tcp" , "53/udp" ],
753
812
}
754
813
)
755
814
validators .validate_application_security_groups (
0 commit comments