1 /*
2 * Copyright (c) 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 #include "gtest/gtest.h"
17 #include "inspector/connect_inspector.h"
18 #include "inspector/connect_server.h"
19 #include "websocket/client/websocket_client.h"
20 #include "websocket/server/websocket_server.h"
21
22 using namespace OHOS::ArkCompiler::Toolchain;
23
24 namespace panda::test {
25 class ConnectServerTest : public testing::Test {
26 public:
SetUpTestCase()27 static void SetUpTestCase()
28 {
29 GTEST_LOG_(INFO) << "ConnectServerTest::SetUpTestCase";
30 }
31
TearDownTestCase()32 static void TearDownTestCase()
33 {
34 GTEST_LOG_(INFO) << "ConnectServerTest::TearDownTestCase";
35 }
36
37 void SetUp() override {}
38
39 void TearDown() override {}
40 void ReturnValueVerification();
41 #if defined(OHOS_PLATFORM)
42 static constexpr char CONNECTED_MESSAGE_TEST[] = "connected";
43 static constexpr char OPEN_MESSAGE_TEST[] = "layoutOpen";
44 static constexpr char CLOSE_MESSAGE_TEST[] = "layoutClose";
45 static constexpr char REQUEST_MESSAGE_TEST[] = "tree";
46 static constexpr char STOPDEBUGGER_MESSAGE_TEST[] = "stopDebugger";
47 static constexpr char OPEN_ARKUI_STATE_PROFILER_TEST[] = "ArkUIStateProfilerOpen";
48 static constexpr char CLOSE_ARKUI_STATE_PROFILER_TEST[] = "ArkUIStateProfilerClose";
49 static constexpr char START_RECORD_MESSAGE_TEST[] = "rsNodeStartRecord";
50 static constexpr char STOP_RECORD_MESSAGE_TEST[] = "rsNodeStopRecord";
51
52 static constexpr char HELLO_INSPECTOR_CLIENT[] = "hello inspector client";
53 static constexpr char INSPECTOR_SERVER_OK[] = "inspector server ok";
54 static constexpr char INSPECTOR_RUN[] = "inspector run";
55 static constexpr char INSPECTOR_QUIT[] = "inspector quit";
56 static constexpr int32_t WAIT_TIME = 2;
57 #endif
58 };
59
60 bool g_profilerFlag = false;
61 bool g_connectFlag = false;
62 bool g_switchStatus = false;
63 int32_t g_createInfoId = 0;
64 int32_t g_instanceId = 1;
65
CallbackInit()66 void CallbackInit()
67 {
68 auto profilerCb = [](bool flag) -> void {
69 g_profilerFlag = flag;
70 };
71 SetProfilerCallback(profilerCb);
72
73 auto connectCb = [](bool flag) -> void {
74 g_connectFlag = flag;
75 };
76 SetConnectCallback(connectCb);
77
78 auto debugModeCb = []() -> void {
79 GTEST_LOG_(INFO) << "Execute DebugModeCallBack.";
80 };
81 SetDebugModeCallBack(debugModeCb);
82
83 auto switchStatusCb = [](bool flag) -> void {
84 g_switchStatus = flag;
85 };
86 auto createInfoCb = [](int32_t id) -> void {
87 g_createInfoId = id;
88 };
89 SetSwitchCallBack(switchStatusCb, createInfoCb, g_instanceId);
90
91 auto startRecordFunc = []() -> void {};
92 auto stopRecordFunc = []() -> void {};
93 SetRecordCallback(startRecordFunc, stopRecordFunc);
94 GTEST_LOG_(INFO) << "ConnectServerTest::CallbackInit finished";
95 }
96
ReturnValueVerification()97 void ConnectServerTest::ReturnValueVerification()
98 {
99 #if defined(OHOS_PLATFORM)
100 // Waiting for executing the message instruction sent by the client
101 sleep(WAIT_TIME);
102 ASSERT_TRUE(g_profilerFlag);
103 ASSERT_EQ(g_createInfoId, g_instanceId);
104 ASSERT_TRUE(g_switchStatus);
105 ASSERT_TRUE(g_connectFlag);
106 ASSERT_FALSE(WaitForConnection());
107
108 SendMessage(INSPECTOR_SERVER_OK);
109 SendLayoutMessage(INSPECTOR_RUN);
110
111 // Waiting for executing the message instruction sent by the client
112 sleep(WAIT_TIME);
113 ASSERT_FALSE(g_profilerFlag);
114 ASSERT_FALSE(g_switchStatus);
115 ASSERT_FALSE(g_connectFlag);
116
117 SendProfilerMessage(INSPECTOR_QUIT);
118
119 StopServer("InspectorConnectTest");
120 #endif
121 }
122
HWTEST_F(ConnectServerTest, InspectorBasicTest, testing::ext::TestSize.Level0)123 HWTEST_F(ConnectServerTest, InspectorBasicTest, testing::ext::TestSize.Level0)
124 {
125 ASSERT_TRUE(WaitForConnection());
126 }
127
HWTEST_F(ConnectServerTest, InspectorConnectTest, testing::ext::TestSize.Level0)128 HWTEST_F(ConnectServerTest, InspectorConnectTest, testing::ext::TestSize.Level0)
129 {
130 CallbackInit();
131 #if defined(OHOS_PLATFORM)
132 int appPid = getprocpid();
133 int oldProcessfd = -2;
134 ASSERT_TRUE(StartServerForSocketPair(oldProcessfd));
135 // test ConnectServer is not nullptr
136 ASSERT_TRUE(StartServerForSocketPair(oldProcessfd));
137 StoreMessage(g_instanceId, HELLO_INSPECTOR_CLIENT);
138 pid_t pid = fork();
139 if (pid == 0) {
140 WebSocketClient clientSocket;
141 ASSERT_TRUE(clientSocket.InitToolchainWebSocketForSockName(std::to_string(appPid)));
142 ASSERT_TRUE(clientSocket.ClientSendWSUpgradeReq());
143 ASSERT_TRUE(clientSocket.ClientRecvWSUpgradeRsp());
144 EXPECT_TRUE(clientSocket.SendReply(OPEN_ARKUI_STATE_PROFILER_TEST));
145 EXPECT_TRUE(clientSocket.SendReply(REQUEST_MESSAGE_TEST));
146 EXPECT_TRUE(clientSocket.SendReply(OPEN_MESSAGE_TEST));
147 EXPECT_TRUE(clientSocket.SendReply(CONNECTED_MESSAGE_TEST));
148 EXPECT_TRUE(clientSocket.SendReply(START_RECORD_MESSAGE_TEST));
149 EXPECT_TRUE(clientSocket.SendReply(STOP_RECORD_MESSAGE_TEST));
150
151 EXPECT_STREQ(clientSocket.Decode().c_str(), HELLO_INSPECTOR_CLIENT);
152
153 std::string recv = clientSocket.Decode();
154 EXPECT_STREQ(recv.c_str(), INSPECTOR_SERVER_OK);
155 if (strcmp(recv.c_str(), INSPECTOR_SERVER_OK) == 0) {
156 EXPECT_TRUE(clientSocket.SendReply(CLOSE_MESSAGE_TEST));
157 EXPECT_TRUE(clientSocket.SendReply(STOPDEBUGGER_MESSAGE_TEST));
158 EXPECT_TRUE(clientSocket.SendReply(CLOSE_ARKUI_STATE_PROFILER_TEST));
159 }
160
161 EXPECT_STREQ(clientSocket.Decode().c_str(), INSPECTOR_RUN);
162 EXPECT_STREQ(clientSocket.Decode().c_str(), INSPECTOR_QUIT);
163 clientSocket.Close();
164 exit(0);
165 } else if (pid > 0) {
166 ReturnValueVerification();
167 }
168 else {
169 std::cerr << "InspectorConnectTest::fork failed" << std::endl;
170 }
171 RemoveMessage(g_instanceId);
172 #endif
173 }
174
175 } // namespace panda::test