extension/module c++ test (#3710)

Summary:
* Install extension_module_static as well
* Keep all symbols (whole-archive) for portable ops
* Add test for extension/module

Pull Request resolved: #3710

Test Plan: `sh test/run_oss_cpp_tests.sh extension/module/test`

Reviewed By: shoumikhin

Differential Revision: D57940456

Pulled By: kirklandsign

fbshipit-source-id: 4a34633250a8c06c9d0db99cd1320cb2b53824a4

Author: kirklandsign

build/Test.cmake

@@ -12,11 +12,13 @@
 #
 # This file should be formatted with
 # ~~~
-# cmake-format -i Utils.cmake
+# cmake-format -i Test.cmake
 # ~~~
 # It should also be cmake-lint clean.
 #
 
+include(${EXECUTORCH_ROOT}/build/Utils.cmake)
+
 # Add code coverage flags to supported compilers
 if(EXECUTORCH_USE_CPP_CODE_COVERAGE)
   if("${CMAKE_CXX_COMPILER_ID}" STREQUAL "GNU")
@@ -54,6 +56,9 @@ cmake_parse_arguments(ET_CXX_TEST "" "" "${multi_arg_names}" ${ARGN})
 # Find prebuilt executorch library
 find_package(executorch CONFIG REQUIRED)
 
+target_link_options_shared_lib(extension_data_loader)
+target_link_options_shared_lib(portable_ops_lib)
+
 enable_testing()
 find_package(GTest CONFIG REQUIRED)
 

build/executorch-config.cmake

@@ -43,6 +43,7 @@ set(lib_list
     qnn_executorch_backend
     portable_ops_lib
     extension_module
+    extension_module_static
     xnnpack_backend
     XNNPACK
     cpuinfo

extension/module/CMakeLists.txt

@@ -46,7 +46,7 @@ target_compile_options(
 
 # Install libraries
 install(
-  TARGETS extension_module
+  TARGETS extension_module extension_module_static
   DESTINATION lib
   INCLUDES
   DESTINATION ${_common_include_directories}

extension/module/test/CMakeLists.txt

@@ -0,0 +1,37 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the BSD-style license found in the
+# LICENSE file in the root directory of this source tree.
+
+# @generated by test/utils/generate_gtest_cmakelists.py
+#
+# This file should be formatted with
+# ~~~
+# cmake-format -i CMakeLists.txt
+# ~~~
+# It should also be cmake-lint clean.
+#
+
+cmake_minimum_required(VERSION 3.19)
+project(extension_module_test)
+
+# Use C++17 for test.
+set(CMAKE_CXX_STANDARD 17)
+
+set(EXECUTORCH_ROOT ${CMAKE_CURRENT_SOURCE_DIR}/../../..)
+
+include(${EXECUTORCH_ROOT}/build/Test.cmake)
+
+set(_test_srcs module_test.cpp)
+
+et_cxx_test(
+  extension_module_test
+  SOURCES
+  ${_test_srcs}
+  EXTRA_LIBS
+  extension_data_loader
+  extension_module_static
+  portable_kernels
+  portable_ops_lib
+)

extension/module/test/module_test.cpp

@@ -9,6 +9,7 @@
 #include <gtest/gtest.h>
 
 #include <executorch/extension/module/module.h>
+#include <array>
 
 using namespace ::testing;
 

test/run_oss_cpp_tests.sh

@@ -27,6 +27,7 @@ build_executorch() {
     -DCMAKE_INSTALL_PREFIX=cmake-out \
     -DEXECUTORCH_USE_CPP_CODE_COVERAGE=ON \
     -DEXECUTORCH_BUILD_EXTENSION_DATA_LOADER=ON \
+    -DEXECUTORCH_BUILD_EXTENSION_MODULE=ON \
     -Bcmake-out
   cmake --build cmake-out -j9 --target install
 }
@@ -49,6 +50,8 @@ build_and_run_test() {
     -Bcmake-out/"${test_dir}"
   cmake --build cmake-out/"${test_dir}" -j9
 
+  export RESOURCES_PATH=extension/module/test/resources
+
   for t in cmake-out/"${test_dir}"/*test; do
     if [ -e "$t" ]; then
       LLVM_PROFILE_FILE="cmake-out/$(basename $t).profraw" ./"$t";

test/utils/OSSTestConfig.json

@@ -11,6 +11,19 @@
             "extension_data_loader"
         ]
     },
+    {
+        "directory": "extension/module/test",
+        "sources": [
+            "module_test.cpp"
+        ],
+        "additional_libs": [
+            "extension_data_loader",
+            "extension_module_static",
+            "portable_kernels",
+            "portable_ops_lib"
+
+        ]
+    },
     {
         "directory": "runtime/core/portable_type/test",
         "sources": [