def create_plugin(self):
self.network_config = {
"name": self.network_name,
"type": self.plugin_type,
"ipam": {
"type": self.ipam_type,
}
}
self.env = {
CNI_COMMAND_ENV: self.command,
CNI_CONTAINERID_ENV: self.container_id,
CNI_IFNAME_ENV: self.cni_ifname,
CNI_ARGS_ENV: self.cni_args,
CNI_PATH_ENV: self.cni_path,
CNI_NETNS_ENV: self.cni_netns
}
# Create the CniPlugin to test.
plugin = CniPlugin(self.network_config, self.env)
# Mock out the datastore client.
plugin._client = self.client
plugin.policy_driver._client = self.client
return plugin
def setUp(self):
"""
Per-test setup method.
"""
self.container_id = "ff3afbd1-17ad-499d-b514-72438c009e81"
self.network_config = {
"name": "ut-network",
"type": "calico",
"ipam": {
"type": "calico-ipam",
"subnet": "10.22.0.0/16",
"routes": [{
"dst": "0.0.0.0/0"
}],
"range-start": "",
"range-end": ""
}
}
self.env = {
CNI_CONTAINERID_ENV: self.container_id,
CNI_IFNAME_ENV: "eth0",
CNI_ARGS_ENV: "",
CNI_COMMAND_ENV: CNI_CMD_ADD,
CNI_PATH_ENV: "/usr/bin/rkt/",
CNI_NETNS_ENV: "netns",
}
# Create the CniPlugin to test.
self.plugin = CniPlugin(self.network_config, self.env)
# Mock out policy driver.
self.plugin.policy_driver = MagicMock(spec=DefaultPolicyDriver)
# Mock out the datastore client.
self.m_datastore_client = MagicMock(spec=DatastoreClient)
self.plugin._client = self.m_datastore_client
