/*
 * Copyright (c) 2023 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 <gmock/gmock.h>
#include <gtest/gtest.h>

#include "dfs_error.h"
#include "i_daemon_mock.h"
#include "i_file_trans_listener.h"
#include "ipc/daemon.h"
#include "ipc_skeleton.h"
#include "iremote_object.h"
#include "remote_file_share.h"
#include "system_ability_definition.h"
#include "utils_log.h"

namespace {
    bool g_getCallingUidTrue = true;
    int32_t g_uidMode = 0;
    int32_t g_getDfsUrisDirFromLocal = OHOS::FileManagement::E_OK;
    constexpr pid_t PASTE_BOARD_UID = 3816;
    constexpr pid_t UDMF_UID = 3012;
    constexpr pid_t DAEMON_UID = 1009;
    static pid_t UID = DAEMON_UID;
}

namespace OHOS {
#ifdef CONFIG_IPC_SINGLE
using namespace IPC_SINGLE;
#endif
pid_t IPCSkeleton::GetCallingUid()
{
    if (!g_getCallingUidTrue) {
        return -1;
    }
    switch (g_uidMode) {
        case 0:
            UID = PASTE_BOARD_UID;
            break;
        case 1:
            UID = UDMF_UID;
            break;
        default:
            UID = -1;
    }
    return UID;
}
}

namespace OHOS::AppFileService::ModuleRemoteFileShare {
    int32_t RemoteFileShare::GetDfsUrisDirFromLocal(const std::vector<std::string> &uriList, const int32_t &userId,
        std::unordered_map<std::string, AppFileService::ModuleRemoteFileShare::HmdfsUriInfo> &uriToDfsUriMaps)
    {
        return g_getDfsUrisDirFromLocal;
    }
}

namespace OHOS::Storage::DistributedFile::Test {
using namespace testing;
using namespace testing::ext;
using namespace std;

class DaemonTest : public testing::Test {
public:
    static void SetUpTestCase(void);
    static void TearDownTestCase(void);
    void SetUp();
    void TearDown();
    std::shared_ptr<Daemon> daemon_;
};

class MockFileTransListener : public IRemoteStub<IFileTransListener> {
public:
    MOCK_METHOD2(OnFileReceive, int32_t(uint64_t totalBytes, uint64_t processedBytes));
    MOCK_METHOD2(OnFailed, int32_t(const std::string &sessionName, int32_t errorCode));
    MOCK_METHOD1(OnFinished, int32_t(const std::string &sessionName));
};

DistributedHardware::DmDeviceInfo deviceInfo = {
    .deviceId = "testdevid",
    .networkId = "testnetworkid",
};

void DaemonTest::SetUpTestCase(void)
{
    GTEST_LOG_(INFO) << "SetUpTestCase";
}

void DaemonTest::TearDownTestCase(void)
{
    GTEST_LOG_(INFO) << "TearDownTestCase";
}

void DaemonTest::SetUp(void)
{
    int32_t saID = FILEMANAGEMENT_DISTRIBUTED_FILE_DAEMON_SA_ID;
    bool runOnCreate = true;
    daemon_ = std::make_shared<Daemon>(saID, runOnCreate);
    GTEST_LOG_(INFO) << "SetUp";
}

void DaemonTest::TearDown(void)
{
    GTEST_LOG_(INFO) << "TearDown";
}

/**
 * @tc.name: OnStopTest
 * @tc.desc: Verify the OnStop function
 * @tc.type: FUNC
 * @tc.require: issueI7M6L1
 */
HWTEST_F(DaemonTest, OnStopTest, TestSize.Level0)
{
    GTEST_LOG_(INFO) << "OnStop Start";
    try {
        daemon_->state_ = ServiceRunningState::STATE_NOT_START;
        daemon_->registerToService_ = false;
        daemon_->OnStop();
        EXPECT_TRUE(true);
    } catch (...) {
        EXPECT_TRUE(false);
        GTEST_LOG_(INFO) << "OnStop  ERROR";
    }
    GTEST_LOG_(INFO) << "OnStop End";
}

