def test_compress(self, mock_lib, mock_fetch):
"""Function: test_compress
Description: Test with compression option.
Arguments:
"""
mock_lib.chk_crt_dir.return_value = (True, None)
mock_lib.dir_file_match.side_effect = [["File1", "File2"],
["File1", "File2"]]
mock_lib.mv_file.return_value = True
mock_lib.is_missing_lists.return_value = ["File1"]
mock_lib.compress.return_value = True
mock_fetch.return_value = {
"parsed": {
"systemLog": {
"path": self.filepath
}
}
}
self.assertEqual(mongo_db_admin.rotate(self.server, self.args_array3),
(False, None))
def test_no_log_dir(self):
"""Function: test_no_log_dir
Description: Test with no log directory passed.
Arguments:
"""
self.assertEqual(mongo_db_admin.rotate(self.server, self.args_array),
(False, None))
def test_file_chk_fail(self, mock_check):
"""Function: test_file_chk_fail
Description: Test with file checking fails.
Arguments:
"""
mock_check.return_value = (False, "ErrorMsg")
self.assertEqual(mongo_db_admin.rotate(self.server, self.args_array2),
(True, "ErrorMsg"))
def test_rotate(self, mock_check, mock_fetch, mock_match, mock_mv,
mock_diff):
"""Function: test_rotate
Description: Test rotate function.
Arguments:
"""
mock_check.return_value = (True, None)
mock_fetch.return_value = {
"parsed": {
"systemLog": {
"path": self.filepath
}
}
}
mock_match.side_effect = [["File1", "File2"], ["File1", "File2"]]
mock_mv.return_value = True
mock_diff.return_value = ["File1"]
self.assertEqual(mongo_db_admin.rotate(self.server, self.args_array2),
(False, None))
def test_too_many_logs(self, mock_check, mock_fetch, mock_match,
mock_diff):
"""Function: test_too_many_logs
Description: Test with too many logs to rotate.
Arguments:
"""
err_msg = "Error: Too many files to move: ['File1', 'File2']"
mock_check.return_value = (True, None)
mock_fetch.return_value = {
"parsed": {
"systemLog": {
"path": self.filepath
}
}
}
mock_match.side_effect = [["File1", "File2"], ["File1"]]
mock_diff.return_value = ["File1", "File2"]
self.assertEqual(mongo_db_admin.rotate(self.server, self.args_array2),
(True, err_msg))