def test_collect_logs(self, mkdirs_mock, copy_mock, tarfile_open_mock,
                          get_files_mock, server_log_mock, launcher_log_mock):
        downloaded_logs_loc = path.join(TMP_PRESTO_DEBUG, "logs")

        collect.logs()

        mkdirs_mock.assert_called_with(downloaded_logs_loc)
        copy_mock.assert_called_with(
            path.join(get_log_directory(), PRESTOADMIN_LOG_NAME),
            downloaded_logs_loc)

        tarfile_open_mock.assert_called_with(OUTPUT_FILENAME_FOR_LOGS, 'w:gz')
        tar = tarfile_open_mock.return_value
        tar.add.assert_called_with(downloaded_logs_loc,
                                   arcname=path.basename(downloaded_logs_loc))
Example #2
0
    def test_collect_logs(self, mkdirs_mock, copy_mock,
                          tarfile_open_mock, get_files_mock, server_log_mock,
                          launcher_log_mock):
        downloaded_logs_loc = path.join(TMP_PRESTO_DEBUG, "logs")

        collect.logs()

        mkdirs_mock.assert_called_with(downloaded_logs_loc)
        copy_mock.assert_called_with(path.join(get_log_directory(),
                                               PRESTOADMIN_LOG_NAME),
                                     downloaded_logs_loc)

        tarfile_open_mock.assert_called_with(OUTPUT_FILENAME_FOR_LOGS, 'w:bz2')
        tar = tarfile_open_mock.return_value
        tar.add.assert_called_with(downloaded_logs_loc,
                                   arcname=path.basename(downloaded_logs_loc))
Example #3
0
    def test_collect_logs(self, path_exists_mock, mkdirs_mock, copy_mock,
                          tarfile_open_mock, file_get_mock, server_log_mock,
                          launcher_log_mock):
        downloaded_logs_loc = path.join(TMP_PRESTO_DEBUG, "logs")
        path_exists_mock.return_value = False

        collect.logs()

        mkdirs_mock.assert_called_with(downloaded_logs_loc)
        copy_mock.assert_called_with(
            path.join(PRESTOADMIN_LOG_DIR, PRESTOADMIN_LOG_NAME),
            downloaded_logs_loc)

        tarfile_open_mock.assert_called_with(OUTPUT_FILENAME_FOR_LOGS, 'w:bz2')
        tar = tarfile_open_mock.return_value
        tar.add.assert_called_with(downloaded_logs_loc,
                                   arcname=path.basename(downloaded_logs_loc))
Example #4
0
    def test_collect_logs(self, path_exists_mock, mkdirs_mock, copy_mock,
                          tarfile_open_mock, mock_execute):
        downloaded_logs_loc = path.join(TMP_PRESTO_DEBUG, "logs")
        path_exists_mock.return_value = False

        collect.logs()

        mkdirs_mock.assert_any_call(TMP_PRESTO_DEBUG)
        mkdirs_mock.assert_called_with(downloaded_logs_loc)
        copy_mock.assert_called_with(
            path.join(PRESTOADMIN_LOG_DIR, PRESTOADMIN_LOG_NAME),
            downloaded_logs_loc)

        mock_execute.assert_called_with(collect.file_get,
                                        REMOTE_PRESTO_LOG_DIR,
                                        downloaded_logs_loc,
                                        roles=[])
        tarfile_open_mock.assert_called_with(OUTPUT_FILENAME_FOR_LOGS, 'w:bz2')
        tar = tarfile_open_mock.return_value
        tar.add.assert_called_with(downloaded_logs_loc,
                                   arcname=path.basename(downloaded_logs_loc))
Example #5
0
    def test_collect_logs(self, path_exists_mock, mkdirs_mock, copy_mock,
                          tarfile_open_mock, mock_execute):
        downloaded_logs_loc = path.join(TMP_PRESTO_DEBUG, "logs")
        path_exists_mock.return_value = False

        collect.logs()

        mkdirs_mock.assert_any_call(TMP_PRESTO_DEBUG)
        mkdirs_mock.assert_called_with(downloaded_logs_loc)
        copy_mock.assert_called_with(path.join(PRESTOADMIN_LOG_DIR,
                                               PRESTOADMIN_LOG_NAME),
                                     downloaded_logs_loc)

        mock_execute.assert_called_with(collect.file_get,
                                        REMOTE_PRESTO_LOG_DIR,
                                        downloaded_logs_loc,
                                        roles=[])
        tarfile_open_mock.assert_called_with(OUTPUT_FILENAME_FOR_LOGS, 'w:bz2')
        tar = tarfile_open_mock.return_value
        tar.add.assert_called_with(downloaded_logs_loc,
                                   arcname=path.basename(downloaded_logs_loc))