# Copyright (c) 2021-2024 Huawei Device Co., Ltd.
# 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.

import("../../../../core/common/dfx/dsoftbus_dfx.gni")
import("../../../../dsoftbus.gni")
import("../../../../sdk/transmission/trans_channel/udp/trans_udp_sdk.gni")

dsoftbus_root_path = "../../../.."

trans_sdk_test_common_inc = [
  "$dsoftbus_root_path/core/common/include",
  "$dsoftbus_root_path/core/connection/interface",
  "$dsoftbus_root_path/sdk/transmission/session/include",
  "$dsoftbus_root_path/interfaces/inner_kits/transport",
  "$dsoftbus_root_path/interfaces/kits/transport",
  "$dsoftbus_root_path/tests/sdk/common/include",
  "$dsoftbus_root_path/sdk/transmission/trans_channel/tcp_direct/src",
  "$dsoftbus_root_path/adapter/common/include",
  "$dsoftbus_root_path/sdk/transmission/trans_channel/tcp_direct/include",
  "$dsoftbus_root_path/sdk/transmission/session/include",
  "$dsoftbus_root_path/core/frame/common/include",
  "$dsoftbus_root_path/sdk/transmission/trans_channel/udp/common/include",
  "$dsoftbus_root_path/core/transmission/trans_channel/udp_negotiation/include",
  "$dsoftbus_root_path/sdk/transmission/trans_channel/udp/file/include",
  "$dsoftbus_root_path/sdk/transmission/ipc/include",
]
trans_sdk_test_common_deps = [
  "$dsoftbus_root_path/core/common:softbus_utils",
  "$dsoftbus_root_path/sdk:softbus_client",
  "$dsoftbus_root_path/tests/sdk/common:softbus_access_token_test",
]

