def setUp(self):
        self.vnc_lib = mock.Mock()

        self.cassandra = mock.Mock()
        self.logger = mock.Mock()

        self.svc = mock.Mock()
        self.svc.netns_manager = instance_manager.NetworkNamespaceManager(
            self.vnc_lib, self.cassandra, self.logger, None, None, None)

        self.snat_agent = snat_agent.SNATAgent(self.svc, self.vnc_lib,
                                               self.cassandra, None)
        DBBase.init(self, self.logger, self.cassandra)

        # register the project
        proj_fq_name = ['default-domain', 'demo']
        config_db.ProjectSM.locate(
            ROUTER_1['parent_uuid'],
            {'fq_name': proj_fq_name})

        project = Project(name=proj_fq_name[-1])
        self.vnc_lib.project_read = mock.Mock(
            return_value=project)

        # register the public network
        config_db.VirtualNetworkSM.locate(
            ROUTER_1['virtual_network_refs'][0]['uuid'],
            {'fq_name': ROUTER_1['virtual_network_refs'][0]['to'], 'parent_type': 'project'})

        # register interfaces
        config_db.VirtualMachineInterfaceSM.locate(
            ROUTER_1['virtual_machine_interface_refs'][0]['uuid'],
            {'fq_name': ROUTER_1['virtual_machine_interface_refs'][0]['to'],
             'virtual_network_refs': [{'uuid': 'private1-uuid'}], 'parent_type': 'project'})
Exemple #2
0
    def setUp(self):
        self.vnc_lib = mock.Mock()

        self.cassandra = mock.Mock()
        self.logger = mock.Mock()

        self.svc = mock.Mock()
        self.svc.netns_manager = instance_manager.NetworkNamespaceManager(
            self.vnc_lib, self.cassandra, self.logger, None, None, None)

        # Mock service module logger.
        self.logger = mock.MagicMock()
        self.module_logger = ServiceMonitorModuleLogger(self.logger)

        self.snat_agent = snat_agent.SNATAgent(self.svc,
                                               self.vnc_lib,
                                               self.cassandra,
                                               None,
                                               logger=self.module_logger)
        DBBase.init(self, self.logger, self.cassandra)

        # register the project
        config_db.ProjectSM.locate(PROJECT_UUID, {'fq_name': PROJECT_FQ_NAME})

        project = Project(name=PROJECT_FQ_NAME[-1])
        self.vnc_lib.project_read = mock.Mock(return_value=project)

        for router in ROUTERS:
            # register the public network
            config_db.VirtualNetworkSM.locate(
                router['virtual_network_refs'][0]['uuid'], {
                    'fq_name': router['virtual_network_refs'][0]['to'],
                    'parent_type': 'project'
                })

            for vmi_ref in router['virtual_machine_interface_refs']:
                # register interfaces
                config_db.VirtualMachineInterfaceSM.locate(
                    vmi_ref['uuid'],
                    {
                        'fq_name': vmi_ref['to'],
                        'virtual_network_refs': [{
                            'uuid': 'private1-uuid'
                        }],
                        'parent_type': 'project',
                    },
                )