def get_agent_api(self, **kwargs):
if self.agent_api:
return self.agent_api
sg_agent = kwargs.get("sg_agent")
iptables_manager = self._get_iptables_manager(sg_agent)
self.agent_api = agent_extension_api.LinuxbridgeAgentExtensionAPI(
iptables_manager)
return self.agent_api
def setUp(self):
super(TestLinuxbridgeAgentExtensionAPI, self).setUp()
self.iptables_manager = mock.Mock()
self.extension_api = ext_api.LinuxbridgeAgentExtensionAPI(
self.iptables_manager)