if (defined(ohos_lite)) {
  import("//build/lite/config/component/lite_component.gni")
  import("//build/lite/config/test.gni")

  if (ohos_build_type == "debug") {
    unittest("TransSdkStreamTest") {
      output_extension = "bin"
      output_dir = "$root_out_dir/test/unittest/dsoftbus"
      sources = [ "udp/stream/trans_sdk_stream_test.cpp" ]
      include_dirs = trans_sdk_test_common_inc
      include_dirs += [
        "$dsoftbus_root_path/core/common/softbus_property/include",
        "$softbus_adapter_config/spec_config",
        "$dsoftbus_root_path/adapter/common/include/OS_adapter_define/linux",
      ]
      deps = trans_sdk_test_common_deps
      deps += [
        "$dsoftbus_root_path/adapter:softbus_adapter",
        "$hilog_lite_deps_path",
      ]
    }
  }
} else {
  import("//build/test.gni")

  trans_sdk_proxy_test_inc = [
    "$dsoftbus_root_path/sdk/transmission/trans_channel/proxy/include",
    "$dsoftbus_root_path/sdk/transmission/trans_channel/udp/file/include",
  ]

  trans_sdk_proxy_test_deps =
      [ "$dsoftbus_root_path/tests/sdk:softbus_client_static" ]

  module_output_path = "dsoftbus/transmission"
  ohos_unittest("TransSdkStreamTest") {
    module_out_path = module_output_path
    sources = [ "udp/stream/trans_sdk_stream_test.cpp" ]
    include_dirs = trans_sdk_test_common_inc
    deps = trans_sdk_test_common_deps
    deps += trans_sdk_proxy_test_deps
    if (is_standard_system) {
      external_deps = [ "hilog:libhilog" ]
    } else {
      external_deps = [ "hilog:libhilog" ]
    }
  }

  module_output_path = "dsoftbus/transmission"
  ohos_unittest("TransSdkCommStreamTest") {
    module_out_path = module_output_path
    sources = [ "udp/stream/trans_sdk_comm_stream_test.cpp" ]
    include_dirs = trans_sdk_test_common_inc
    include_dirs += trans_sdk_proxy_test_inc
    deps = trans_sdk_test_common_deps
    deps += trans_sdk_proxy_test_deps
    if (is_standard_system) {
      external_deps = [ "hilog:libhilog" ]
    } else {
      external_deps = [ "hilog:libhilog" ]
    }
  }

  module_output_path = "dsoftbus/transmission"
  ohos_unittest("ClientTransProxyFileCommonTest") {
    module_out_path = module_output_path
    sources = [ "proxy/client_trans_proxy_file_common_test.cpp" ]
    include_dirs = trans_sdk_test_common_inc
    include_dirs += trans_sdk_proxy_test_inc
    deps = trans_sdk_test_common_deps
    deps += trans_sdk_proxy_test_deps
    if (is_standard_system) {
      external_deps = [ "hilog:libhilog" ]
    } else {
      external_deps = [ "hilog:libhilog" ]
    }
  }

  module_output_path = "dsoftbus/transmission"
  ohos_unittest("ClientTransPendingTest") {
    module_out_path = module_output_path
    sources = [ "proxy/client_trans_pending_test.cpp" ]
    include_dirs = trans_sdk_test_common_inc
    include_dirs += trans_sdk_proxy_test_inc
    deps = trans_sdk_test_common_deps
    deps += trans_sdk_proxy_test_deps
    if (is_standard_system) {
      external_deps = [ "hilog:libhilog" ]
    } else {
      external_deps = [ "hilog:libhilog" ]
    }
  }

  module_output_path = "dsoftbus/transmission"
  ohos_unittest("ClientTransProxyManagerTest") {
    sanitize = {
      cfi = true
      cfi_cross_dso = true
      debug = false
    }
    module_out_path = module_output_path
    sources = [ "proxy/client_trans_proxy_manager_test.cpp" ]
    include_dirs = trans_sdk_test_common_inc
    include_dirs += trans_sdk_proxy_test_inc
    include_dirs +=
        [ "$dsoftbus_root_path/sdk/transmission/trans_channel/proxy/src" ]
    deps = trans_sdk_test_common_deps
    deps += trans_sdk_proxy_test_deps
    if (is_standard_system) {
      external_deps = [ "hilog:libhilog" ]
    } else {
      external_deps = [ "hilog:libhilog" ]
    }
  }

  module_output_path = "dsoftbus/transmission"
  ohos_unittest("ClientTransUdpManagerTest") {
    sanitize = {
      cfi = true
      cfi_cross_dso = true
      debug = false
    }
    module_out_path = module_output_path
    sources = [ "udp/common/client_trans_udp_manager_test.cpp" ]
    include_dirs = trans_sdk_test_common_inc
    include_dirs += trans_sdk_proxy_test_inc
    deps = trans_sdk_test_common_deps
    deps += trans_sdk_proxy_test_deps
    if (is_standard_system) {
      external_deps = [ "hilog:libhilog" ]
    } else {
      external_deps = [ "hilog:libhilog" ]
    }
  }

  module_output_path = "dsoftbus/transmission"
  ohos_unittest("ClientTransUdpManagerStaticTest") {
    module_out_path = module_output_path
    sources = [ "udp/common/client_trans_udp_manager_static_test.cpp" ]
    include_dirs = trans_sdk_test_common_inc
    include_dirs += [
      "$dsoftbus_root_path/sdk/transmission/trans_channel/udp/common/src",
      "$dsoftbus_root_path/sdk/transmission/trans_channel/udp/file/include",
      "$dsoftbus_root_path/sdk/transmission/trans_channel/udp/stream/include",
      "$dsoftbus_root_path/sdk/transmission/ipc/include",
      "$dsoftbus_root_path/sdk/transmission/trans_channel/manager/include",
    ]
    deps = trans_sdk_test_common_deps
    deps += trans_sdk_proxy_test_deps
    native_source_path = rebase_path("$dsoftbus_root_path")
    dep_file = "components/nstackx_enhanced/nstackx_core/dfile/BUILD.gn"
    enhanced = exec_script("$dsoftbus_root_path/check_sub_module.py",
                           [
                             "$native_source_path",
                             "$dep_file",
                           ],
                           "value")

    if (enhanced) {
      include_dirs += [ "$dsoftbus_root_path/components/nstackx_enhanced/nstackx_util/interface" ]
      deps += [ "$dsoftbus_root_path/components/nstackx_enhanced/nstackx_core/dfile:nstackx_dfile" ]
    } else {
      include_dirs +=
          [ "$dsoftbus_root_path/components/nstackx/nstackx_util/interface" ]
      deps += [ "$dsoftbus_root_path/components/nstackx/nstackx_core/dfile:nstackx_dfile.open" ]
    }
    if (is_standard_system) {
      external_deps = [ "hilog:libhilog" ]
    } else {
      external_deps = [ "hilog:libhilog" ]
    }
  }

  module_output_path = "dsoftbus/transmission"
  ohos_unittest("ClientTransProxyFileManagerTest") {
    sanitize = {
      cfi = true
      cfi_cross_dso = true
      debug = false
    }
    module_out_path = module_output_path
    sources = [ "proxy/client_trans_proxy_file_manager_test.cpp" ]
    include_dirs = trans_sdk_test_common_inc
    include_dirs += [
      "$dsoftbus_root_path/sdk/transmission/trans_channel/proxy/include",
      "$dsoftbus_root_path/sdk/transmission/trans_channel/udp/file/include",
      "$dsoftbus_root_path/sdk/transmission/ipc/include",
      "$dsoftbus_root_path/sdk/bus_center/manager/include",
      "$dsoftbus_root_path/sdk/transmission/trans_channel/proxy/src",
      "$dsoftbus_root_path/sdk/transmission/session/src",
      "$dsoftbus_root_path/sdk/transmission/trans_channel/udp/file/include",
      "$dsoftbus_root_path/sdk/transmission/trans_channel/manager/include",
      "$dsoftbus_root_path/core/frame/common/include",
      "$dsoftbus_root_path/core/transmission/common/include",
      "$dsoftbus_root_path/core/bus_center/lnn/lane_hub/lane_manager/include",
      "$dsoftbus_root_path/interfaces/inner_kits/lnn",
    ]
    deps = trans_sdk_test_common_deps
    deps += trans_sdk_proxy_test_deps
    if (is_standard_system) {
      external_deps = [ "hilog:libhilog" ]
    } else {
      external_deps = [ "hilog:libhilog" ]
    }
  }

  module_output_path = "dsoftbus/transmission"
  ohos_unittest("ClientTransProxyFileManagerMockTest") {
    module_out_path = module_output_path
    sources = [
      "proxy/client_trans_proxy_file_manager_mock_test.cpp",
      "proxy/mock/client_trans_proxy_file_manager_mock.cpp",
    ]
    include_dirs = trans_sdk_test_common_inc
    include_dirs += [
      "$dsoftbus_root_path/sdk/transmission/trans_channel/proxy/include",
      "$dsoftbus_root_path/sdk/transmission/trans_channel/udp/file/include",
      "$dsoftbus_root_path/sdk/transmission/ipc/include",
      "$dsoftbus_root_path/sdk/bus_center/manager/include",
      "$dsoftbus_root_path/sdk/transmission/trans_channel/proxy/src",
      "$dsoftbus_root_path/sdk/transmission/trans_channel/udp/file/include",
      "$dsoftbus_root_path/sdk/transmission/trans_channel/manager/include",
      "$dsoftbus_root_path/core/frame/common/include",
      "$dsoftbus_root_path/core/transmission/common/include",
      "$dsoftbus_root_path/core/bus_center/lnn/lane_hub/lane_manager/include",
      "$dsoftbus_root_path/interfaces/inner_kits/lnn",
      "$dsoftbus_root_path/tests/sdk/transmission/trans_channel/proxy/mock",
    ]
    deps = trans_sdk_test_common_deps
    deps += trans_sdk_proxy_test_deps
    external_deps = [
      "googletest:gmock",
      "hilog:libhilog",
    ]
  }

  module_output_path = "dsoftbus/transmission"
  ohos_unittest("ClientTransChannelManagerTest") {
    sanitize = {
      cfi = true
      cfi_cross_dso = true
      debug = false
    }
    module_out_path = module_output_path
    sources = [ "manager/client_trans_channel_manager_test.cpp" ]
    include_dirs = trans_sdk_test_common_inc
    include_dirs += [
      "$dsoftbus_root_path/sdk/transmission/trans_channel/manager/include",
      "$dsoftbus_root_path/sdk/transmission/trans_channel/auth/include",
      "$dsoftbus_root_path/sdk/transmission/trans_channel/proxy/include",
      "$dsoftbus_root_path/sdk/transmission/trans_channel/udp/file/include",
      "$dsoftbus_root_path/sdk/transmission/trans_channel/udp/common/include",
    ]
    deps = trans_sdk_test_common_deps
    deps += trans_sdk_proxy_test_deps
    if (is_standard_system) {
      external_deps = [ "hilog:libhilog" ]
    } else {
      external_deps = [ "hilog:libhilog" ]
    }
  }

  module_output_path = "dsoftbus/transmission"
  ohos_unittest("ClientTransChannelCallbackTest") {
    sanitize = {
      cfi = true
      cfi_cross_dso = true
      debug = false
    }
    module_out_path = module_output_path
    sources = [ "manager/client_trans_channel_callback_test.cpp" ]
    include_dirs = trans_sdk_test_common_inc
    include_dirs += [
      "$dsoftbus_root_path/sdk/transmission/trans_channel/manager/include",
      "$dsoftbus_root_path/sdk/transmission/trans_channel/auth/include",
      "$dsoftbus_root_path/sdk/transmission/trans_channel/proxy/include",
      "$dsoftbus_root_path/sdk/transmission/trans_channel/udp/file/include",
      "$dsoftbus_root_path/sdk/transmission/trans_channel/udp/common/include",
    ]
    deps = trans_sdk_test_common_deps
    deps += trans_sdk_proxy_test_deps
    if (is_standard_system) {
      external_deps = [ "hilog:libhilog" ]
    } else {
      external_deps = [ "hilog:libhilog" ]
    }
  }

  module_output_path = "dsoftbus/transmission"
  ohos_unittest("ClientTransSessionImplTest") {
    module_out_path = module_output_path
    sources = [ "manager/client_trans_session_impl_test.cpp" ]
    include_dirs = trans_sdk_test_common_inc
    include_dirs += [
      "$dsoftbus_root_path/sdk/transmission/trans_channel/manager/include",
      "$dsoftbus_root_path/sdk/transmission/session/cpp/src",
      "$dsoftbus_root_path/sdk/transmission/session/cpp/include",
      "$dsoftbus_root_path/sdk/transmission/trans_channel/auth/include",
      "$dsoftbus_root_path/sdk/transmission/trans_channel/proxy/include",
      "$dsoftbus_root_path/sdk/transmission/trans_channel/udp/file/include",
      "$dsoftbus_root_path/sdk/transmission/trans_channel/udp/common/include",
    ]
    deps = trans_sdk_test_common_deps
    deps += trans_sdk_proxy_test_deps
    if (is_standard_system) {
      external_deps = [ "hilog:libhilog" ]
    } else {
      external_deps = [ "hilog:libhilog" ]
    }
  }

  ohos_unittest("ClientTransUdpStreamInterfaceTest") {
    sanitize = {
      cfi = true
      cfi_cross_dso = true
      debug = false
    }
    module_out_path = module_output_path
    sources = [ "udp/stream/client_trans_udp_stream_interface_test.cpp" ]
    include_dirs = trans_udp_channel_sdk_inc
    deps = trans_sdk_test_common_deps
    deps += trans_sdk_proxy_test_deps
    if (is_standard_system) {
      external_deps = [ "hilog:libhilog" ]
    } else {
      external_deps = [ "hilog:libhilog" ]
    }
  }

  ohos_unittest("ClientTransStreamTest") {
    module_out_path = module_output_path
    sources = [ "udp/stream/client_trans_stream_test.cpp" ]
    include_dirs = trans_udp_channel_sdk_inc
    include_dirs += trans_sdk_test_common_inc
    include_dirs += [
      "$dsoftbus_root_path/sdk/transmission/trans_channel/udp/stream/src",
      "$dsoftbus_root_path/sdk/transmission/ipc/standard/src",
      "$dsoftbus_root_path/sdk/transmission/ipc/standard/include",
      "$dsoftbus_root_path/core/frame/standard/init/include",
      "$dsoftbus_root_path/sdk/transmission/session/include",
    ]

    deps = trans_sdk_test_common_deps
    deps += trans_sdk_proxy_test_deps
    if (is_standard_system) {
      external_deps = [ "hilog:libhilog" ]
    } else {
      external_deps = [ "hilog:libhilog" ]
    }
  }

  ohos_unittest("StreamAdaptorTest") {
    module_out_path = module_output_path
    sources = [ "udp/stream/stream_adaptor_test.cpp" ]
    include_dirs = trans_udp_channel_sdk_inc
    deps = trans_sdk_test_common_deps
    deps += trans_sdk_proxy_test_deps
    if (is_standard_system) {
      external_deps = [ "hilog:libhilog" ]
    } else {
      external_deps = [ "hilog:libhilog" ]
    }
  }

  group("unittest") {
    testonly = true
    deps = [
      ":ClientTransChannelCallbackTest",
      ":ClientTransChannelManagerTest",
      ":ClientTransPendingTest",
      ":ClientTransProxyFileCommonTest",
      ":ClientTransProxyFileManagerMockTest",
      ":ClientTransProxyFileManagerTest",
      ":ClientTransProxyManagerTest",
      ":ClientTransSessionImplTest",
      ":ClientTransUdpManagerStaticTest",
      ":ClientTransUdpManagerTest",
      ":TransSdkCommStreamTest",
      ":TransSdkStreamTest",
      "auth:unittest",
      "monitor_permission_test:CheckPermissionTest",
      "permission:TransDynamicPermissionTest",
      "qos:unittest",
      "udp/distributed_stream:distributed_stream_test",
      "udp/file:unittest",
    ]
    if (dsoftbus_feature_trans_udp == true) {
      if (dsoftbus_feature_trans_udp_stream == true) {
        if (!defined(ohos_lite)) {
          deps += [
            ":ClientTransStreamTest",
            ":ClientTransUdpStreamInterfaceTest",
            ":StreamAdaptorTest",
            "tcp_direct:unittest",
            "udp/stream/libsoftbus_stream:unittest",
          ]
        }
      }
    }
  }
}