def test_execute_script_remote(self, remote_execute_mock): node_name = 'node5' info = {} obj = node.NodeContext() self.addCleanup(obj._delete_context) obj._execute_script(node_name, info) remote_execute_mock.assert_called_once()
def test_node_info(self): node_name_args = 'node5' obj = node.NodeContext() self.addCleanup(obj._delete_context) obj.nodes = [{'name': node_name_args, 'check': node_name_args}] node_info = obj._get_node_info(node_name_args) self.assertEqual(node_info.get('check'), node_name_args)
def test_execute_script_local(self, local_execute_mock): node_name = 'local' info = {} obj = node.NodeContext() self.addCleanup(obj._delete_context) obj._execute_script(node_name, info) local_execute_mock.assert_called_once()
def test_undeploy_anisble(self, dispatch_ansible_mock): obj = node.NodeContext() obj.env = { 'type': 'ansible' } obj.undeploy() dispatch_ansible_mock.assert_called_once()
def test_undeploy(self, dispatch_script_mock): obj = node.NodeContext() obj.env = { 'type': 'script' } obj.undeploy() dispatch_script_mock.assert_called_once()
def test_deploy_anisible(self, dispatch_ansible_mock): obj = node.NodeContext() self.addCleanup(obj._delete_context) obj.env = { 'type': 'ansible' } obj.deploy() dispatch_ansible_mock.assert_called_once()
def test_get_script(self): script_args = 'hello.bash' info_args = {'script': script_args} obj = node.NodeContext() self.addCleanup(obj._delete_context) script, options = obj._get_script(info_args) self.assertEqual(script_args, script) self.assertEqual('', options)
def test_deploy(self, dispatch_script_mock): obj = node.NodeContext() self.addCleanup(obj._delete_context) obj.env = { 'type': 'script' } obj.deploy() dispatch_script_mock.assert_called_once()
def setUp(self): self.test_context = node.NodeContext() self.addCleanup(self._remove_contexts) self.os_path_join = os.path.join self.attrs = { 'name': 'foo', 'task_id': '1234567890', 'file': self._get_file_abspath(self.NODES_SAMPLE) }
def test_get_client(self, wait_mock): node_name_args = 'node5' obj = node.NodeContext() obj.nodes = [{ 'name': node_name_args, 'user': '******', 'ip': '10.10.10.10', 'pwd': 'ubuntu', }] obj._get_client(node_name_args) self.assertTrue(wait_mock.called)
def test_get_client(self, wait_mock): node_name_args = 'node5' obj = node.NodeContext() self.addCleanup(obj._delete_context) obj.nodes = [{ 'name': node_name_args, 'user': '******', 'ip': '10.10.10.10', 'pwd': 'ubuntu', }] obj._get_client(node_name_args) wait_mock.assert_called_once()
def test_execute_remote_script(self, execute_mock, put_file_mock): obj = node.NodeContext() obj.env = {'prefix': 'yardstick.benchmark.scenarios.compute'} node_name_args = 'node5' obj.nodes = [{ 'name': node_name_args, 'user': '******', 'ip': '10.10.10.10', 'pwd': 'ubuntu', }] info = {'script': 'computecapacity.bash'} execute_mock.return_value = (0, '', '') obj._execute_remote_script('node5', info) self.assertTrue(put_file_mock.called) self.assertTrue(execute_mock.called)
def test__add_resources_to_template_raw(self): test_context = node.NodeContext() self.addCleanup(test_context._delete_context) test_context._name = 'foo' test_context.template_file = '/tmp/some-heat-file' test_context.heat_parameters = {'image': 'cirros'} test_context.key_filename = "/tmp/1234" test_context.keypair_name = "foo-key" test_context.secgroup_name = "foo-secgroup" test_context.key_uuid = "2f2e4997-0a8e-4eb7-9fa4-f3f8fbbc393b" test_context.tmpfile = tempfile.NamedTemporaryFile(delete=True, mode='w+t') test_context.tmpfile.write("heat_template_version: 2015-04-30") test_context.tmpfile.flush() test_context.tmpfile.seek(0) heat_template = heat.HeatTemplate('template name') heat_template.resources = {} heat_template.add_network("network1") heat_template.add_network("network2") heat_template.add_security_group("sec_group1") heat_template.add_security_group("sec_group2") heat_template.add_subnet("subnet1", "network1", "cidr1") heat_template.add_subnet("subnet2", "network2", "cidr2") heat_template.add_router("router1", "gw1", "subnet1") heat_template.add_router_interface("router_if1", "router1", "subnet1") network1 = mock.MagicMock() network1.stack_name = "network1" network1.subnet_stack_name = "subnet1" network1.vnic_type = "normal" network2 = mock.MagicMock() network2.stack_name = "network2" network2.subnet_stack_name = "subnet2" network2.vnic_type = "normal" heat_template.add_port("port1", network1) heat_template.add_port("port2", network2, sec_group_id="sec_group1", provider="not-sriov") heat_template.add_port("port3", network2, sec_group_id="sec_group1", provider="sriov") heat_template.add_floating_ip("floating_ip1", "network1", "port1", "router_if1") heat_template.add_floating_ip("floating_ip2", "network2", "port2", "router_if2", "foo-secgroup") heat_template.add_floating_ip_association("floating_ip1_association", "floating_ip1", "port1") heat_template.add_servergroup("server_grp2", "affinity") heat_template.add_servergroup("server_grp3", "anti-affinity") heat_template.add_security_group("security_group") heat_template.add_server(name="server1", image="image1", flavor="flavor1", flavors=[]) heat_template.add_server_group(name="servergroup", policies=["policy1", "policy2"]) heat_template.add_server_group(name="servergroup", policies="policy1") heat_template.add_server(name="server2", image="image1", flavor="flavor1", flavors=[], ports=["port1", "port2"], networks=["network1", "network2"], scheduler_hints="hints1", user="******", key_name="foo-key", user_data="user", metadata={ "cat": 1, "doc": 2 }, additional_properties={ "prop1": 1, "prop2": 2 }) heat_template.add_server(name="server2", image="image1", flavor="flavor1", flavors=["flavor1", "flavor2"], ports=["port1", "port2"], networks=["network1", "network2"], scheduler_hints="hints1", user="******", key_name="foo-key", user_data="user", metadata={ "cat": 1, "doc": 2 }, additional_properties={ "prop1": 1, "prop2": 2 }) heat_template.add_server(name="server2", image="image1", flavor="flavor1", flavors=["flavor3", "flavor4"], ports=["port1", "port2"], networks=["network1", "network2"], scheduler_hints="hints1", user="******", key_name="foo-key", user_data="user", metadata={ "cat": 1, "doc": 2 }, additional_properties={ "prop1": 1, "prop2": 2 }) heat_template.add_flavor(name="flavor1", vcpus=1, ram=2048, disk=1, extra_specs={ "cat": 1, "dog": 2 }) heat_template.add_flavor(name=None, vcpus=1, ram=2048) heat_template.add_server(name="server1", image="image1", flavor="flavor1", flavors=[], ports=["port1", "port2"], networks=["network1", "network2"], scheduler_hints="hints1", user="******", key_name="foo-key", user_data="user", metadata={ "cat": 1, "doc": 2 }, additional_properties={ "prop1": 1, "prop2": 2 }) heat_template.add_network("network1") heat_template.add_flavor("test") self.assertEqual(heat_template.resources['test']['type'], 'OS::Nova::Flavor')
def setUp(self): self.test_context = node.NodeContext()
def test_get_script(self): script_args = 'hello.bash' info_args = {'script': script_args} script, options = node.NodeContext()._get_script(info_args) self.assertEqual(script_args, script) self.assertEqual('', options)
def test_execute_script_remote(self, remote_execute_mock): node_name = 'node5' info = {} node.NodeContext()._execute_script(node_name, info) self.assertTrue(remote_execute_mock.called)
def test_execute_script_local(self, local_execute_mock): node_name = 'local' info = {} node.NodeContext()._execute_script(node_name, info) self.assertTrue(local_execute_mock.called)
def test_undeploy(self, dispatch_script_mock): obj = node.NodeContext() obj.env = {'type': 'script'} obj.undeploy() self.assertTrue(dispatch_script_mock.called)
def test_undeploy_anisble(self, dispatch_ansible_mock): obj = node.NodeContext() obj.env = {'type': 'ansible'} obj.undeploy() self.assertTrue(dispatch_ansible_mock.called)
def setUp(self): self.test_context = node.NodeContext() self.os_path_join = os.path.join