/*
 * Copyright (c) 2022-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.
 */

#include "UTTest_permission_manager.h"
#include "accesstoken_kit.h"
#include "nativetoken_kit.h"
#include "token_setproc.h"

namespace OHOS {
namespace DistributedHardware {
void PermissionManagerTest::SetUp()
{
    const int32_t permsNum = 2;
    const int32_t indexZero = 0;
    const int32_t indexOne = 1;
    uint64_t tokenId;
    const char *perms[permsNum];
    perms[indexZero] = "ohos.permission.ACCESS_SERVICE_DM";
    perms[indexOne] = "ohos.permission.DISTRIBUTED_DATASYNC";
    NativeTokenInfoParams infoInstance = {
        .dcapsNum = 0,
        .permsNum = permsNum,
        .aclsNum = 0,
        .dcaps = NULL,
        .perms = perms,
        .acls = NULL,
        .processName = "dsoftbus_service",
        .aplStr = "system_core",
    };
    tokenId = GetAccessTokenId(&infoInstance);
    SetSelfTokenID(tokenId);
    OHOS::Security::AccessToken::AccessTokenKit::ReloadNativeTokenInfo();
}
constexpr int32_t PAKAGE_NAME_SIZE_MAX = 256;
#define AUTH_CODE_WHITE_LIST_NUM (3)
constexpr const static char g_authCodeWhiteList[AUTH_CODE_WHITE_LIST_NUM][PAKAGE_NAME_SIZE_MAX] = {
    "com.huawei.msdp.hmringgenerator",
    "com.huawei.msdp.hmringdiscriminator",
    "CollaborationFwk",
};

#define PIN_HOLDER_WHITE_LIST_NUM (1)
constexpr const static char g_pinHolderWhiteList[PIN_HOLDER_WHITE_LIST_NUM][PKG_NAME_SIZE_MAX] = {
    "CollaborationFwk",
};

#define SYSTEM_SA_WHITE_LIST_NUM (4)
constexpr const static char systemSaWhiteList[SYSTEM_SA_WHITE_LIST_NUM][PAKAGE_NAME_SIZE_MAX] = {
    "Samgr_Networking",
    "ohos.distributeddata.service",
    "ohos.dslm",
    "ohos.deviceprofile",
};
void PermissionManagerTest::TearDown()
{
}

void PermissionManagerTest::SetUpTestCase()
{
}

void PermissionManagerTest::TearDownTestCase()
{
}

namespace {

/**
 * @tc.name: PinAuthUi::CheckPermission_001
 * @tc.desc: the return value is true
 * @tc.type: FUNC
 * @tc.require: AR000GHSJK
 */
HWTEST_F(PermissionManagerTest, CheckPermission_001, testing::ext::TestSize.Level0)
{
    bool ret = PermissionManager::GetInstance().CheckPermission();
    ASSERT_EQ(ret, true);
}

/**
 * @tc.name: PinAuthUi::GetCallerProcessName_001
 * @tc.desc: the return value is not DM_OK
 * @tc.type: FUNC
 * @tc.require: AR000GHSJK
 */
HWTEST_F(PermissionManagerTest, GetCallerProcessName_001, testing::ext::TestSize.Level0)
{
    std::string processName;
    int32_t ret = PermissionManager::GetInstance().GetCallerProcessName(processName);
    ASSERT_EQ(ret, DM_OK);
}

/**
 * @tc.name: PinAuthUi::CheckProcessNameValidOnAuthCode_001
 * @tc.desc: the return value is false
 * @tc.type: FUNC
 * @tc.require: AR000GHSJK
 */
HWTEST_F(PermissionManagerTest, CheckProcessNameValidOnAuthCode_001, testing::ext::TestSize.Level0)
{
    std::string processName;
    bool ret = PermissionManager::GetInstance().CheckProcessNameValidOnAuthCode(processName);
    ASSERT_EQ(ret, false);
}

/**
 * @tc.name: PinAuthUi::CheckProcessNameValidOnAuthCode_002
 * @tc.desc: the return value is false
 * @tc.type: FUNC
 * @tc.require: AR000GHSJK
 */
HWTEST_F(PermissionManagerTest, CheckProcessNameValidOnAuthCode_002, testing::ext::TestSize.Level0)
{
    std::string processName = "processName";
    bool ret = PermissionManager::GetInstance().CheckProcessNameValidOnAuthCode(processName);
    ASSERT_EQ(ret, false);
}

/**
 * @tc.name: PinAuthUi::CheckProcessNameValidOnAuthCode_003
 * @tc.desc: the return value is true
 * @tc.type: FUNC
 * @tc.require: AR000GHSJK
 */
HWTEST_F(PermissionManagerTest, CheckProcessNameValidOnAuthCode_003, testing::ext::TestSize.Level0)
{
    std::string processName1(g_authCodeWhiteList[0]);
    bool ret = PermissionManager::GetInstance().CheckProcessNameValidOnAuthCode(processName1);
    ASSERT_EQ(ret, true);
    std::string processName2(g_authCodeWhiteList[1]);
    ret = PermissionManager::GetInstance().CheckProcessNameValidOnAuthCode(processName2);
    ASSERT_EQ(ret, true);
    std::string processName3(g_authCodeWhiteList[2]);
    ret = PermissionManager::GetInstance().CheckProcessNameValidOnAuthCode(processName3);
    ASSERT_EQ(ret, true);
}

HWTEST_F(PermissionManagerTest, CheckProcessNameValidOnAuthCode_004, testing::ext::TestSize.Level0)
{
    std::string processName = "CollaborationFwk";
    bool ret = PermissionManager::GetInstance().CheckProcessNameValidOnAuthCode(processName);
    ASSERT_TRUE(ret);
}
/**
 * @tc.name: PinAuthUi::CheckProcessNameValidOnPinHolder_001
 * @tc.desc: the return value is false
 * @tc.type: FUNC
 * @tc.require: AR000GHSJK
 */
HWTEST_F(PermissionManagerTest, CheckProcessNameValidOnPinHolder_001, testing::ext::TestSize.Level0)
{
    std::string processName;
    bool ret = PermissionManager::GetInstance().CheckProcessNameValidOnPinHolder(processName);
    ASSERT_EQ(ret, false);
}

/**
 * @tc.name: PinAuthUi::CheckProcessNameValidOnPinHolder_002
 * @tc.desc: the return value is false
 * @tc.type: FUNC
 * @tc.require: AR000GHSJK
 */
HWTEST_F(PermissionManagerTest, CheckProcessNameValidOnPinHolder_002, testing::ext::TestSize.Level0)
{
    std::string processName = "processName";
    bool ret = PermissionManager::GetInstance().CheckProcessNameValidOnPinHolder(processName);
    ASSERT_EQ(ret, false);
}

/**
 * @tc.name: PinAuthUi::CheckProcessNameValidOnPinHolder_003
 * @tc.desc: the return value is true
 * @tc.type: FUNC
 * @tc.require: AR000GHSJK
 */
HWTEST_F(PermissionManagerTest, CheckProcessNameValidOnPinHolder_003, testing::ext::TestSize.Level0)
{
    std::string processName1(g_pinHolderWhiteList[0]);
    bool ret = PermissionManager::GetInstance().CheckProcessNameValidOnPinHolder(processName1);
    ASSERT_EQ(ret, true);
}

HWTEST_F(PermissionManagerTest, CheckProcessNameValidOnPinHolder_004, testing::ext::TestSize.Level0)
{
    std::string processName = "CollaborationFwk";
    bool ret = PermissionManager::GetInstance().CheckProcessNameValidOnPinHolder(processName);
    ASSERT_TRUE(ret);
}
/**
 * @tc.name: PinAuthUi::CheckSystemSA_001
 * @tc.desc: the return value is false
 * @tc.type：FUNC
 * @tc.require: AR000GHSJK
*/
HWTEST_F(PermissionManagerTest, CheckSystemSA_001, testing::ext::TestSize.Level0)
{
    std::string pkgName;
    bool ret = PermissionManager::GetInstance().CheckSystemSA(pkgName);
    ASSERT_EQ(ret, false);
    pkgName = "pkgName";
    ret = PermissionManager::GetInstance().CheckSystemSA(pkgName);
    ASSERT_EQ(ret, false);
}

/**
 * @tc.name: PinAuthUi::CheckSystemSA_002
 * @tc.desc: the return value is true
 * @tc.type：FUNC
 * @tc.require: AR000GHSJK
*/
HWTEST_F(PermissionManagerTest, CheckSystemSA_002, testing::ext::TestSize.Level0)
{
    std::string pkgName1(systemSaWhiteList[0]);
    bool ret = PermissionManager::GetInstance().CheckSystemSA(pkgName1);
    ASSERT_EQ(ret, true);
    std::string pkgName2(systemSaWhiteList[1]);
    ret = PermissionManager::GetInstance().CheckSystemSA(pkgName2);
    ASSERT_EQ(ret, true);
    std::string pkgName3(systemSaWhiteList[2]);
    ret = PermissionManager::GetInstance().CheckSystemSA(pkgName3);
    ASSERT_EQ(ret, true);
    std::string pkgName4(systemSaWhiteList[3]);
    ret = PermissionManager::GetInstance().CheckSystemSA(pkgName4);
    ASSERT_EQ(ret, true);
}

HWTEST_F(PermissionManagerTest, CheckSystemSA_101, testing::ext::TestSize.Level0)
{
    std::string pkgName = "pkgName";
    bool ret = PermissionManager::GetInstance().CheckSystemSA(pkgName);
    ASSERT_FALSE(ret);
}

HWTEST_F(PermissionManagerTest, CheckSystemSA_102, testing::ext::TestSize.Level0)
{
    std::string pkgName = "ohos.dhardware";
    bool ret = PermissionManager::GetInstance().CheckSystemSA(pkgName);
    ASSERT_TRUE(ret);
}
}
} // namespace DistributedHardware
} // namespace OHOS