Example #1
0
    def setUp(self):
        super(PortForwardingExtensionBaseTestCase, self).setUp()

        self.fip_pf_ext = pf.PortForwardingAgentExtension()

        self.context = context.get_admin_context()
        self.connection = mock.Mock()
        self.floatingip2 = router.FloatingIP(context=None, id=_uuid(),
                                             floating_ip_address='172.24.6.12',
                                             floating_network_id=_uuid(),
                                             router_id=_uuid(),
                                             status='ACTIVE')
        self.portforwarding1 = pf_obj.PortForwarding(
            context=None, id=_uuid(), floatingip_id=self.floatingip2.id,
            external_port=1111, protocol='tcp', internal_port_id=_uuid(),
            external_port_range='1111:1111',
            internal_port_range='11111:11111',
            internal_ip_address='1.1.1.1', internal_port=11111,
            floating_ip_address=self.floatingip2.floating_ip_address,
            router_id=self.floatingip2.router_id)

        self.agent = l3_agent.L3NATAgent(HOSTNAME, self.conf)
        self.ex_gw_port = {'id': _uuid()}
        self.fip = {'id': _uuid(),
                    'floating_ip_address': TEST_FIP,
                    'fixed_ip_address': '192.168.0.1',
                    'floating_network_id': _uuid(),
                    'port_id': _uuid(),
                    'host': HOSTNAME}
        self.router = {'id': self.floatingip2.router_id,
                       'gw_port': self.ex_gw_port,
                       'ha': False,
                       'distributed': False,
                       lib_const.FLOATINGIP_KEY: [self.fip]}
        self.router_info = l3router.RouterInfo(
            self.agent, self.floatingip2.router_id, self.router,
            **self.ri_kwargs)
        self.centralized_port_forwarding_fip_set = set(
            [str(self.floatingip2.floating_ip_address) + '/32'])
        self.pf_managed_fips = [self.floatingip2.id]
        self.router_info.ex_gw_port = self.ex_gw_port
        self.router_info.fip_managed_by_port_forwardings = self.pf_managed_fips
        self.agent.router_info[self.router['id']] = self.router_info

        self.get_router_info = mock.patch(
            'neutron.agent.l3.l3_agent_extension_api.'
            'L3AgentExtensionAPI.get_router_info').start()
        self.get_router_info.return_value = self.router_info

        self.agent_api = l3_ext_api.L3AgentExtensionAPI(None, None)
        self.fip_pf_ext.consume_api(self.agent_api)

        self.port_forwardings = [self.portforwarding1]
 def setUp(self):
     super(L3AgentFipPortForwardingExtensionTestFramework, self).setUp()
     self.conf.set_override('extensions', ['port_forwarding'], 'agent')
     self.agent = neutron_l3_agent.L3NATAgentWithStateReport('agent1',
                                                             self.conf)
     self.fip_pf_ext = pf.PortForwardingAgentExtension()
     self.fip_id1 = _uuid()
     self.fip_id2 = _uuid()
     self.fip_id3 = _uuid()
     self.portforwarding1 = pf_obj.PortForwarding(
         context=None, id=_uuid(), floatingip_id=self.fip_id1,
         external_port=1111, protocol='tcp', internal_port_id=_uuid(),
         external_port_range='1111:1111',
         internal_port_range='11111:11111',
         internal_ip_address='1.1.1.1', internal_port=11111,
         floating_ip_address='111.111.111.111', router_id=_uuid())
     self.portforwarding2 = pf_obj.PortForwarding(
         context=None, id=_uuid(), floatingip_id=self.fip_id1,
         external_port=1112, protocol='tcp', internal_port_id=_uuid(),
         external_port_range='1112:1112',
         internal_port_range='11112:11112',
         internal_ip_address='1.1.1.2', internal_port=11112,
         floating_ip_address='111.111.111.111', router_id=_uuid())
     self.portforwarding3 = pf_obj.PortForwarding(
         context=None, id=_uuid(), floatingip_id=self.fip_id2,
         external_port=1113, protocol='tcp', internal_port_id=_uuid(),
         internal_ip_address='1.1.1.3', internal_port=11113,
         external_port_range='1113:1113',
         internal_port_range='11113:11113',
         floating_ip_address='111.222.111.222', router_id=_uuid())
     self.portforwarding4 = pf_obj.PortForwarding(
         context=None, id=_uuid(), floatingip_id=self.fip_id3,
         external_port=2222, protocol='tcp', internal_port_id=_uuid(),
         external_port_range='2222:2222',
         internal_port_range='22222:22222',
         internal_ip_address='2.2.2.2', internal_port=22222,
         floating_ip_address='222.222.222.222', router_id=_uuid())
     self.port_forwardings = [self.portforwarding1, self.portforwarding2,
                              self.portforwarding3, self.portforwarding4]
     self._set_bulk_pull_mock()
     self.managed_fips = [self.fip_id1, self.fip_id2, self.fip_id3]
     self.fip_list_for_pf = ['111.111.111.111/32', '111.222.111.222/32',
                             '222.222.222.222/32']