# Copyright (C) 2019-2020 Zilliz. All rights reserved. # # Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance # with the License. You may obtain a copy of the License at # # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, software distributed under the License # is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express # or implied. See the License for the specific language governing permissions and limitations under the License include_directories(${CMAKE_HOME_DIRECTORY}/src) include_directories(${CMAKE_HOME_DIRECTORY}/src/thirdparty) include_directories(${CMAKE_CURRENT_SOURCE_DIR}) include_directories( ${KNOWHERE_INCLUDE_DIR} ${SIMDJSON_INCLUDE_DIR} ${TANTIVY_INCLUDE_DIR} ${CONAN_INCLUDE_DIRS} ${MILVUS_COMMON_INCLUDE_DIR} ${MILVUS_STORAGE_INCLUDE_DIR} ) add_definitions(-DMILVUS_TEST_SEGCORE_YAML_PATH="${CMAKE_CURRENT_SOURCE_DIR}/test_utils/test_segcore.yaml") # Collect test files from source directories using glob pattern file(GLOB_RECURSE SOURCE_TEST_FILES "${CMAKE_HOME_DIRECTORY}/src/**/*Test.cpp" "${CMAKE_HOME_DIRECTORY}/src/**/*_test.cpp" ) # TODO: better to use ls/find pattern set(MILVUS_TEST_FILES ${SOURCE_TEST_FILES} init_gtest.cpp test_loading.cpp test_exec.cpp test_expr_materialized_view.cpp test_float16.cpp test_group_by.cpp test_iterative_filter.cpp test_indexing.cpp test_index_wrapper.cpp test_integer_overflow.cpp test_query.cpp test_scorer.cpp test_sealed.cpp test_storage.cpp test_string_expr.cpp test_rust_result.cpp test_storage_v2_index_raw_data.cpp test_group_by_json.cpp ) if ( NOT (INDEX_ENGINE STREQUAL "cardinal") ) list(FILTER MILVUS_TEST_FILES EXCLUDE REGEX "KmeansClusteringTest\\.cpp$") endif() # need update aws-sdk-cpp, see more from https://github.com/aws/aws-sdk-cpp/issues/1757. # now we always remove this file from MILVUS_TEST_FILES thus it is never compiled. # once done, compile this test file only if `BUILD_DISK_ANN STREQUAL "ON"`. # if ( BUILD_DISK_ANN STREQUAL "OFF" ) list(FILTER MILVUS_TEST_FILES EXCLUDE REGEX "MinioChunkManagerTest\\.cpp$") # endif() # bitset has its own test binary list(FILTER MILVUS_TEST_FILES EXCLUDE REGEX "BitsetTest\\.cpp$") if (NOT (LINUX OR APPLE)) list(FILTER MILVUS_TEST_FILES EXCLUDE REGEX "ScalarIndexCreatorTest\\.cpp$") list(FILTER MILVUS_TEST_FILES EXCLUDE REGEX "StringIndexTest\\.cpp$") list(FILTER MILVUS_TEST_FILES EXCLUDE REGEX "ArrayTest\\.cpp$") list(FILTER MILVUS_TEST_FILES EXCLUDE REGEX "ExprArrayTest\\.cpp$") endif() if (ENABLE_AZURE_FS) set(AZURE_BUILD_DIR ON) add_definitions(-DAZURE_BUILD_DIR) else() list(FILTER MILVUS_TEST_FILES EXCLUDE REGEX "AzureChunkManagerTest\\.cpp$") list(FILTER MILVUS_TEST_FILES EXCLUDE REGEX "AzureBlobChunkManagerTest\\.cpp$") endif() # need update aws-sdk-cpp, see more from https://github.com/aws/aws-sdk-cpp/issues/2119 # once done, move this line to the else branch of `if (DEFINED AZURE_BUILD_DIR)` list(FILTER MILVUS_TEST_FILES EXCLUDE REGEX "RemoteChunkManagerTest\\.cpp$") if (ENABLE_GCP_NATIVE) add_definitions(-DENABLE_GCP_NATIVE) else() list(FILTER MILVUS_TEST_FILES EXCLUDE REGEX "GcpNativeChunkManagerTest\\.cpp$") endif() if (LINUX) message( STATUS "Building Milvus Unit Test on Linux") option(USE_ASAN "Whether to use AddressSanitizer" OFF) if ( USE_ASAN ) message( STATUS "Building Milvus using AddressSanitizer") add_compile_options(-fno-stack-protector -fno-omit-frame-pointer -fno-var-tracking -fsanitize=address) add_link_options(-fno-stack-protector -fno-omit-frame-pointer -fno-var-tracking -fsanitize=address) endif() endif() add_executable(all_tests ${MILVUS_TEST_FILES} ) target_link_libraries(all_tests gtest gmock milvus_core knowhere milvus-storage milvus-common ) install(TARGETS all_tests DESTINATION unittest) add_subdirectory(bench) add_subdirectory(test_json_stats) # Optionally include external scalar-benchmark project option(ENABLE_SCALAR_BENCH "Enable fetching and building scalar-benchmark" OFF) set(SCALAR_BENCHMARK_GIT_URL "https://github.com/zilliztech/scalar-benchmark" CACHE STRING "Scalar benchmark git repo URL") set(SCALAR_BENCHMARK_GIT_TAG "main" CACHE STRING "Scalar benchmark git tag/branch") if (ENABLE_SCALAR_BENCH) include(FetchContent) if (DEFINED SCALAR_BENCHMARK_SOURCE_DIR AND EXISTS ${SCALAR_BENCHMARK_SOURCE_DIR}/CMakeLists.txt) message(STATUS "Using local scalar-benchmark from ${SCALAR_BENCHMARK_SOURCE_DIR}") add_subdirectory(${SCALAR_BENCHMARK_SOURCE_DIR} ${CMAKE_CURRENT_BINARY_DIR}/scalar-benchmark-build) else() message(STATUS "Fetching scalar-benchmark from ${SCALAR_BENCHMARK_GIT_URL} (${SCALAR_BENCHMARK_GIT_TAG})") FetchContent_Declare( scalar_benchmark GIT_REPOSITORY ${SCALAR_BENCHMARK_GIT_URL} GIT_TAG ${SCALAR_BENCHMARK_GIT_TAG} SOURCE_DIR ${CMAKE_CURRENT_BINARY_DIR}/scalar-benchmark-src BINARY_DIR ${CMAKE_CURRENT_BINARY_DIR}/scalar-benchmark-build ) FetchContent_GetProperties(scalar_benchmark) if (NOT scalar_benchmark_POPULATED) FetchContent_Populate(scalar_benchmark) add_subdirectory(${scalar_benchmark_SOURCE_DIR} ${scalar_benchmark_BINARY_DIR}) endif() endif() endif() # bitset unit test include(CheckCXXCompilerFlag) include(CheckIncludeFileCXX) check_cxx_compiler_flag("-march=armv8-a+sve" COMPILER_SUPPORTS_SVE) check_include_file_cxx("arm_sve.h" COMPILER_HAS_ARM_SVE_HEADER) add_executable(bitset_test ${CMAKE_HOME_DIRECTORY}/src/bitset/BitsetTest.cpp ) if (COMPILER_SUPPORTS_SVE AND COMPILER_HAS_ARM_SVE_HEADER) message(STATUS "SVE support for the bitset library UT is enabled") target_compile_definitions(bitset_test PRIVATE BITSET_ENABLE_SVE_SUPPORT=1) set_source_files_properties(${CMAKE_HOME_DIRECTORY}/src/bitset/BitsetTest.cpp PROPERTIES COMPILE_FLAGS "-march=armv8-a+sve") else() message(STATUS "SVE support for the bitset library UT is disabled") endif() target_link_libraries(bitset_test milvus_bitset gtest ${CONAN_LIBS} ) install(TARGETS bitset_test DESTINATION unittest)