#
# rocprofv3 attachment test
#
cmake_minimum_required(VERSION 3.21.0 FATAL_ERROR)

project(
    rocprofiler-sdk-tests-rocprofv3-attachment-attach-once
    LANGUAGES CXX
    VERSION 0.0.0)

find_package(rocprofiler-sdk REQUIRED)

set(ROCPROFILER_MEMCHECK_TYPES "AddressSanitizer" "UndefinedBehaviorSanitizer"
                               "ThreadSanitizer")

if(ROCPROFILER_MEMCHECK AND ROCPROFILER_MEMCHECK IN_LIST ROCPROFILER_MEMCHECK_TYPES)
    set(IS_DISABLED ON)
else()
    set(IS_DISABLED OFF)
endif()

if(ROCPROFILER_MEMCHECK STREQUAL "LeakSanitizer")
    set(LOG_LEVEL "warning") # info produces memory leak
else()
    set(LOG_LEVEL "info")
endif()

string(REPLACE "LD_PRELOAD=" "ROCPROF_PRELOAD=" PRELOAD_ENV
               "${ROCPROFILER_MEMCHECK_PRELOAD_ENV}")

set(attachment-env
    "LD_LIBRARY_PATH=$<TARGET_FILE_DIR:rocprofiler-sdk::rocprofiler-sdk-shared-library>:$ENV{LD_LIBRARY_PATH}"
    )

# Test that launches the app and attaches to it (CSV format)
rocprofiler_add_integration_execute_test(
    rocprofv3-test-attachment-attach-once
    COMMAND
        ${CMAKE_CURRENT_SOURCE_DIR}/run_attachment_test_unified.sh
        $<TARGET_FILE:attachment-test> $<TARGET_FILE:rocprofiler-sdk::rocprofv3>
        ${CMAKE_CURRENT_BINARY_DIR} ${LOG_LEVEL} out
    DEPENDS rocprofiler-sdk::rocprofv3
    TIMEOUT 60
    LABELS "integration-tests;attachment"
    DISABLED ${IS_DISABLED}
    PRELOAD "${PRELOAD_ENV}"
    ENVIRONMENT "${attachment-env}"
    SKIP_REGULAR_EXPRESSION "This test is skipped."
    FIXTURES_SETUP rocprofv3-test-attachment-attach-once
    FAIL_REGULAR_EXPRESSION "ERROR|FATAL|${ROCPROFILER_DEFAULT_FAIL_REGEX}")
# Validate the output from the attached profiling
rocprofiler_add_integration_validate_test(
    rocprofv3-test-attachment-attach-once-csv
    TEST_PATHS validate.py
    COPY conftest.py
    CONFIG pytest.ini
    TIMEOUT 30
    LABELS "integration-tests;attachment"
    DISABLED ${IS_DISABLED}
    SKIP_REGULAR_EXPRESSION "SKIPPED"
    FIXTURES_REQUIRED rocprofv3-test-attachment-attach-once
    ARGS --kernel-input
         ${CMAKE_CURRENT_BINARY_DIR}/attachment-output/out_kernel_trace.csv
         --hsa-input
         ${CMAKE_CURRENT_BINARY_DIR}/attachment-output/out_hsa_api_trace.csv
         --memory-copy-input
         ${CMAKE_CURRENT_BINARY_DIR}/attachment-output/out_memory_copy_trace.csv
         --agent-input
         ${CMAKE_CURRENT_BINARY_DIR}/attachment-output/out_agent_info.csv
         --skip-if
         ${CMAKE_CURRENT_BINARY_DIR}/attachment-output/skipped)

rocprofiler_add_integration_validate_test(
    rocprofv3-test-attachment-attach-once-json
    TEST_PATHS validate.py
    COPY conftest.py
    CONFIG pytest.ini
    TIMEOUT 30
    LABELS "integration-tests;attachment"
    DISABLED ${IS_DISABLED}
    SKIP_REGULAR_EXPRESSION "SKIPPED"
    FIXTURES_REQUIRED rocprofv3-test-attachment-attach-once
    ARGS --kernel-input
         ${CMAKE_CURRENT_BINARY_DIR}/attachment-output/out_results.json
         --hsa-input
         ${CMAKE_CURRENT_BINARY_DIR}/attachment-output/out_results.json
         --memory-copy-input
         ${CMAKE_CURRENT_BINARY_DIR}/attachment-output/out_results.json
         --agent-input
         ${CMAKE_CURRENT_BINARY_DIR}/attachment-output/out_results.json
         --skip-if
         ${CMAKE_CURRENT_BINARY_DIR}/attachment-output/skipped)
