Esempio n. 1
0
 def load_router_data(self, yaml_vt_data):
     """Load Router virtual topology data from the yaml file. """
     virtual_topology_data = yaml.load(yaml_vt_data)
     self._router_data = virtual_topology_data['routers'][0].get('router')
     self._router = Router(self._api, self._context, self._router_data)
     self._router._get_tenant_id = MagicMock(return_value='tenant_0')
     self._router.build()
Esempio n. 2
0
 def load_router_data(self, yaml_vt_data):
     """ Load Router virtual topology data from the yaml file. """
     virtual_topology_data = yaml.load(yaml_vt_data)
     self._router_data = virtual_topology_data['routers'][0].get('router')
     self._router = Router(self._api, self._context, self._router_data)
     self._router._get_tenant_id = MagicMock(return_value='tenant_0')
     self._router.build()
 def add_router(self, router):
     router_obj = Router(self._api, self, router)
     router_obj.build()
     self._routers[router['name']] = router_obj
     self._bridge_router[router['name']] = router_obj
 def add_router(self, router):
     router_obj = Router(self._api, self, router)
     router_obj.build()
     self._routers[router['name']] = router_obj
     self._bridge_router[router['name']] = router_obj
Esempio n. 5
0
class RouterTest(unittest.TestCase):

    def setUp(self):
        self._api = MagicMock()
        self._context = MagicMock()
        self._router_data = None
        self._router = None
        self._chain = MagicMock()
        self._chain._mn_resource.get_id.return_value = 'chain_0'
        self._mn_router = MagicMock()
        self._api.add_router.return_value = self._mn_router

    def load_router_data(self, yaml_vt_data):
        """ Load Router virtual topology data from the yaml file. """
        virtual_topology_data = yaml.load(yaml_vt_data)
        self._router_data = virtual_topology_data['routers'][0].get('router')
        self._router = Router(self._api, self._context, self._router_data)
        self._router._get_tenant_id = MagicMock(return_value='tenant_0')
        self._router.build()

    def test_assign_filters_on_build_no_filters(self):
        """ Test if filters are correctly set when no filters are specified. """
        self.load_router_data("""
                routers:
                  - router:
                      name: router-000-001
                """)

        self.assertEqual(3, len(self._mn_router.mock_calls))
        self._mn_router.create.assert_called_with()
        self.assertEqual(None, self._router._inbound_filter)
        self.assertEqual(None, self._router._outbound_filter)

    def test_inbound_filter_resolution(self):
        """ Tests if in-bound filter is correctly looked up."""
        self.load_router_data("""
                routers:
                  - router:
                      name: router-000-001
                      inbound_filter_id:
                        chain_name: in_filter_001
                """)

        self._context.look_up_resource.assert_called_with(self._mn_router,
                'inbound_filter_id', {'chain_name': 'in_filter_001'})

    def test_outbound_filter_resolution(self):
        """ Tests if out-bound filter is correctly looked up."""
        self.load_router_data("""
                routers:
                  - router:
                      name: router-000-001
                      outbound_filter_id:
                        chain_name: out_filter_001
                """)

        self._context.look_up_resource.assert_called_with(self._mn_router,
                'outbound_filter_id', {'chain_name': 'out_filter_001'})

    def test_set_inbound_filter(self):
        """ Tests if setting an in-bound filter to a router dynamically updates
            the topology data for the router resource.
        """
        self.load_router_data("""
            routers:
              - router:
                  name: router-000-001
            """)
        self.assertEqual(None, self._router.get_inbound_filter())

        # Sets a new rule chain. The router resource data needs to be updated.
        self._router.set_inbound_filter(self._chain)
        self.assertEqual(self._chain, self._router.get_inbound_filter())
        self._mn_router.inbound_filter_id.assert_called_with('chain_0')
        self._mn_router.update.assert_called_with()

        # Deletes the rule chain. The router resource data needs to be updated.
        self._router.set_inbound_filter(None)
        self.assertEqual(None, self._router.get_inbound_filter())
        self._mn_router.inbound_filter_id.assert_called_with(None)
        self._mn_router.update.assert_called_with()

    def test_set_outbound_filter(self):
        """ Tests if setting an out-bound filter to a router dynamically updates
            the topology data for the router resource.
        """
        self.load_router_data("""
            routers:
              - router:
                  name: router-000-001
            """)
        self.assertEqual(None, self._router.get_outbound_filter())

        # Sets a new rule chain. The router resource data needs to be updated.
        self._router.set_outbound_filter(self._chain)
        self.assertEqual(self._chain, self._router.get_outbound_filter())
        self._mn_router.outbound_filter_id.assert_called_with('chain_0')
        self._mn_router.update.assert_called_with()

        # Deletes the rule chain. The router resource data needs to be updated.
        self._router.set_outbound_filter(None)
        self.assertEqual(None, self._router.get_outbound_filter())
        self._mn_router.outbound_filter_id.assert_called_with(None)
        self._mn_router.update.assert_called_with()
