def test_set_target_raid_config_stdin(self, stdin_read_mock): node_manager_mock = mock.MagicMock(spec=['set_target_raid_config']) client_mock = mock.MagicMock(spec=['node'], node=node_manager_mock) target_raid_config_string = ( '{"logical_disks": [{"size_gb": 100, "raid_level": "1"}]}') stdin_read_mock.return_value = target_raid_config_string args_mock = mock.MagicMock(node='node', target_raid_config='-') expected_target_raid_config_string = json.loads( target_raid_config_string) n_shell.do_node_set_target_raid_config(client_mock, args_mock) stdin_read_mock.assert_called_once_with('target_raid_config') client_mock.node.set_target_raid_config.assert_called_once_with( 'node', expected_target_raid_config_string)
def test_do_node_set_target_raid_config_string(self): node_manager_mock = mock.MagicMock(spec=['set_target_raid_config']) client_mock = mock.MagicMock(spec=['node'], node=node_manager_mock) target_raid_config_string = ( '{"logical_disks": [{"size_gb": 100, "raid_level": "1"}]}') expected_target_raid_config_string = json.loads( target_raid_config_string) args = mock.MagicMock(node='node', target_raid_config=target_raid_config_string) n_shell.do_node_set_target_raid_config(client_mock, args) node_manager_mock.set_target_raid_config.assert_called_once_with( 'node', expected_target_raid_config_string)
def test_set_target_raid_config_stdin(self, stdin_read_mock): node_manager_mock = mock.MagicMock(spec=['set_target_raid_config']) client_mock = mock.MagicMock(spec=['node'], node=node_manager_mock) target_raid_config_string = ( '{"logical_disks": [{"size_gb": 100, "raid_level": "1"}]}') stdin_read_mock.return_value = target_raid_config_string args_mock = mock.MagicMock(node='node', target_raid_config='-') expected_target_raid_config_string = json.loads( target_raid_config_string) n_shell.do_node_set_target_raid_config(client_mock, args_mock) stdin_read_mock.assert_called_once_with('target_raid_config') client_mock.node.set_target_raid_config.assert_called_once_with( 'node', expected_target_raid_config_string)
def test_do_node_set_target_raid_config_string(self): node_manager_mock = mock.MagicMock(spec=['set_target_raid_config']) client_mock = mock.MagicMock(spec=['node'], node=node_manager_mock) target_raid_config_string = ( '{"logical_disks": [{"size_gb": 100, "raid_level": "1"}]}') expected_target_raid_config_string = json.loads( target_raid_config_string) args = mock.MagicMock(node='node', target_raid_config=target_raid_config_string) n_shell.do_node_set_target_raid_config(client_mock, args) node_manager_mock.set_target_raid_config.assert_called_once_with( 'node', expected_target_raid_config_string)
def test_do_node_set_target_raid_config_file(self): contents = '{"raid": "config"}' with tempfile.NamedTemporaryFile(mode='w') as f: f.write(contents) f.flush() node_manager_mock = mock.MagicMock(spec=['set_target_raid_config']) client_mock = mock.MagicMock(spec=['node'], node=node_manager_mock) args = mock.MagicMock() args.node = 'node_uuid' args.target_raid_config = f.name n_shell.do_node_set_target_raid_config(client_mock, args) node_manager_mock.set_target_raid_config.assert_called_once_with( 'node_uuid', json.loads(contents))
def test_do_node_set_target_raid_config_file(self): contents = '{"raid": "config"}' with tempfile.NamedTemporaryFile(mode='w') as f: f.write(contents) f.flush() node_manager_mock = mock.MagicMock(spec=['set_target_raid_config']) client_mock = mock.MagicMock(spec=['node'], node=node_manager_mock) args = mock.MagicMock() args.node = 'node_uuid' args.target_raid_config = f.name n_shell.do_node_set_target_raid_config(client_mock, args) node_manager_mock.set_target_raid_config.assert_called_once_with( 'node_uuid', json.loads(contents))