/*
 * Copyright (c) 2021 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 "test.h"

#include "native_engine/impl/ark/ark_native_engine.h"
#include "utils/log.h"

#include <unistd.h>

using ArkIdleMonitor = panda::ecmascript::ArkIdleMonitor;
using panda::RuntimeOption;

struct ThreadArgs {
    NativeEngine* engine = nullptr;
    bool initialState = false;
    bool suspendState = false;
    bool resumeState = false;
};

static NativeEngine* g_nativeEngine = nullptr;
LoggerCollector* LoggerCollector::last_ = nullptr;

NativeEngineTest::NativeEngineTest()
{
    engine_ = g_nativeEngine;
}

NativeEngineTest::~NativeEngineTest()
{}

void *NativeEngineTest::Run(void *args)
{
    ThreadArgs* threadArgs = reinterpret_cast<ThreadArgs*>(args);
    NativeEngine* engine = threadArgs->engine;
    threadArgs->initialState = engine->IsSuspended();
    engine->SuspendVM();
    threadArgs->suspendState = engine->IsSuspended();
    engine->ResumeVM();
    sleep(1);
    threadArgs->resumeState = engine->IsSuspended();
    return nullptr;
}

void DeathTest::Run()
{
    size_t stackSize = GetCurrentStackSize();
    ASSERT_NE(pipe(coutPipe_), -1);
    ASSERT_NE(pipe(cerrPipe_), -1);

    // Malloc stack of child process
    ASSERT_FALSE(stackSize < 0 || stackSize > MAX_ALLOWED_CHILD_PROCESS_STACK_SIZE);
    char* childStack = new char[stackSize];
    ASSERT_NE(childStack, nullptr);

    pid_t childPid = clone(RunInChild, childStack + stackSize, SIGCHLD, (void*)this);
    if (childPid != -1) {
        int status = 0;
        ASSERT_EQ(waitpid(childPid, &status, 0), childPid);

        isExit_ = WIFEXITED(status);
        exitCode_ = WEXITSTATUS(status);
        isSignal_ = WIFSIGNALED(status);
        signal_ = WTERMSIG(status);

        coutResult_ = ReadFd(coutPipe_[0]);
        cerrResult_ = ReadFd(cerrPipe_[0]);
        AssertResult();
    }

    close(coutPipe_[0]);
    close(cerrPipe_[0]);

    free(childStack);
    ASSERT_NE(childPid, -1);
}

size_t DeathTest::GetCurrentStackSize()
{
    //  Get stack size of current thread
    pthread_attr_t attr;
    size_t stackSize = 0;
    pthread_getattr_np(pthread_self(), &attr);
    pthread_attr_getstacksize(&attr, &stackSize);
    pthread_attr_destroy(&attr);
    return stackSize;
}

int DeathTest::RunInChild(void* arg)
{
    // block to generate cppcrash log
    if (signal(SIGABRT, SIG_IGN) == SIG_ERR) {
        std::cerr << "Failed to register abort signal handler." << std::endl;
    }

    DeathTest* that = reinterpret_cast<DeathTest*>(arg);
    // close unused write pipe port
    close(that->coutPipe_[0]);
    close(that->cerrPipe_[0]);
    // redirect cout/cerr to pipe
    dup2(that->coutPipe_[1], STDOUT_FILENO);
    dup2(that->cerrPipe_[1], STDERR_FILENO);

    // redirect hilog to stdout
    LOG_SetCallback(RedirectHilog);

    // execute test case.
    that->TestBody();

    // close read pipe port
    close(that->coutPipe_[1]);
    close(that->cerrPipe_[1]);
    return 0;
}

void DeathTest::RedirectHilog(const LogType, const LogLevel level, const unsigned int, const char*, const char* msg)
{
    if (level >= LogLevel::LOG_WARN) {
        std::cerr << msg << std::endl;
    } else {
        std::cout << msg << std::endl;
    }
}

std::string DeathTest::ReadFd(int fd)
{
    std::string result;
    // set pipe to non-block mode
    if (fcntl(fd, F_SETFL, fcntl(fd, F_GETFL) | O_NONBLOCK) == -1) {
        std::cerr << "Failed to set fd to non-block mode" << std::endl;
        return result;
    }
    constexpr int bufferSize = 64;
    char buffer[bufferSize];
    ssize_t readSize = 0;
    while ((readSize = read(fd, buffer, sizeof(buffer))) > 0) {
        result += std::string(buffer, readSize);
    };
    return result;
}

int main(int argc, char** argv)
{
    testing::GTEST_FLAG(output) = "xml:./";
    testing::InitGoogleTest(&argc, argv);

    // Setup
    RuntimeOption option;
    option.SetGcType(RuntimeOption::GC_TYPE::GEN_GC);
    const int64_t poolSize = 0x1000000;  // 16M
    option.SetGcPoolSize(poolSize);
    option.SetLogLevel(RuntimeOption::LOG_LEVEL::ERROR);
    option.SetDebuggerLibraryPath("");
    EcmaVM* vm = panda::JSNApi::CreateJSVM(option);
    if (vm == nullptr) {
        return 0;
    }

    g_nativeEngine = new ArkNativeEngine(vm, nullptr);

    int ret = testing::UnitTest::GetInstance()->Run();

    g_nativeEngine->Loop(LOOP_NOWAIT);

    delete g_nativeEngine;
    g_nativeEngine = nullptr;
    panda::JSNApi::DestroyJSVM(vm);
    vm = nullptr;

    return ret;
}

HWTEST_F(NativeEngineTest, SuspendVM001, testing::ext::TestSize.Level0)
{
    pthread_t tids;
    struct ThreadArgs *args = new ThreadArgs;
    args->engine = engine_;
    int res = pthread_create(&tids, NULL, Run, (void*)args);
    if (res != 0) {
        std::cout << "thread create failed";
        return;
    }
    for (int i = 0; i < 3; ++i) { // 3:Loop 3 times
        std::this_thread::sleep_for(std::chrono::milliseconds(100));
        engine_->CheckSafepoint();
    }
    ASSERT_TRUE(!args->initialState);
    ASSERT_TRUE(args->suspendState);
    ASSERT_TRUE(!args->resumeState);
    delete args;
    args = nullptr;
}

HWTEST_F(NativeEngineTest, CreateRuntimeFunc001, testing::ext::TestSize.Level0)
{
    std::thread t([this]() {
        auto result = engine_->CreateRuntime(true);
        ASSERT_TRUE(result);
        delete reinterpret_cast<NativeEngine *>(result);
    });
    t.join();
}

HWTEST_F(NativeEngineTest, ExecuteTranslateBySourceMapFunc001, testing::ext::TestSize.Level0)
{
    std::string stack = engine_->ExecuteTranslateBySourceMap("test1/test2/test3/test.ts");
    ASSERT_EQ(stack, "test1/test2/test3/test.ts");
}

HWTEST_F(NativeEngineTest, NapiErrorManagerTest001, testing::ext::TestSize.Level0)
{
    auto onWorkerErrorCallback =
        [] (napi_env env, napi_value exception, std::string vm_name, uint32_t type) -> bool {
        EXPECT_EQ(type, 2);
        EXPECT_EQ(vm_name, "test");
        return true;
    };
    auto onMainThreadErrorCallback =
        [] (napi_env env, std::string summary, std::string name,
            std::string message, std::string stack) -> bool {
        EXPECT_EQ(name, "name");
        EXPECT_EQ(message, "message");
        return true;
    };
    auto hasOnErrorCb = [] () -> bool {
        return true;
    };
    NapiErrorManager::GetInstance()->RegisterHasOnErrorCallback(hasOnErrorCb);
    NapiErrorManager::GetInstance()->RegisterOnErrorCallback(onWorkerErrorCallback, onMainThreadErrorCallback);
    ASSERT_NE(NapiErrorManager::GetInstance()->GetOnWorkerErrorCallback(), nullptr);
    ASSERT_NE(NapiErrorManager::GetInstance()->GetOnMainThreadErrorCallback(), nullptr);
    napi_value exception = nullptr;
    NapiErrorManager::GetInstance()->NotifyUncaughtException(reinterpret_cast<napi_env>(engine_),
        exception, "test", 2);
    ASSERT_EQ(NapiErrorManager::GetInstance()->NotifyUncaughtException(reinterpret_cast<napi_env>(engine_),
        "", "name", "message", ""), true);
}

HWTEST_F(NativeEngineTest, NapiErrorManagerTest002, testing::ext::TestSize.Level0)
{
    auto allUnhandledRejectionCallbackTest =
        [] (napi_env env, napi_value* exception, std::string vm_name, uint32_t type) -> bool {
        EXPECT_EQ(type, 0);
        EXPECT_EQ(vm_name, "test");
        return true;
    };
    auto hasAllUnhandledRejectionCb = [] () -> bool {
        return false;
    };

    NapiErrorManager::GetInstance()->RegisterHasAllUnhandledRejectionCallback(hasAllUnhandledRejectionCb);
    napi_value* exception = nullptr;
    NapiErrorManager::GetInstance()->RegisterAllUnhandledRejectionCallback(allUnhandledRejectionCallbackTest);
    ASSERT_NE(NapiErrorManager::GetInstance()->GetAllUnhandledRejectionCallback(), nullptr);
    NapiErrorManager::GetInstance()->NotifyUnhandledRejection(reinterpret_cast<napi_env>(engine_),
        exception, "test", 0);
}

HWTEST_F(NativeEngineTest, NapiErrorManagerTest003, testing::ext::TestSize.Level0)
{
    auto hasOnErrorCb = [] () -> bool {
        return true;
    };

    NapiErrorManager::GetInstance()->RegisterHasOnErrorCallback(hasOnErrorCb);
    auto callback = NapiErrorManager::GetInstance()->GetHasErrorCallback();
    ASSERT_NE(callback, nullptr);
    ASSERT_EQ(callback(), true);
}

HWTEST_F(NativeEngineTest, NapiErrorManagerTest004, testing::ext::TestSize.Level0)
{
    auto hasAllUnhandledRejectionCb = [] () -> bool {
        return false;
    };

    NapiErrorManager::GetInstance()->RegisterHasAllUnhandledRejectionCallback(hasAllUnhandledRejectionCb);
    auto callback = NapiErrorManager::GetInstance()->GetHasAllUnhandledRejectionCallback();
    ASSERT_NE(callback, nullptr);
    ASSERT_EQ(callback(), false);
}

HWTEST_F(NativeEngineTest, FinalizersCallbackTest001, testing::ext::TestSize.Level0)
{
    ASSERT_NE(engine_, nullptr);
    napi_env env = (napi_env)engine_;
    const EcmaVM *vm = reinterpret_cast<ArkNativeEngine*>(engine_)->GetEcmaVm();

    const char *str = "FinalizersCallbackTest001";
    size_t size = 2 * ArkNativeEngine::FINALIZERS_PACK_PENDING_NATIVE_BINDING_SIZE_THRESHOLD;
    static bool finalizersCallbackDone[2] = {false, false};

    for (int i = 0; i < 2; ++i) {
        {
            panda::LocalScope scope(vm);
            napi_value object = nullptr;
            napi_create_object(env, &object);
            napi_wrap_with_size(env, object, (void*)str, [](napi_env env, void *data, void *hint) {
                bool *result = reinterpret_cast<bool*>(hint);
                ASSERT_FALSE(*result);
                *result = true;
            }, reinterpret_cast<void*>(&finalizersCallbackDone[i]), nullptr, size);
        }
        panda::JSNApi::TriggerGC(vm, panda::ecmascript::GCReason::OTHER, panda::JSNApi::TRIGGER_GC_TYPE::FULL_GC);
    }

    ASSERT_FALSE(finalizersCallbackDone[0]);
    ASSERT_TRUE(finalizersCallbackDone[1]);
}

/**
 * @tc.name: SetMainThreadEcmaVMTest
 * @tc.desc: Test interface of ArkIdleMonitor
 * @tc.type: FUNC
 */
