def test_upload_file(remove_mock, move_mock, chmod_mock, mock_rand, mock_time,
test_manager, input_file, output_file):
"""
Tests uploading a file to the manager
"""
input_file, output_file = getattr(test_manager, input_file), getattr(
test_manager, output_file)
with open(os.path.join(test_data_path, input_file)) as f:
xml_file = f.read()
m = mock_open(read_data=xml_file)
with patch('builtins.open', m):
upload_file(input_file, output_file, 'application/xml')
m.assert_any_call(
os.path.join(test_manager.api_tmp_path, 'api_tmp_file_0_0.xml'))
m.assert_any_call(
os.path.join(test_manager.api_tmp_path, 'api_tmp_file_0_0.xml'), 'w')
m.assert_any_call(os.path.join(test_data_path, input_file))
move_mock.assert_called_once_with(os.path.join(test_manager.api_tmp_path,
'api_tmp_file_0_0.xml'),
os.path.join(test_data_path,
output_file),
copy_function=copyfile)
remove_mock.assert_called_once_with(
os.path.join(test_data_path, input_file))
def test_upload_file(self):
# rules
upload_file(self.input_rules_file, self.output_rules_file,
'application/xml')
self.assertTrue(os.path.isfile(self.output_rules_file))
# decoders
upload_file(self.input_decoders_file, self.output_decoders_file,
'application/xml')
self.assertTrue(os.path.isfile(self.output_decoders_file))
# CDB lists
upload_file(self.input_lists_file, self.output_lists_file,
'application/octet-stream')
self.assertTrue(os.path.isfile(self.output_lists_file))