1/*
2 * Copyright (c) 2023-2024 Huawei Device Co., Ltd.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at
6 *
7 *     http://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15
16#ifndef TEST_CONVERTXML_UNITTEST_TEST_H
17#define TEST_CONVERTXML_UNITTEST_TEST_H
18
19#include "native_engine.h"
20
21#include "gtest/gtest.h"
22
23#if defined(ENABLE_TASKPOOL_FFRT)
24#include "c/executor_task.h"
25#include "ffrt_inner.h"
26#endif
27
28namespace Commonlibrary::Concurrent::TaskPoolModule {
29class NativeEngineTest : public testing::Test {
30public:
31    NativeEngineTest();
32    virtual ~NativeEngineTest();
33    void SetUp() override {}
34    void TearDown() override {}
35
36    static napi_value IsConcurrent(napi_env env, napi_value argv[], size_t argc);
37    static napi_value GetTaskPoolInfo(napi_env env, napi_value argv[], size_t argc);
38    static napi_value TerminateTask(napi_env env, napi_value argv[], size_t argc);
39    static napi_value Execute(napi_env env, napi_value argv[], size_t argc);
40    static napi_value ExecuteDelayed(napi_env env, napi_value argv[], size_t argc);
41    static napi_value Cancel(napi_env env, napi_value argv[], size_t argc);
42    static void DelayTask(uv_timer_t* handle);
43    static napi_value ExecuteGroup(napi_env env, napi_value taskGroup);
44    static napi_value ExecutePeriodically(napi_env env, napi_value argv[], size_t argc);
45    static void PeriodicTaskCallback(uv_timer_t* handle);
46    static void UpdateGroupInfoByResult(napi_env env, uv_timer_t* handle, napi_value res, bool success);
47
48    static void TaskGroupDestructor(napi_env env, void* data);
49    static void SequenceRunnerDestructor(napi_env env, void* data);
50    static void TryTriggerExpand();
51    static void CheckForBlockedWorkers(napi_env env);
52    static void TriggerShrink(napi_env env);
53    static void foo(const uv_async_t* req);
54    static void NotifyShrink(napi_env env);
55    static void TryExpand(napi_env env);
56    static void CancelTask(napi_env env);
57    static void NotifyWorkerIdle(napi_env env);
58    static void EnqueueTaskId(napi_env env);
59    static void GetTaskByPriority(napi_env env);
60    static void RestoreWorker(napi_env env);
61    static void StoreDependentId(uint64_t taskId, uint64_t dependentId);
62    static void StoreDependentTaskId(uint64_t taskId, uint64_t dependentId);
63    static void StoreTaskDuration(uint64_t taskId);
64    static void InitTaskManager(napi_env env);
65    static void NotifyDependencyTaskInfo(napi_env env);
66    static void StoreTaskDependency(napi_env env);
67    static void RemoveTaskDependency(napi_env env);
68    static void ReleaseTaskData(napi_env env);
69    static void CheckTask(napi_env env);
70    static void CancelGroupTask(napi_env env);
71    static void TriggerSeqRunner(napi_env env);
72    static void UpdateGroupState(napi_env env);
73    static void ReleaseWorkerHandles(napi_env env);
74    static void DebuggerOnPostTask(napi_env env);
75    static void PerformTask(napi_env env);
76    static void NotifyHandleTaskResult(napi_env env);
77    static void TaskResultCallback(napi_env env);
78    static void HandleFunctionException(napi_env env);
79    static void* WorkerConstructor(napi_env env);
80    static pid_t GetWorkerTid(uv_timer_t* handle);
81
82    class ExceptionScope {
83    public:
84        explicit ExceptionScope(napi_env env) : env_(env) {}
85        ~ExceptionScope()
86        {
87            napi_value exception = nullptr;
88            napi_get_and_clear_last_exception(env_, &exception);
89        }
90    private:
91        napi_env env_ = nullptr;
92    };
93
94protected:
95    NativeEngine *engine_ = nullptr;
96};
97
98class SendableUtils {
99public:
100    SendableUtils() = default;
101    ~SendableUtils() = default;
102
103    static napi_value CreateSendableClass(napi_env env);
104    static napi_value CreateSendableInstance(napi_env env);
105    static napi_value Foo(napi_env env, napi_callback_info info);
106    static napi_value Bar(napi_env env, napi_callback_info info);
107};
108} // namespace Commonlibrary::Concurrent::TaskPoolModule
109#endif // TEST_CONVERTXML_UNITTEST_TEST_H