def _assert(self):
self.assertEqual(1, self.mock_com.call_node.call_count)
self.assertEqual(self.node, self.mock_com.call_node.call_args[0][0])
self.assertEqual(
"remote/booth_set_config", self.mock_com.call_node.call_args[0][1]
)
data = url_decode(self.mock_com.call_node.call_args[0][2])
self.assertTrue("data_json" in data)
self.assertEqual(
{
"config": {
"name": "cfg_name.conf",
"data": "cfg"
}
},
json.loads(data["data_json"][0])
)
assert_report_item_list_equal(
self.mock_rep.report_item_list,
[(
Severities.INFO,
report_codes.BOOTH_CONFIGS_SAVED_ON_NODE,
{
"node": self.node.label,
"name": "cfg_name",
"name_list": ["cfg_name"]
}
)]
)
def test_with_authfile(self):
lib._set_config_on_node(self.mock_com,
self.mock_rep,
self.node,
"cfg_name",
"cfg",
authfile="/abs/path/my-key.key",
authfile_data="test key".encode("utf-8"))
self.assertEqual(1, self.mock_com.call_node.call_count)
self.assertEqual(self.node, self.mock_com.call_node.call_args[0][0])
self.assertEqual("remote/booth_set_config",
self.mock_com.call_node.call_args[0][1])
data = url_decode(self.mock_com.call_node.call_args[0][2])
self.assertTrue("data_json" in data)
self.assertEqual(
{
"config": {
"name": "cfg_name.conf",
"data": "cfg"
},
"authfile": {
"name": "my-key.key",
"data": to_b64("test key")
}
}, json.loads(data["data_json"][0]))
assert_report_item_list_equal(
self.mock_rep.report_item_list,
[(Severities.INFO, report_codes.BOOTH_CONFIG_ACCEPTED_BY_NODE, {
"node": self.node.label,
"name_list": ["cfg_name"]
})])
def test_communication_failure(self, mock_read_authfile, mock_read_configs,
mock_parse, mock_authfile):
mock_parse.side_effect = self.mock_parse_fn
mock_authfile.side_effect = self.mock_authfile_fn
mock_read_authfile.side_effect = self.mock_read_authfile_fn
mock_read_configs.return_value = {
"name1.conf": "config1",
"name2.conf": "config2"
}
self.mock_communicator.call_node.side_effect = NodeConnectionException(
self.node.label, "command", "reason")
assert_raise_library_error(
lambda: lib.send_all_config_to_node(self.mock_communicator, self.
mock_reporter, self.node),
(Severities.ERROR,
report_codes.NODE_COMMUNICATION_ERROR_UNABLE_TO_CONNECT, {
"node": self.node.label,
"command": "command",
"reason": "reason"
}))
self.assertEqual(2, mock_parse.call_count)
mock_parse.assert_has_calls(
[mock.call("config1"), mock.call("config2")])
self.assertEqual(2, mock_authfile.call_count)
mock_authfile.assert_has_calls(
[mock.call("config1"), mock.call("config2")])
self.assertEqual(2, mock_read_authfile.call_count)
mock_read_authfile.assert_has_calls([
mock.call(self.mock_reporter, "/path/to/file1.key"),
mock.call(self.mock_reporter, "/path/to/file2.key")
])
mock_read_configs.assert_called_once_with(self.mock_reporter, False)
self.assertEqual(1, self.mock_communicator.call_node.call_count)
self.assertEqual(self.node,
self.mock_communicator.call_node.call_args[0][0])
self.assertEqual("remote/booth_save_files",
self.mock_communicator.call_node.call_args[0][1])
data = url_decode(self.mock_communicator.call_node.call_args[0][2])
self.assertFalse("rewrite_existing" in data)
self.assertTrue("data_json" in data)
self.assertEqual([{
"name": "name1.conf",
"data": "config1",
"is_authfile": False
}, {
"name": "file1.key",
"data": to_b64("some key"),
"is_authfile": True
}, {
"name": "name2.conf",
"data": "config2",
"is_authfile": False
}, {
"name": "file2.key",
"data": to_b64("another key"),
"is_authfile": True
}], json.loads(data["data_json"][0]))
def test_response_not_json(self, mock_read_authfile, mock_read_configs,
mock_parse, mock_authfile):
mock_parse.side_effect = self.mock_parse_fn
mock_authfile.side_effect = self.mock_authfile_fn
mock_read_authfile.side_effect = self.mock_read_authfile_fn
mock_read_configs.return_value = {
"name1.conf": "config1",
"name2.conf": "config2"
}
self.mock_communicator.call_node.return_value = "not json"
assert_raise_library_error(
lambda: lib.send_all_config_to_node(self.mock_communicator, self.
mock_reporter, self.node),
(Severities.ERROR, report_codes.INVALID_RESPONSE_FORMAT, {
"node": self.node.label
}))
self.assertEqual(2, mock_parse.call_count)
mock_parse.assert_has_calls(
[mock.call("config1"), mock.call("config2")])
self.assertEqual(2, mock_authfile.call_count)
mock_authfile.assert_has_calls(
[mock.call("config1"), mock.call("config2")])
self.assertEqual(2, mock_read_authfile.call_count)
mock_read_authfile.assert_has_calls([
mock.call(self.mock_reporter, "/path/to/file1.key"),
mock.call(self.mock_reporter, "/path/to/file2.key")
])
mock_read_configs.assert_called_once_with(self.mock_reporter, False)
self.assertEqual(1, self.mock_communicator.call_node.call_count)
self.assertEqual(self.node,
self.mock_communicator.call_node.call_args[0][0])
self.assertEqual("remote/booth_save_files",
self.mock_communicator.call_node.call_args[0][1])
data = url_decode(self.mock_communicator.call_node.call_args[0][2])
self.assertFalse("rewrite_existing" in data)
self.assertTrue("data_json" in data)
self.assertEqual([{
"name": "name1.conf",
"data": "config1",
"is_authfile": False
}, {
"name": "file1.key",
"data": to_b64("some key"),
"is_authfile": True
}, {
"name": "name2.conf",
"data": "config2",
"is_authfile": False
}, {
"name": "file2.key",
"data": to_b64("another key"),
"is_authfile": True
}], json.loads(data["data_json"][0]))
def _assert(self):
self.assertEqual(1, self.mock_com.call_node.call_count)
self.assertEqual(self.node, self.mock_com.call_node.call_args[0][0])
self.assertEqual("remote/booth_set_config",
self.mock_com.call_node.call_args[0][1])
data = url_decode(self.mock_com.call_node.call_args[0][2])
self.assertTrue("data_json" in data)
self.assertEqual({"config": {
"name": "cfg_name.conf",
"data": "cfg"
}}, json.loads(data["data_json"][0]))
assert_report_item_list_equal(
self.mock_rep.report_item_list,
[(Severities.INFO, report_codes.BOOTH_CONFIG_ACCEPTED_BY_NODE, {
"node": self.node.label,
"name_list": ["cfg_name"]
})])
def test_with_authfile(self):
lib._set_config_on_node(
self.mock_com,
self.mock_rep,
self.node,
"cfg_name",
"cfg",
authfile="/abs/path/my-key.key",
authfile_data="test key".encode("utf-8")
)
self.assertEqual(1, self.mock_com.call_node.call_count)
self.assertEqual(self.node, self.mock_com.call_node.call_args[0][0])
self.assertEqual(
"remote/booth_set_config", self.mock_com.call_node.call_args[0][1]
)
data = url_decode(self.mock_com.call_node.call_args[0][2])
self.assertTrue("data_json" in data)
self.assertEqual(
{
"config": {
"name": "cfg_name.conf",
"data": "cfg"
},
"authfile": {
"name": "my-key.key",
"data": to_b64("test key")
}
},
json.loads(data["data_json"][0])
)
assert_report_item_list_equal(
self.mock_rep.report_item_list,
[(
Severities.INFO,
report_codes.BOOTH_CONFIGS_SAVED_ON_NODE,
{
"node": self.node.label,
"name": "cfg_name",
"name_list": ["cfg_name"]
}
)]
)
def test_configs_without_authfiles(
self, mock_read_authfile, mock_read_configs, mock_parse, mock_authfile
):
def mock_authfile_fn(parsed_config):
if parsed_config == "config1":
return None
elif parsed_config == "config2":
return "/path/to/file2.key"
else:
raise AssertionError(
"unexpected input: {0}".format(parsed_config)
)
mock_parse.side_effect = self.mock_parse_fn
mock_authfile.side_effect = mock_authfile_fn
mock_read_authfile.return_value = "another key".encode("utf-8")
mock_read_configs.return_value = {
"name1.conf": "config1",
"name2.conf": "config2"
}
self.mock_communicator.call_node.return_value = """
{
"existing": [],
"failed": {},
"saved": ["name1.conf", "name2.conf", "file2.key"]
}
"""
lib.send_all_config_to_node(
self.mock_communicator, self.mock_reporter, self.node
)
self.assertEqual(2, mock_parse.call_count)
mock_parse.assert_has_calls([
mock.call("config1"), mock.call("config2")
])
self.assertEqual(2, mock_authfile.call_count)
mock_authfile.assert_has_calls([
mock.call("config1"), mock.call("config2")
])
mock_read_authfile.assert_called_once_with(
self.mock_reporter, "/path/to/file2.key"
)
mock_read_configs.assert_called_once_with(self.mock_reporter, False)
self.assertEqual(1, self.mock_communicator.call_node.call_count)
self.assertEqual(
self.node, self.mock_communicator.call_node.call_args[0][0]
)
self.assertEqual(
"remote/booth_save_files",
self.mock_communicator.call_node.call_args[0][1]
)
data = url_decode(self.mock_communicator.call_node.call_args[0][2])
self.assertFalse("rewrite_existing" in data)
self.assertTrue("data_json" in data)
self.assertEqual(
[
{
"name": "name1.conf",
"data": "config1",
"is_authfile": False
},
{
"name": "name2.conf",
"data": "config2",
"is_authfile": False
},
{
"name": "file2.key",
"data": to_b64("another key"),
"is_authfile": True
}
],
json.loads(data["data_json"][0])
)
assert_report_item_list_equal(
self.mock_reporter.report_item_list,
[
(
Severities.INFO,
report_codes.BOOTH_CONFIGS_SAVING_ON_NODE,
{"node": self.node.label}
),
(
Severities.INFO,
report_codes.BOOTH_CONFIGS_SAVED_ON_NODE,
{
"node": self.node.label,
"name": "name1.conf, name2.conf, file2.key",
"name_list": ["name1.conf", "name2.conf", "file2.key"]
}
)
]
)
def test_response_not_json(
self, mock_read_authfile, mock_read_configs, mock_parse, mock_authfile
):
mock_parse.side_effect = self.mock_parse_fn
mock_authfile.side_effect = self.mock_authfile_fn
mock_read_authfile.side_effect = self.mock_read_authfile_fn
mock_read_configs.return_value = {
"name1.conf": "config1",
"name2.conf": "config2"
}
self.mock_communicator.call_node.return_value = "not json"
assert_raise_library_error(
lambda: lib.send_all_config_to_node(
self.mock_communicator, self.mock_reporter, self.node
),
(
Severities.ERROR,
report_codes.INVALID_RESPONSE_FORMAT,
{"node": self.node.label}
)
)
self.assertEqual(2, mock_parse.call_count)
mock_parse.assert_has_calls([
mock.call("config1"), mock.call("config2")
])
self.assertEqual(2, mock_authfile.call_count)
mock_authfile.assert_has_calls([
mock.call("config1"), mock.call("config2")
])
self.assertEqual(2, mock_read_authfile.call_count)
mock_read_authfile.assert_has_calls([
mock.call(self.mock_reporter, "/path/to/file1.key"),
mock.call(self.mock_reporter, "/path/to/file2.key")
])
mock_read_configs.assert_called_once_with(self.mock_reporter, False)
self.assertEqual(1, self.mock_communicator.call_node.call_count)
self.assertEqual(
self.node, self.mock_communicator.call_node.call_args[0][0]
)
self.assertEqual(
"remote/booth_save_files",
self.mock_communicator.call_node.call_args[0][1]
)
data = url_decode(self.mock_communicator.call_node.call_args[0][2])
self.assertFalse("rewrite_existing" in data)
self.assertTrue("data_json" in data)
self.assertEqual(
[
{
"name": "name1.conf",
"data": "config1",
"is_authfile": False
},
{
"name": "file1.key",
"data": to_b64("some key"),
"is_authfile": True
},
{
"name": "name2.conf",
"data": "config2",
"is_authfile": False
},
{
"name": "file2.key",
"data": to_b64("another key"),
"is_authfile": True
}
],
json.loads(data["data_json"][0])
)
def test_communication_failure(
self, mock_read_authfile, mock_read_configs, mock_parse, mock_authfile
):
mock_parse.side_effect = self.mock_parse_fn
mock_authfile.side_effect = self.mock_authfile_fn
mock_read_authfile.side_effect = self.mock_read_authfile_fn
mock_read_configs.return_value = {
"name1.conf": "config1",
"name2.conf": "config2"
}
self.mock_communicator.call_node.side_effect = NodeConnectionException(
self.node.label, "command", "reason"
)
assert_raise_library_error(
lambda: lib.send_all_config_to_node(
self.mock_communicator, self.mock_reporter, self.node
),
(
Severities.ERROR,
report_codes.NODE_COMMUNICATION_ERROR_UNABLE_TO_CONNECT,
{
"node": self.node.label,
"command": "command",
"reason": "reason"
}
)
)
self.assertEqual(2, mock_parse.call_count)
mock_parse.assert_has_calls([
mock.call("config1"), mock.call("config2")
])
self.assertEqual(2, mock_authfile.call_count)
mock_authfile.assert_has_calls([
mock.call("config1"), mock.call("config2")
])
self.assertEqual(2, mock_read_authfile.call_count)
mock_read_authfile.assert_has_calls([
mock.call(self.mock_reporter, "/path/to/file1.key"),
mock.call(self.mock_reporter, "/path/to/file2.key")
])
mock_read_configs.assert_called_once_with(self.mock_reporter, False)
self.assertEqual(1, self.mock_communicator.call_node.call_count)
self.assertEqual(
self.node, self.mock_communicator.call_node.call_args[0][0]
)
self.assertEqual(
"remote/booth_save_files",
self.mock_communicator.call_node.call_args[0][1]
)
data = url_decode(self.mock_communicator.call_node.call_args[0][2])
self.assertFalse("rewrite_existing" in data)
self.assertTrue("data_json" in data)
self.assertEqual(
[
{
"name": "name1.conf",
"data": "config1",
"is_authfile": False
},
{
"name": "file1.key",
"data": to_b64("some key"),
"is_authfile": True
},
{
"name": "name2.conf",
"data": "config2",
"is_authfile": False
},
{
"name": "file2.key",
"data": to_b64("another key"),
"is_authfile": True
}
],
json.loads(data["data_json"][0])
)
def test_write_failure(
self, mock_read_authfile, mock_read_configs, mock_parse, mock_authfile
):
mock_parse.side_effect = self.mock_parse_fn
mock_authfile.side_effect = self.mock_authfile_fn
mock_read_authfile.side_effect = self.mock_read_authfile_fn
mock_read_configs.return_value = {
"name1.conf": "config1",
"name2.conf": "config2"
}
self.mock_communicator.call_node.return_value = """
{
"existing": [],
"failed": {
"name1.conf": "Error message",
"file1.key": "Another error message"
},
"saved": ["name2.conf", "file2.key"]
}
"""
assert_raise_library_error(
lambda: lib.send_all_config_to_node(
self.mock_communicator, self.mock_reporter, self.node
),
(
Severities.ERROR,
report_codes.BOOTH_CONFIG_WRITE_ERROR,
{
"node": self.node.label,
"name": "name1.conf",
"reason": "Error message"
}
),
(
Severities.ERROR,
report_codes.BOOTH_CONFIG_WRITE_ERROR,
{
"node": self.node.label,
"name": "file1.key",
"reason": "Another error message"
}
)
)
self.assertEqual(2, mock_parse.call_count)
mock_parse.assert_has_calls([
mock.call("config1"), mock.call("config2")
])
self.assertEqual(2, mock_authfile.call_count)
mock_authfile.assert_has_calls([
mock.call("config1"), mock.call("config2")
])
self.assertEqual(2, mock_read_authfile.call_count)
mock_read_authfile.assert_has_calls([
mock.call(self.mock_reporter, "/path/to/file1.key"),
mock.call(self.mock_reporter, "/path/to/file2.key")
])
mock_read_configs.assert_called_once_with(self.mock_reporter, False)
self.assertEqual(1, self.mock_communicator.call_node.call_count)
self.assertEqual(
self.node, self.mock_communicator.call_node.call_args[0][0]
)
self.assertEqual(
"remote/booth_save_files",
self.mock_communicator.call_node.call_args[0][1]
)
data = url_decode(self.mock_communicator.call_node.call_args[0][2])
self.assertFalse("rewrite_existing" in data)
self.assertTrue("data_json" in data)
self.assertEqual(
[
{
"name": "name1.conf",
"data": "config1",
"is_authfile": False
},
{
"name": "file1.key",
"data": to_b64("some key"),
"is_authfile": True
},
{
"name": "name2.conf",
"data": "config2",
"is_authfile": False
},
{
"name": "file2.key",
"data": to_b64("another key"),
"is_authfile": True
}
],
json.loads(data["data_json"][0])
)
assert_report_item_list_equal(
self.mock_reporter.report_item_list,
[
(
Severities.INFO,
report_codes.BOOTH_CONFIGS_SAVING_ON_NODE,
{"node": self.node.label}
),
(
Severities.ERROR,
report_codes.BOOTH_CONFIG_WRITE_ERROR,
{
"node": self.node.label,
"name": "name1.conf",
"reason": "Error message"
}
),
(
Severities.ERROR,
report_codes.BOOTH_CONFIG_WRITE_ERROR,
{
"node": self.node.label,
"name": "file1.key",
"reason": "Another error message"
}
)
]
)
def test_rewrite_existing(
self, mock_read_authfile, mock_read_configs, mock_parse, mock_authfile
):
mock_parse.side_effect = self.mock_parse_fn
mock_authfile.side_effect = self.mock_authfile_fn
mock_read_authfile.side_effect = self.mock_read_authfile_fn
mock_read_configs.return_value = {
"name1.conf": "config1",
"name2.conf": "config2"
}
self.mock_communicator.call_node.return_value = """
{
"existing": ["name1.conf", "file1.key"],
"failed": {},
"saved": ["name2.conf", "file2.key"]
}
"""
lib.send_all_config_to_node(
self.mock_communicator,
self.mock_reporter,
self.node,
rewrite_existing=True
)
mock_read_configs.assert_called_once_with(self.mock_reporter, False)
self.assertEqual(2, mock_parse.call_count)
mock_parse.assert_has_calls([
mock.call("config1"), mock.call("config2")
])
self.assertEqual(2, mock_authfile.call_count)
mock_authfile.assert_has_calls([
mock.call("config1"), mock.call("config2")
])
self.assertEqual(2, mock_read_authfile.call_count)
mock_read_authfile.assert_has_calls([
mock.call(self.mock_reporter, "/path/to/file1.key"),
mock.call(self.mock_reporter, "/path/to/file2.key")
])
self.assertEqual(1, self.mock_communicator.call_node.call_count)
self.assertEqual(
self.node, self.mock_communicator.call_node.call_args[0][0]
)
self.assertEqual(
"remote/booth_save_files",
self.mock_communicator.call_node.call_args[0][1]
)
data = url_decode(self.mock_communicator.call_node.call_args[0][2])
self.assertTrue("rewrite_existing" in data)
self.assertTrue("data_json" in data)
self.assertEqual(
[
{
"name": "name1.conf",
"data": "config1",
"is_authfile": False
},
{
"name": "file1.key",
"data": to_b64("some key"),
"is_authfile": True
},
{
"name": "name2.conf",
"data": "config2",
"is_authfile": False
},
{
"name": "file2.key",
"data": to_b64("another key"),
"is_authfile": True
}
],
json.loads(data["data_json"][0])
)
assert_report_item_list_equal(
self.mock_reporter.report_item_list,
[
(
Severities.INFO,
report_codes.BOOTH_CONFIGS_SAVING_ON_NODE,
{"node": self.node.label}
),
(
Severities.WARNING,
report_codes.FILE_ALREADY_EXISTS,
{
"file_role": None,
"file_path": "name1.conf",
"node": self.node.label
}
),
(
Severities.WARNING,
report_codes.FILE_ALREADY_EXISTS,
{
"file_role": None,
"file_path": "file1.key",
"node": self.node.label
}
),
(
Severities.INFO,
report_codes.BOOTH_CONFIGS_SAVED_ON_NODE,
{
"node": self.node.label,
"name": "name2.conf, file2.key",
"name_list": ["name2.conf", "file2.key"]
}
)
]
)
def test_unable_to_parse_config(self, mock_read_authfile,
mock_read_configs, mock_parse,
mock_authfile):
def mock_parse_fn(config_data):
if config_data == "config1":
raise LibraryError()
elif config_data == "config2":
return "config2"
else:
raise AssertionError(
"unexpected input: {0}".format(config_data))
mock_parse.side_effect = mock_parse_fn
mock_authfile.return_value = "/path/to/file2.key"
mock_read_authfile.return_value = "another key".encode("utf-8")
mock_read_configs.return_value = {
"name1.conf": "config1",
"name2.conf": "config2"
}
self.mock_communicator.call_node.return_value = """
{
"existing": [],
"failed": {},
"saved": ["name2.conf", "file2.key"]
}
"""
lib.send_all_config_to_node(self.mock_communicator, self.mock_reporter,
self.node)
self.assertEqual(2, mock_parse.call_count)
mock_parse.assert_has_calls(
[mock.call("config1"), mock.call("config2")])
mock_authfile.assert_called_once_with("config2")
mock_read_authfile.assert_called_once_with(self.mock_reporter,
"/path/to/file2.key")
mock_read_configs.assert_called_once_with(self.mock_reporter, False)
self.assertEqual(1, self.mock_communicator.call_node.call_count)
self.assertEqual(self.node,
self.mock_communicator.call_node.call_args[0][0])
self.assertEqual("remote/booth_save_files",
self.mock_communicator.call_node.call_args[0][1])
data = url_decode(self.mock_communicator.call_node.call_args[0][2])
self.assertFalse("rewrite_existing" in data)
self.assertTrue("data_json" in data)
self.assertEqual([{
"name": "name2.conf",
"data": "config2",
"is_authfile": False
}, {
"name": "file2.key",
"data": to_b64("another key"),
"is_authfile": True
}], json.loads(data["data_json"][0]))
assert_report_item_list_equal(
self.mock_reporter.report_item_list,
[(Severities.INFO, report_codes.BOOTH_CONFIG_DISTRIBUTION_STARTED,
{}),
(Severities.WARNING, report_codes.BOOTH_SKIPPING_CONFIG, {
"config_file": "name1.conf"
}),
(Severities.INFO, report_codes.BOOTH_CONFIG_ACCEPTED_BY_NODE, {
"node": self.node.label,
"name_list": ["name2.conf", "file2.key"]
})])
def test_write_failure(self, mock_read_authfile, mock_read_configs,
mock_parse, mock_authfile):
mock_parse.side_effect = self.mock_parse_fn
mock_authfile.side_effect = self.mock_authfile_fn
mock_read_authfile.side_effect = self.mock_read_authfile_fn
mock_read_configs.return_value = {
"name1.conf": "config1",
"name2.conf": "config2"
}
self.mock_communicator.call_node.return_value = """
{
"existing": [],
"failed": {
"name1.conf": "Error message",
"file1.key": "Another error message"
},
"saved": ["name2.conf", "file2.key"]
}
"""
assert_raise_library_error(
lambda: lib.send_all_config_to_node(self.mock_communicator, self.
mock_reporter, self.node),
(Severities.ERROR,
report_codes.BOOTH_CONFIG_DISTRIBUTION_NODE_ERROR, {
"node": self.node.label,
"name": "name1.conf",
"reason": "Error message"
}), (Severities.ERROR,
report_codes.BOOTH_CONFIG_DISTRIBUTION_NODE_ERROR, {
"node": self.node.label,
"name": "file1.key",
"reason": "Another error message"
}))
self.assertEqual(2, mock_parse.call_count)
mock_parse.assert_has_calls(
[mock.call("config1"), mock.call("config2")])
self.assertEqual(2, mock_authfile.call_count)
mock_authfile.assert_has_calls(
[mock.call("config1"), mock.call("config2")])
self.assertEqual(2, mock_read_authfile.call_count)
mock_read_authfile.assert_has_calls([
mock.call(self.mock_reporter, "/path/to/file1.key"),
mock.call(self.mock_reporter, "/path/to/file2.key")
])
mock_read_configs.assert_called_once_with(self.mock_reporter, False)
self.assertEqual(1, self.mock_communicator.call_node.call_count)
self.assertEqual(self.node,
self.mock_communicator.call_node.call_args[0][0])
self.assertEqual("remote/booth_save_files",
self.mock_communicator.call_node.call_args[0][1])
data = url_decode(self.mock_communicator.call_node.call_args[0][2])
self.assertFalse("rewrite_existing" in data)
self.assertTrue("data_json" in data)
self.assertEqual([{
"name": "name1.conf",
"data": "config1",
"is_authfile": False
}, {
"name": "file1.key",
"data": to_b64("some key"),
"is_authfile": True
}, {
"name": "name2.conf",
"data": "config2",
"is_authfile": False
}, {
"name": "file2.key",
"data": to_b64("another key"),
"is_authfile": True
}], json.loads(data["data_json"][0]))
assert_report_item_list_equal(
self.mock_reporter.report_item_list,
[(Severities.INFO, report_codes.BOOTH_CONFIG_DISTRIBUTION_STARTED,
{}),
(Severities.ERROR,
report_codes.BOOTH_CONFIG_DISTRIBUTION_NODE_ERROR, {
"node": self.node.label,
"name": "name1.conf",
"reason": "Error message"
}),
(Severities.ERROR,
report_codes.BOOTH_CONFIG_DISTRIBUTION_NODE_ERROR, {
"node": self.node.label,
"name": "file1.key",
"reason": "Another error message"
})])
def test_rewrite_existing(self, mock_read_authfile, mock_read_configs,
mock_parse, mock_authfile):
mock_parse.side_effect = self.mock_parse_fn
mock_authfile.side_effect = self.mock_authfile_fn
mock_read_authfile.side_effect = self.mock_read_authfile_fn
mock_read_configs.return_value = {
"name1.conf": "config1",
"name2.conf": "config2"
}
self.mock_communicator.call_node.return_value = """
{
"existing": ["name1.conf", "file1.key"],
"failed": {},
"saved": ["name2.conf", "file2.key"]
}
"""
lib.send_all_config_to_node(self.mock_communicator,
self.mock_reporter,
self.node,
rewrite_existing=True)
mock_read_configs.assert_called_once_with(self.mock_reporter, False)
self.assertEqual(2, mock_parse.call_count)
mock_parse.assert_has_calls(
[mock.call("config1"), mock.call("config2")])
self.assertEqual(2, mock_authfile.call_count)
mock_authfile.assert_has_calls(
[mock.call("config1"), mock.call("config2")])
self.assertEqual(2, mock_read_authfile.call_count)
mock_read_authfile.assert_has_calls([
mock.call(self.mock_reporter, "/path/to/file1.key"),
mock.call(self.mock_reporter, "/path/to/file2.key")
])
self.assertEqual(1, self.mock_communicator.call_node.call_count)
self.assertEqual(self.node,
self.mock_communicator.call_node.call_args[0][0])
self.assertEqual("remote/booth_save_files",
self.mock_communicator.call_node.call_args[0][1])
data = url_decode(self.mock_communicator.call_node.call_args[0][2])
self.assertTrue("rewrite_existing" in data)
self.assertTrue("data_json" in data)
self.assertEqual([{
"name": "name1.conf",
"data": "config1",
"is_authfile": False
}, {
"name": "file1.key",
"data": to_b64("some key"),
"is_authfile": True
}, {
"name": "name2.conf",
"data": "config2",
"is_authfile": False
}, {
"name": "file2.key",
"data": to_b64("another key"),
"is_authfile": True
}], json.loads(data["data_json"][0]))
assert_report_item_list_equal(
self.mock_reporter.report_item_list,
[(Severities.INFO, report_codes.BOOTH_CONFIG_DISTRIBUTION_STARTED,
{}),
(Severities.WARNING, report_codes.FILE_ALREADY_EXISTS, {
"file_role": None,
"file_path": "name1.conf",
"node": self.node.label
}),
(Severities.WARNING, report_codes.FILE_ALREADY_EXISTS, {
"file_role": None,
"file_path": "file1.key",
"node": self.node.label
}),
(Severities.INFO, report_codes.BOOTH_CONFIG_ACCEPTED_BY_NODE, {
"node": self.node.label,
"name_list": ["name2.conf", "file2.key"]
})])