Esempio n. 6
0
class RouterTest(unittest.TestCase):

    def setUp(self):
        self._api = MagicMock()
        self._context = MagicMock()
        self._router_data = None
        self._router = None
        self._chain = MagicMock()
        self._chain._mn_resource.get_id.return_value = 'chain_0'
        self._mn_router = MagicMock()
        self._api.add_router.return_value = self._mn_router

    def load_router_data(self, yaml_vt_data):
        """Load Router virtual topology data from the yaml file. """
        virtual_topology_data = yaml.load(yaml_vt_data)
        self._router_data = virtual_topology_data['routers'][0].get('router')
        self._router = Router(self._api, self._context, self._router_data)
        self._router._get_tenant_id = MagicMock(return_value='tenant_0')
        self._router.build()

    def test_assign_filters_on_build_no_filters(self):
        """Test if filters are correctly set when no filters are specified. """
        self.load_router_data("""
                routers:
                  - router:
                      name: router-000-001
                """)

        self.assertEqual(3, len(self._mn_router.mock_calls))
        self._mn_router.create.assert_called_with()
        self.assertEqual(None, self._router._inbound_filter)
        self.assertEqual(None, self._router._outbound_filter)

    def test_inbound_filter_resolution(self):
        """Tests if in-bound filter is correctly looked up."""
        self.load_router_data("""
                routers:
                  - router:
                      name: router-000-001
                      inbound_filter_id:
                        chain_name: in_filter_001
                """)

        self._context.look_up_resource.assert_called_with(self._mn_router,
                'inbound_filter_id', {'chain_name': 'in_filter_001'})

    def test_outbound_filter_resolution(self):
        """Tests if out-bound filter is correctly looked up."""
        self.load_router_data("""
                routers:
                  - router:
                      name: router-000-001
                      outbound_filter_id:
                        chain_name: out_filter_001
                """)

        self._context.look_up_resource.assert_called_with(self._mn_router,
                'outbound_filter_id', {'chain_name': 'out_filter_001'})

    def test_set_inbound_filter(self):
        """Tests if setting an in-bound filter to a router dynamically updates
           the topology data for the router resource.
        """
        self.load_router_data("""
            routers:
              - router:
                  name: router-000-001
            """)
        self.assertEqual(None, self._router.get_inbound_filter())

        # Sets a new rule chain. The router resource data needs to be updated.
        self._router.set_inbound_filter(self._chain)
        self.assertEqual(self._chain, self._router.get_inbound_filter())
        self._mn_router.inbound_filter_id.assert_called_with('chain_0')
        self._mn_router.update.assert_called_with()

        # Deletes the rule chain. The router resource data needs to be updated.
        self._router.set_inbound_filter(None)
        self.assertEqual(None, self._router.get_inbound_filter())
        self._mn_router.inbound_filter_id.assert_called_with(None)
        self._mn_router.update.assert_called_with()

    def test_set_outbound_filter(self):
        """Tests if setting an out-bound filter to a router dynamically updates
           the topology data for the router resource.
        """
        self.load_router_data("""
            routers:
              - router:
                  name: router-000-001
            """)
        self.assertEqual(None, self._router.get_outbound_filter())

        # Sets a new rule chain. The router resource data needs to be updated.
        self._router.set_outbound_filter(self._chain)
        self.assertEqual(self._chain, self._router.get_outbound_filter())
        self._mn_router.outbound_filter_id.assert_called_with('chain_0')
        self._mn_router.update.assert_called_with()

        # Deletes the rule chain. The router resource data needs to be updated.
        self._router.set_outbound_filter(None)
        self.assertEqual(None, self._router.get_outbound_filter())
        self._mn_router.outbound_filter_id.assert_called_with(None)
        self._mn_router.update.assert_called_with()