/**
 * @tc.name: DaemonTest_PublishSA_0100
 * @tc.desc: Verify the PublishSA function.
 * @tc.type: FUNC
 * @tc.require: issueI7SP3A
 */
HWTEST_F(DaemonTest, DaemonTest_PublishSA_0100, TestSize.Level0)
{
    GTEST_LOG_(INFO) << "DaemonTest_PublishSA_0100 start";
    try {
        daemon_->registerToService_ = true;
        daemon_->PublishSA();
        EXPECT_TRUE(daemon_->registerToService_);
    } catch (const exception &e) {
        LOGE("Error:%{public}s", e.what());
        EXPECT_TRUE(false);
    }
    GTEST_LOG_(INFO) << "DaemonTest_PublishSA_0100 end";
}

/**
 * @tc.name: DaemonTest_OnStart_0100
 * @tc.desc: Verify the OnStart function.
 * @tc.type: FUNC
 * @tc.require: issueI7SP3A
 */
HWTEST_F(DaemonTest, DaemonTest_OnStart_0100, TestSize.Level0)
{
    GTEST_LOG_(INFO) << "DaemonTest_OnStart_0100 start";
    try {
        daemon_->state_ = ServiceRunningState::STATE_RUNNING;
        daemon_->OnStart();
        EXPECT_EQ(daemon_->state_, ServiceRunningState::STATE_RUNNING);
    } catch (const exception &e) {
        LOGE("Error:%{public}s", e.what());
        EXPECT_TRUE(false);
    }
    GTEST_LOG_(INFO) << "DaemonTest_OnStart_0100 end";
}

/**
 * @tc.name: DaemonTest_OnRemoveSystemAbility_0100
 * @tc.desc: Verify the OnRemoveSystemAbility function.
 * @tc.type: FUNC
 * @tc.require: issueI7SP3A
 */
HWTEST_F(DaemonTest, DaemonTest_OnRemoveSystemAbility_0100, TestSize.Level0)
{
    GTEST_LOG_(INFO) << "DaemonTest_OnRemoveSystemAbility_0100 start";
    try {
        daemon_->OnRemoveSystemAbility(COMMON_EVENT_SERVICE_ID + 1, "");
        EXPECT_TRUE(true);
    } catch (const exception &e) {
        LOGE("Error:%{public}s", e.what());
        EXPECT_TRUE(false);
    }
    GTEST_LOG_(INFO) << "DaemonTest_OnRemoveSystemAbility_0100 end";
}

/**
 * @tc.name: DaemonTest_OnRemoveSystemAbility_0200
 * @tc.desc: Verify the OnRemoveSystemAbility function.
 * @tc.type: FUNC
 * @tc.require: issueI7SP3A
 */
HWTEST_F(DaemonTest, DaemonTest_OnRemoveSystemAbility_0200, TestSize.Level0)
{
    GTEST_LOG_(INFO) << "DaemonTest_OnRemoveSystemAbility_0200 start";
    try {
        daemon_->subScriber_ = nullptr;
        daemon_->OnRemoveSystemAbility(COMMON_EVENT_SERVICE_ID, "");
        EXPECT_TRUE(true);
    } catch (const exception &e) {
        LOGE("Error:%{public}s", e.what());
        EXPECT_TRUE(false);
    }
    GTEST_LOG_(INFO) << "DaemonTest_OnRemoveSystemAbility_0200 end";
}

/**
 * @tc.name: DaemonTest_PrepareSession_0100
 * @tc.desc: Verify the PrepareSession function.
 * @tc.type: FUNC
 * @tc.require: issueI90MOB
 */