HWTEST_F(NativeEngineTest, SetMainThreadEcmaVMTest001, testing::ext::TestSize.Level0)
{
    ASSERT_NE(engine_, nullptr);
    EcmaVM *vm = const_cast<EcmaVM*>(reinterpret_cast<ArkNativeEngine*>(engine_)->GetEcmaVm());

    auto arkIdleMonitor = ArkIdleMonitor::GetInstance();
    arkIdleMonitor->SetMainThreadEcmaVM(vm);

    int64_t oldIdleDuration = arkIdleMonitor->GetTotalIdleDuration();
    for (int i = 0; i < 10; i++) {
        arkIdleMonitor->NotifyLooperIdleStart(i * 500, 20);
        ASSERT_EQ(arkIdleMonitor->IsIdleState(), true);
        arkIdleMonitor->NotifyLooperIdleEnd(i * 500 + 250);
        ASSERT_EQ(arkIdleMonitor->IsIdleState(), false);
    }

    arkIdleMonitor->SetMainThreadEcmaVM(nullptr);

    for (int i = 10; i < 20; i++) {
        arkIdleMonitor->NotifyLooperIdleStart(i * 500, 20);
        ASSERT_EQ(arkIdleMonitor->IsIdleState(), true);
        arkIdleMonitor->NotifyLooperIdleEnd(i * 500 + 250);
        ASSERT_EQ(arkIdleMonitor->IsIdleState(), false);
    }

    int64_t newIdleDuration = arkIdleMonitor->GetTotalIdleDuration();
    ASSERT_TRUE(newIdleDuration > oldIdleDuration);
    ASSERT_TRUE(arkIdleMonitor->GetIdleNotifyCount() > 0);
}

