def test__get_protocol_port_udp_code_type(self): sg_rule = self._prepare_rule(protocol="icmp", port_range_min=1, port_range_max=8) self.assertEqual("icmp (type:1, code:8)", securitygroup._get_protocol_port(sg_rule))
def test__get_protocol_port_udp_port_range(self): sg_rule = self._prepare_rule(protocol="udp", port_range_min=5000, port_range_max=5010) self.assertEqual("5000-5010/udp", securitygroup._get_protocol_port(sg_rule))
def test__get_protocol_port_icmp_all(self): sg_rule = self._prepare_rule(protocol="icmp") self.assertEqual("icmp", securitygroup._get_protocol_port(sg_rule))
def test__get_protocol_port_all_none(self): sg_rule = self._prepare_rule() self.assertIsNone(securitygroup._get_protocol_port(sg_rule))
def test__get_protocol_port_udp_one_port(self): sg_rule = self._prepare_rule(protocol="udp", port_range_min=22, port_range_max=22) self.assertEqual("22/udp", securitygroup._get_protocol_port(sg_rule))
def test__get_protocol_port_tcp_port_range(self): sg_rule = self._prepare_rule(protocol='tcp', port_range_min=5000, port_range_max=5010) self.assertEqual('5000-5010/tcp', securitygroup._get_protocol_port(sg_rule))
def test__get_protocol_port_udp_all_port(self): sg_rule = self._prepare_rule(protocol='udp') self.assertEqual('udp', securitygroup._get_protocol_port(sg_rule))
def test__get_protocol_port_udp_code_type(self): sg_rule = self._prepare_rule(protocol='icmp', port_range_min=1, port_range_max=8) self.assertEqual('icmp (type:1, code:8)', securitygroup._get_protocol_port(sg_rule))
def test__get_protocol_port_tcp_one_port(self): sg_rule = self._prepare_rule(protocol='tcp', port_range_min=22, port_range_max=22) self.assertEqual('22/tcp', securitygroup._get_protocol_port(sg_rule))
def test__get_protocol_port_icmp_all(self): sg_rule = self._prepare_rule(protocol='icmp') self.assertEqual('icmp', securitygroup._get_protocol_port(sg_rule))
def test__get_protocol_port_udp_port_range(self): sg_rule = self._prepare_rule(protocol='udp', port_range_min=5000, port_range_max=5010) self.assertEqual('5000-5010/udp', securitygroup._get_protocol_port(sg_rule))
def test__get_protocol_port_udp_one_port(self): sg_rule = self._prepare_rule(protocol='udp', port_range_min=22, port_range_max=22) self.assertEqual('22/udp', securitygroup._get_protocol_port(sg_rule))
def setup_columns(self, info, parsed_args): for rule in info: rule['port/protocol'] = _get_protocol_port(rule) rule['remote'] = _get_remote(rule) return super(ListIsoflatRule, self).setup_columns(info, parsed_args)