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))
