def test_get_ports_strings_from_list(self): self.assertEqual(PortsRangeHelper.get_ports_strings_from_list([]), []) self.assertEqual(PortsRangeHelper.get_ports_strings_from_list([10, 11]), ['10-11']) self.assertEqual(PortsRangeHelper.get_ports_strings_from_list([10, 12, 13, 14]), ['10', '12-14']) self.assertEqual(PortsRangeHelper.get_ports_strings_from_list([10, 12, 13, 14, 20, 21, 22]), ['10', '12-14', '20-22'])
def test_get_ports_strings_from_list(self): self.assertEqual(PortsRangeHelper.get_ports_strings_from_list([]), []) self.assertEqual(PortsRangeHelper.get_ports_strings_from_list([10, 11]), ['10-11']) self.assertEqual(PortsRangeHelper.get_ports_strings_from_list([10, 12, 13, 14]), ['10', '12-14']) self.assertEqual(PortsRangeHelper.get_ports_strings_from_list([10, 12, 13, 14, 20, 21, 22]), ['10', '12-14', '20-22'])
def _build_ports_strings(self, nsg, direction_key, ip_protocol): nsg_ports = PortsRangeHelper.build_ports_dict(nsg, direction_key, ip_protocol) IsAllowed = StringUtils.equal(self.access_action, ALLOW_OPERATION) # Find ports with different access level from NSG and this action diff_ports = sorted([p for p in self.action_ports if nsg_ports.get(p, False) != IsAllowed]) return PortsRangeHelper.get_ports_strings_from_list(diff_ports)