Example #1
0
 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)