def setUp(self):
     super(TestProfileRules, self).setUp()
     self.m_mgr = Mock(spec=RulesManager)
     self.m_ipt_updater = Mock(spec=IptablesUpdater)
     self.m_ips_mgr = Mock(spec=IpsetManager)
     self.rules = ProfileRules("prof1", 4, self.m_ipt_updater,
                               self.m_ips_mgr)
     self.rules._manager = self.m_mgr
     self.rules._id = "prof1"
示例#2
0
    def setUp(self):
        super(TestProfileRules, self).setUp()

        self.config = load_config("felix_default.cfg")

        self.m_mgr = Mock(spec=RulesManager)
        self.m_ipt_updater = Mock(spec=IptablesUpdater)
        self.m_ips_mgr = Mock(spec=IpsetManager)
        self.rules = ProfileRules(self.config.plugins["iptables_generator"],
                                  "prof1", 4, self.m_ipt_updater,
                                  self.m_ips_mgr)
        self.rules._manager = self.m_mgr
        self.rules._id = "prof1"