HWTEST_F(DaemonTest, DaemonTest_PrepareSession_0100, TestSize.Level0)
{
    GTEST_LOG_(INFO) << "DaemonTest_PrepareSession_0100 start";
    try {
        const std::string srcUri = "file://docs/storage/Users/currentUser/Documents?networkid=xxxxx";
        const std::string dstUri = "file://docs/storage/Users/currentUser/Documents";
        const std::string srcDeviceId = "testSrcDeviceId";
        auto listener = sptr<IRemoteObject>(new MockFileTransListener());
        const std::string copyPath = "tmpDir";
        const std::string sessionName = "DistributedDevice0";
        HmdfsInfo fileInfo = {
            .copyPath = copyPath,
            .dirExistFlag = false,
            .sessionName = sessionName,
        };
        EXPECT_EQ(daemon_->PrepareSession(srcUri, dstUri, srcDeviceId, listener, fileInfo),
                  FileManagement::E_SA_LOAD_FAILED);
    } catch (const exception &e) {
        LOGE("Error:%{public}s", e.what());
        EXPECT_TRUE(false);
    }
    GTEST_LOG_(INFO) << "DaemonTest_PrepareSession_0100 end";
}

/**
 * @tc.name: DaemonTest_GetDfsUrisDirFromLocal_0100
 * @tc.desc: Verify the GetDfsUrisDirFromLocal function.
 * @tc.type: FUNC
 * @tc.require: issueI90MOB
 */
HWTEST_F(DaemonTest, DaemonTest_GetDfsUrisDirFromLocal_0100, TestSize.Level0)
{
    GTEST_LOG_(INFO) << "DaemonTest_GetDfsUrisDirFromLocal_0100 start";
    try {
        std::vector<std::string> uriList;
        int32_t userId = 100;
        std::string normalUri = "file://docs/data/storage/el2/cloud";
        std::string errUri = "file://docs/data/storage/../el2/cloud";
        std::unordered_map<std::string, AppFileService::ModuleRemoteFileShare::HmdfsUriInfo> uriToDfsUriMaps;
        g_getCallingUidTrue = false;
        auto ret = daemon_->GetDfsUrisDirFromLocal(uriList, userId, uriToDfsUriMaps);
        EXPECT_EQ(ret, FileManagement::E_PERMISSION_DENIED);

        g_getCallingUidTrue = true;
        uriList.emplace_back(normalUri);
        uriList.emplace_back(errUri);
        ret = daemon_->GetDfsUrisDirFromLocal(uriList, userId, uriToDfsUriMaps);
        EXPECT_EQ(ret, FileManagement::E_ILLEGAL_URI);

        g_getCallingUidTrue = true;
        uriList.erase(uriList.begin() + 1); // 1: errUri
        g_getDfsUrisDirFromLocal = FileManagement::E_INVAL_ARG;
        ret = daemon_->GetDfsUrisDirFromLocal(uriList, userId, uriToDfsUriMaps);
        EXPECT_EQ(ret, FileManagement::E_INVAL_ARG);

        g_getDfsUrisDirFromLocal = OHOS::FileManagement::E_OK;
        ret = daemon_->GetDfsUrisDirFromLocal(uriList, userId, uriToDfsUriMaps);
        EXPECT_EQ(ret, OHOS::FileManagement::E_OK);

        g_uidMode = 1; // 1: UDMF_UID
        uriList.emplace_back(errUri);
        ret = daemon_->GetDfsUrisDirFromLocal(uriList, userId, uriToDfsUriMaps);
        EXPECT_EQ(ret, FileManagement::E_ILLEGAL_URI);

        uriList.erase(uriList.begin() + 1); // 1: errUri
        g_getDfsUrisDirFromLocal = FileManagement::E_INVAL_ARG;
        ret = daemon_->GetDfsUrisDirFromLocal(uriList, userId, uriToDfsUriMaps);
        EXPECT_EQ(ret, FileManagement::E_INVAL_ARG);

        g_getDfsUrisDirFromLocal = OHOS::FileManagement::E_OK;
        ret = daemon_->GetDfsUrisDirFromLocal(uriList, userId, uriToDfsUriMaps);
        EXPECT_EQ(ret, OHOS::FileManagement::E_OK);
    } catch (const exception &e) {
        LOGE("Error:%{public}s", e.what());
        EXPECT_TRUE(false);
    }
    GTEST_LOG_(INFO) << "DaemonTest_GetDfsUrisDirFromLocal_0100 end";
}
} // namespace OHOS::Storage::DistributedFile::Test