xref: /arkcompiler/toolchain/tooling/test/dispatcher_test.cpp

/*
 * 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 "dispatcher.h"
#include "ecmascript/tests/test_helper.h"
#include "protocol_handler.h"

using namespace panda::ecmascript;
using namespace panda::ecmascript::tooling;

namespace panda::test {
class DispatcherTest : public testing::Test {
public:
    static void SetUpTestCase()
    {
        GTEST_LOG_(INFO) << "SetUpTestCase";
    }

    static void TearDownTestCase()
    {
        GTEST_LOG_(INFO) << "TearDownCase";
    }

    void SetUp() override
    {
        TestHelper::CreateEcmaVMWithScope(ecmaVm, thread, scope);
    }

    void TearDown() override
    {
        TestHelper::DestroyEcmaVMWithScope(ecmaVm, scope);
    }

protected:
    EcmaVM *ecmaVm {nullptr};
    EcmaHandleScope *scope {nullptr};
    JSThread *thread {nullptr};
};

HWTEST_F_L0(DispatcherTest, DispatchRequestTest)
{
    std::string msg = "";
    auto dispatchRequest = std::make_unique<DispatchRequest>(msg);
    bool result = dispatchRequest->IsValid();
    ASSERT_TRUE(!result);
    msg = std::string() + R"({"id":0,"method":"Tracing.xxx","params":{}})";
    auto dispatchRequest1 = std::make_unique<DispatchRequest>(msg);
    result  = dispatchRequest1->IsValid();
    ASSERT_TRUE(result);
    msg = std::string() + R"("":0)";
    auto dispatchRequest2 = std::make_unique<DispatchRequest>(msg);
    result = dispatchRequest2->IsValid();
    ASSERT_TRUE(!result);
    msg = std::string() + R"({"method":"Tracing","params":{}})";
    auto dispatchRequest3 = std::make_unique<DispatchRequest>(msg);
    result = dispatchRequest3->IsValid();
    msg = std::string() + R"({"id":0,"params":{}})";
    auto dispatchRequest9 = std::make_unique<DispatchRequest>(msg);
    result = dispatchRequest9->IsValid();
    ASSERT_TRUE(!result);
    msg = std::string() + R"({"id":0,"method":"Tracing","params":{}})";
    auto dispatchRequest4 = std::make_unique<DispatchRequest>(msg);
    result = dispatchRequest4->IsValid();
    ASSERT_TRUE(!result);
    msg = std::string() + R"({"id":0,"method":".Tracing","params":{}})";
    auto dispatchRequest5 = std::make_unique<DispatchRequest>(msg);
    result = dispatchRequest5->IsValid();
    ASSERT_TRUE(!result);
    msg = std::string() + R"({"id":0,"method":"Tracing.","params":{}})";
    auto dispatchRequest6 = std::make_unique<DispatchRequest>(msg);
    result = dispatchRequest6->IsValid();
    ASSERT_TRUE(!result);
    msg = std::string() + R"({"id":0,"method":"Tracing.end", "params":1})";
    auto dispatchRequest7 = std::make_unique<DispatchRequest>(msg);
    result = dispatchRequest7->IsValid();
    ASSERT_TRUE(!result);
    msg = std::string() + R"({"id":0,"method":"Tracing.end"})";
    auto dispatchRequest8 = std::make_unique<DispatchRequest>(msg);
    result = dispatchRequest8->IsValid();
    ASSERT_TRUE(result);
}

HWTEST_F_L0(DispatcherTest, IsValidTest)
{
    std::string msg = std::string() + R"({"id":0,"method":"Tracing.end","params":{}})";
    auto dispatchRequest = std::make_unique<DispatchRequest>(msg);
    bool result = dispatchRequest->IsValid();
    ASSERT_TRUE(result);
}

HWTEST_F_L0(DispatcherTest, GetDomainTest)
{
    std::string msg = std::string() + R"({"id":0,"method":"Tracing.end","params":{}})";
    auto dispatchRequest = std::make_unique<DispatchRequest>(msg);
    std::string result = dispatchRequest->GetDomain();
    ASSERT_TRUE(result == "Tracing");
}

HWTEST_F_L0(DispatcherTest, DispatchResponseCreateTest)
{
    ResponseCode code = ResponseCode::OK;
    std::string msg = "";
    DispatchResponse response = DispatchResponse::Create(code, msg);
    ASSERT_TRUE(response.IsOk());
    ASSERT_TRUE(response.GetMessage() == "");

    msg = "msg";
    response = DispatchResponse::Create(msg);
    ASSERT_TRUE(!response.IsOk());
    ASSERT_TRUE(response.GetMessage() == "msg");

    std::optional<std::string> error;
    response = DispatchResponse::Create(error);
    ASSERT_TRUE(response.IsOk());
    ASSERT_TRUE(response.GetMessage() == "");
}

HWTEST_F_L0(DispatcherTest, DispatcherDispatchTest)
{
    std::string msg = "";
    DispatchRequest dispatchRequest(msg);
    msg = std::string() + R"({"id":0,"method":"Tracing.end","params":{}})";
    DispatchRequest dispatchRequest1(msg);
    msg = std::string() + R"({"id":0,"method":"Test.end","params":{}})";
    DispatchRequest dispatchRequest2(msg);
    std::string result = "";
    std::function<void(const void*, const std::string &)> callback =
        [&result]([[maybe_unused]] const void *ptr, const std::string &temp) { result = temp; };
    ProtocolChannel *channel =  new ProtocolHandler(callback, ecmaVm);
    auto dispatcher = std::make_unique<Dispatcher>(ecmaVm, channel);
    dispatcher->Dispatch(dispatchRequest);
    ASSERT_TRUE(result == "");
    result = "";
    dispatcher->Dispatch(dispatchRequest1);
    ASSERT_TRUE(result.find("\"id\":0,") != std::string::npos);
    result = "";
    dispatcher->Dispatch(dispatchRequest2);
    ASSERT_TRUE(result == "");
}
}  // namespace panda::test