1 /*
2 * Copyright (c) 2021 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 "event_dispatch_queue_test.h"
17
18 using namespace testing::ext;
19 using namespace OHOS::HiviewDFX;
20
SetUp()21 void EventDispatchQueueTest::SetUp()
22 {
23 /**
24 * @tc.setup: create order and unordered event dispatch queue
25 */
26 printf("SetUp.\n");
27 }
28
TearDown()29 void EventDispatchQueueTest::TearDown()
30 {
31 /**
32 * @tc.teardown: destroy the event dispatch queue we have created
33 */
34 printf("TearDown.\n");
35 }
36
CreateEvent(const std::string & name,int32_t id,const std::string & message,Event::MessageType type)37 std::shared_ptr<Event> EventDispatchQueueTest::CreateEvent(const std::string& name, int32_t id,
38 const std::string& message, Event::MessageType type)
39 {
40 auto event = std::make_shared<Event>(name);
41 event->messageType_ = type;
42 event->eventId_ = id;
43 event->SetValue("message", message);
44 return event;
45 }
46
OnUnorderedEvent(const OHOS::HiviewDFX::Event & msg)47 void ExtendEventListener::OnUnorderedEvent(const OHOS::HiviewDFX::Event& msg)
48 {
49 printf("cur listener:%s OnUnorderedEvent eventId_:%u \n", name_.c_str(), msg.eventId_);
50 unorderEventCount_++;
51 auto message = msg.GetValue("message");
52 processedUnorderedEvents_[message] = msg.sender_;
53 }
54
GetListenerName()55 std::string ExtendEventListener::GetListenerName()
56 {
57 return name_;
58 }
59 /**
60 * @tc.name: EventDispatchQueueCreateTest001
61 * @tc.desc: create and init an event dispatch queue
62 * @tc.type: FUNC
63 * @tc.require: AR000DPTSU
64 */
65 HWTEST_F(EventDispatchQueueTest, EventDispatchQueueCreateTest001, TestSize.Level3)
66 {
67 printf("EventDispatchQueueTest.\n");
68 OHOS::HiviewDFX::HiviewPlatform& platform = HiviewPlatform::GetInstance();
69 if (!platform.InitEnvironment("/data/test/test_data/hiview_platform_config0")) {
70 printf("Fail to init environment. \n");
71 }
72 auto unorderQueue = std::make_shared<EventDispatchQueue>("test1", Event::ManageType::UNORDERED, &platform);
73 ASSERT_EQ(false, unorderQueue->IsRunning());
74 unorderQueue->Start();
75 sleep(1);
76 ASSERT_EQ(true, unorderQueue->IsRunning());
77 unorderQueue->Stop();
78 ASSERT_EQ(false, unorderQueue->IsRunning());
79 }
80
81 /**
82 * @tc.name: UnorderEventDispatchTest001
83 * @tc.desc: create event and send it to a unorder dispatch queue
84 * @tc.type: FUNC
85 * @tc.require: AR000DPTSU
86 */
87 HWTEST_F(EventDispatchQueueTest, UnorderEventDispatchTest001, TestSize.Level3)
88 {
89 printf("EventDispatchQueueTest.\n");
90 OHOS::HiviewDFX::HiviewPlatform& platform = HiviewPlatform::GetInstance();
91 if (!platform.InitEnvironment("/data/test/test_data/hiview_platform_config0")) {
92 printf("Fail to init environment. \n");
93 }
94 auto unorder = std::make_shared<EventDispatchQueue>("disp-unorder", Event::ManageType::UNORDERED, &platform);
95 unorder->Start();
96
97 auto listener1 = std::make_shared<ExtendEventListener>("listener1");
98 platform.RegisterUnorderedEventListener(listener1);
99 listener1->AddListenerInfo(Event::MessageType::RAW_EVENT, EventListener::EventIdRange(EVENT_ID_0, EVENT_ID_2));
100 auto listener2 = std::make_shared<ExtendEventListener>("listener2");
101 platform.RegisterUnorderedEventListener(listener2);
102 listener2->AddListenerInfo(Event::MessageType::RAW_EVENT, EVENT_ID_2);
103
104 std::set<EventListener::EventIdRange> listenerInfo1;
105 auto ret1 = platform.GetListenerInfo(Event::MessageType::RAW_EVENT, listener1->GetListenerName(), listenerInfo1);
106 for (auto& temp : listenerInfo1) {
107 printf("listenerInfo1 begin == %d end == %d\n", temp.begin, temp.end);
108 }
109 ASSERT_EQ(ret1, true);
110
111 std::set<EventListener::EventIdRange> listenerInfo2;
112 auto ret2 = platform.GetListenerInfo(Event::MessageType::RAW_EVENT, listener2->GetListenerName(), listenerInfo2);
113 for (auto& temp : listenerInfo2) {
114 printf("listenerInfo2 begin == %d end == %d\n", temp.begin, temp.end);
115 }
116 ASSERT_EQ(ret2, true);
117
118 auto event1 = CreateEvent("testEvent1", EVENT_ID_0, "test", Event::MessageType::RAW_EVENT);
119 unorder->Enqueue(event1);
120 sleep(1);
121 ASSERT_EQ(listener1->unorderEventCount_, 1ul);
122 ASSERT_EQ(listener2->unorderEventCount_, 0ul);
123 auto event2 = CreateEvent("testEvent1", EVENT_ID_2, "test", Event::MessageType::RAW_EVENT);
124 unorder->Enqueue(event2);
125 sleep(1);
126 ASSERT_EQ(listener1->unorderEventCount_, 2ul);
127 ASSERT_EQ(listener2->unorderEventCount_, 1ul);
128 auto event3 = CreateEvent("testEvent1", EVENT_ID_3, "test", Event::MessageType::RAW_EVENT);
129 unorder->Enqueue(event3);
130 sleep(1);
131 ASSERT_EQ(listener1->unorderEventCount_, 2ul);
132 ASSERT_EQ(listener2->unorderEventCount_, 1ul);
133 auto event4 = CreateEvent("testEvent1", EVENT_ID_2, "test", Event::MessageType::FAULT_EVENT);
134 unorder->Enqueue(event4);
135 sleep(1);
136 ASSERT_EQ(listener1->unorderEventCount_, 2ul);
137 ASSERT_EQ(listener2->unorderEventCount_, 1ul);
138 }
139