/**
 * @tc.name: NotifyChangeBackgroundStateTest
 * @tc.desc: Test interface of ArkIdleMonitor
 * @tc.type: FUNC
 */
HWTEST_F(NativeEngineTest, NotifyChangeBackgroundStateTest001, testing::ext::TestSize.Level0)
{
    ASSERT_NE(engine_, nullptr);
    EcmaVM *vm = const_cast<EcmaVM*>(reinterpret_cast<ArkNativeEngine*>(engine_)->GetEcmaVm());

    auto arkIdleMonitor = ArkIdleMonitor::GetInstance();
    arkIdleMonitor->SetMainThreadEcmaVM(vm);
    arkIdleMonitor->SetStartTimerCallback();
    arkIdleMonitor->NotifyChangeBackgroundState(true);
    ASSERT_TRUE(arkIdleMonitor->IsInBackground() == true);
    arkIdleMonitor->NotifyChangeBackgroundState(false);
    ASSERT_TRUE(arkIdleMonitor->IsInBackground() == false);
}

/**
 * @tc.name: RegisterWorkerEnvTest
 * @tc.desc: Test interface of ArkIdleMonitor
 * @tc.type: FUNC
 */
HWTEST_F(NativeEngineTest, RegisterWorkerEnvTest001, testing::ext::TestSize.Level0)
{
    ASSERT_NE(engine_, nullptr);
    EcmaVM *vm = const_cast<EcmaVM*>(reinterpret_cast<ArkNativeEngine*>(engine_)->GetEcmaVm());

    auto arkIdleMonitor = ArkIdleMonitor::GetInstance();
    arkIdleMonitor->SetMainThreadEcmaVM(vm);

    std::thread t([&arkIdleMonitor]() {
        RuntimeOption option;
        option.SetGcType(RuntimeOption::GC_TYPE::GEN_GC);
        const int64_t poolSize = 0x1000000;  // 16M
        option.SetGcPoolSize(poolSize);
        option.SetLogLevel(RuntimeOption::LOG_LEVEL::ERROR);
        option.SetDebuggerLibraryPath("");
        auto workerVM1 = panda::JSNApi::CreateJSVM(option);
        auto workerVM2 = panda::JSNApi::CreateJSVM(option);
        ArkNativeEngine *workerEngine1 = new ArkNativeEngine(workerVM1, nullptr);
        ArkNativeEngine *workerEngine2 = new ArkNativeEngine(workerVM2, nullptr);
        napi_env wokerEnv1 = reinterpret_cast<napi_env>(workerEngine1);
        napi_env wokerEnv2 = reinterpret_cast<napi_env>(workerEngine2);

        arkIdleMonitor->UnregisterWorkerEnv(wokerEnv1);
        arkIdleMonitor->UnregisterWorkerEnv(wokerEnv2);

        arkIdleMonitor->RegisterWorkerEnv(wokerEnv1);
        arkIdleMonitor->RegisterWorkerEnv(wokerEnv2);

        arkIdleMonitor->UnregisterWorkerEnv(wokerEnv1);
        arkIdleMonitor->UnregisterWorkerEnv(wokerEnv2);

        delete workerEngine1;
        delete workerEngine2;
    });
    t.join();
}

/**
 * @tc.name: SetRawHeapTrimLevel
 * @tc.desc: Test interface of SetRawHeapTrimLevel
 * @tc.type: FUNC
 */
HWTEST_F(NativeEngineTest, SetRawHeapTrimLevelTest001, testing::ext::TestSize.Level0)
{
    ASSERT_NE(engine_, nullptr);
    EcmaVM *vm = const_cast<EcmaVM*>(reinterpret_cast<ArkNativeEngine*>(engine_)->GetEcmaVm());

    auto arkIdleMonitor = ArkIdleMonitor::GetInstance();
    arkIdleMonitor->SetMainThreadEcmaVM(vm);

    engine_->SetRawHeapTrimLevel(1); // test value
}