/*
 * Copyright (c) 2022 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 <gtest/gtest.h>
#include <thread>
#include <sched/workgroup_internal.h>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#if defined(QOS_FRAME_RTG)
#include "rtg_interface.h"
#include "concurrent_task_client.h"
#endif
#include "dfx/log/ffrt_log_api.h"
#include "common.h"

#define GET_TID() syscall(SYS_gettid)

using namespace testing;
#ifdef HWTEST_TESTING_EXT_ENABLE
using namespace testing::ext;
#endif
using namespace ffrt;

class WorkgroupInternalTest : public testing::Test {
protected:
    static void SetUpTestCase()
    {
    }

    static void TearDownTestCase()
    {
    }

    virtual void SetUp()
    {
    }

    virtual void TearDown()
    {
    }
};

void StartApp(int uid)
{
#if defined(QOS_FRAME_RTG)
    std::unordered_map<std::string, std::string> payload;
    payload["uid"] = std::to_string(uid);
    payload["type"] = "appStart";
    OHOS::ConcurrentTask::ConcurrentTaskClient::GetInstance().ReportData(0,
        uid, payload);
#endif
}

void SwapToFront(int uid)
{
#if defined(QOS_FRAME_RTG)
    std::unordered_map<std::string, std::string> payload;
    payload["uid"] = std::to_string(uid);
    payload["type"] = "foreground";
    OHOS::ConcurrentTask::ConcurrentTaskClient::GetInstance().ReportData(0,
        uid, payload);
#endif
}

/**
 * @tc.name: JoinWGTest
 * @tc.desc: Test whether the JoinWG interface are normal.
 * @tc.type: FUNC
 */
HWTEST_F(WorkgroupInternalTest, JoinWGTest, TestSize.Level1)
{
    int tid = GET_TID();
    bool ret = JoinWG(tid);
#if defined(QOS_FRAME_RTG)
    EXPECT_FALSE(ret);
#else
    EXPECT_TRUE(ret);
#endif
}

/**
 * @tc.name: WorkgroupCreateTest
 * @tc.desc: Test whether the WorkgroupCreate interface are normal.
 * @tc.type: FUNC
 */
HWTEST_F(WorkgroupInternalTest, WorkgroupCreateTest, TestSize.Level1)
{
    long interval = 10000;
    (void)setuid(1000);
    struct WorkGroup *ret = WorkgroupCreate(interval);
    EXPECT_NE(ret, nullptr);

    (void)setuid(3039);
    struct WorkGroup *ret1 = WorkgroupCreate(interval);
    EXPECT_NE(ret1, nullptr);

    (void)setuid(0);
    struct WorkGroup *ret2 = WorkgroupCreate(interval);
    EXPECT_NE(ret2, nullptr);
}

/**
 * @tc.name: WorkgroupStartIntervalTest
 * @tc.desc: Test whether the WorkgroupStartInterval interface are normal.
 * @tc.type: FUNC
 */
HWTEST_F(WorkgroupInternalTest, WorkgroupStartIntervalTest, TestSize.Level1)
{
#if defined(QOS_FRAME_RTG)
    struct WorkGroup* wg = nullptr;
    struct WorkGroup wg1 = {true, 0, {0}, 0, WgType(0)};
    WorkgroupStartInterval(wg);

    struct WorkGroup* p = &wg1;
    p->started = true;
    WorkgroupStartInterval(p);

    int SYS_UID = 1000;
    int TEST_UID = 10087;

    (void)setuid(SYS_UID);
    StartApp(TEST_UID);
    SwapToFront(TEST_UID);
    p->started = false;
    p->rtgId = 10;
    WorkgroupStartInterval(p);
    EXPECT_TRUE(p->started);
#endif
}

/**
 * @tc.name: WorkgroupStopIntervalTest
 * @tc.desc: Test whether the WorkgroupStopInterval interface are normal.
 * @tc.type: FUNC
 */
HWTEST_F(WorkgroupInternalTest, WorkgroupStopIntervalTest, TestSize.Level1)
{
#if defined(QOS_FRAME_RTG)
    struct WorkGroup* wg = nullptr;
    struct WorkGroup wg1 = {true, 0, {0}, 0, WgType(0)};
    WorkgroupStopInterval(wg);

    struct WorkGroup* p = &wg1;
    p->started = false;
    WorkgroupStopInterval(p);

    int SYS_UID = 1000;
    int TEST_UID = 10087;

    (void)setuid(SYS_UID);
    StartApp(TEST_UID);
    SwapToFront(TEST_UID);
    p->started = true;
    p->rtgId = 10;
    WorkgroupStopInterval(p);
    EXPECT_FALSE(p->started);
#endif
}

/**
 * @tc.name: WorkgroupJoinTest
 * @tc.desc: Test whether the WorkgroupJoin interface are normal.
 * @tc.type: FUNC
 */
HWTEST_F(WorkgroupInternalTest, WorkgroupJoinTest, TestSize.Level1)
{
    long tid = 10086;
    struct WorkGroup wg1 = {true, 0, {0}, 0, WgType(0)};
    struct WorkGroup* wg = nullptr;

    WorkgroupJoin(wg, tid);
    struct WorkGroup* p = &wg1;
    p->rtgId = 10;
    WorkgroupJoin(p, tid);
}

/**
 * @tc.name: WorkgroupClearTest
 * @tc.desc: Test whether the WorkgroupClear interface are normal.
 * @tc.type: FUNC
 */
HWTEST_F(WorkgroupInternalTest, WorkgroupClearTest, TestSize.Level1)
{
    struct WorkGroup* wg = nullptr;

    int ret = WorkgroupClear(wg);
    EXPECT_EQ(0, ret);

    long interval = 10000;
    (void)setuid(1000);
    struct WorkGroup *wg2 = WorkgroupCreate(interval);
    int ret2 = WorkgroupClear(wg2);
#if defined(QOS_FRAME_RTG)
    EXPECT_EQ(-1, ret2);
#else
    EXPECT_EQ(0, ret2);
#endif

    (void)setuid(3039);
    struct WorkGroup *wg3 = WorkgroupCreate(interval);
    int ret3 = WorkgroupClear(wg3);
#if defined(QOS_FRAME_RTG)
    EXPECT_EQ(-1, ret3);
#else
    EXPECT_EQ(0, ret3);
#endif

    (void)setuid(0);
    struct WorkGroup *wg4 = WorkgroupCreate(interval);
    int ret4 = WorkgroupClear(wg4);
    EXPECT_EQ(0, ret4);
}