1 //===-- BroadcasterTest.cpp -------------------------------------*- C++ -*-===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 
10 #include "gtest/gtest.h"
11 
12 #include "lldb/Utility/Broadcaster.h"
13 #include "lldb/Utility/Event.h"
14 #include "lldb/Utility/Listener.h"
15 #include "lldb/Utility/Predicate.h"
16 
17 #include <thread>
18 
19 using namespace lldb;
20 using namespace lldb_private;
21 
22 TEST(BroadcasterTest, BroadcastEvent) {
23   EventSP event_sp;
24   Broadcaster broadcaster(nullptr, "test-broadcaster");
25   std::chrono::seconds timeout(0);
26 
27   // Create a listener, sign it up, make sure it receives an event.
28   ListenerSP listener1_sp = Listener::MakeListener("test-listener1");
29   const uint32_t event_mask1 = 1;
30   EXPECT_EQ(event_mask1,
31             listener1_sp->StartListeningForEvents(&broadcaster, event_mask1));
32   broadcaster.BroadcastEvent(event_mask1, nullptr);
33   EXPECT_TRUE(listener1_sp->GetEvent(event_sp, timeout));
34   EXPECT_EQ(event_mask1, event_sp->GetType());
35 
36   {
37     // Add one more listener, make sure it works as well.
38     ListenerSP listener2_sp = Listener::MakeListener("test-listener2");
39     const uint32_t event_mask2 = 1;
40     EXPECT_EQ(event_mask2, listener2_sp->StartListeningForEvents(
41                                &broadcaster, event_mask1 | event_mask2));
42     broadcaster.BroadcastEvent(event_mask2, nullptr);
43     EXPECT_TRUE(listener2_sp->GetEvent(event_sp, timeout));
44     EXPECT_EQ(event_mask2, event_sp->GetType());
45 
46     // Both listeners should get this event.
47     broadcaster.BroadcastEvent(event_mask1, nullptr);
48     EXPECT_TRUE(listener1_sp->GetEvent(event_sp, timeout));
49     EXPECT_EQ(event_mask1, event_sp->GetType());
50     EXPECT_TRUE(listener2_sp->GetEvent(event_sp, timeout));
51     EXPECT_EQ(event_mask2, event_sp->GetType());
52   }
53 
54   // Now again only one listener should be active.
55   broadcaster.BroadcastEvent(event_mask1, nullptr);
56   EXPECT_TRUE(listener1_sp->GetEvent(event_sp, timeout));
57   EXPECT_EQ(event_mask1, event_sp->GetType());
58 }
59 
60 TEST(BroadcasterTest, EventTypeHasListeners) {
61   EventSP event_sp;
62   Broadcaster broadcaster(nullptr, "test-broadcaster");
63 
64   const uint32_t event_mask = 1;
65   EXPECT_FALSE(broadcaster.EventTypeHasListeners(event_mask));
66 
67   {
68     ListenerSP listener_sp = Listener::MakeListener("test-listener");
69     EXPECT_EQ(event_mask,
70               listener_sp->StartListeningForEvents(&broadcaster, event_mask));
71     EXPECT_TRUE(broadcaster.EventTypeHasListeners(event_mask));
72   }
73 
74   EXPECT_FALSE(broadcaster.EventTypeHasListeners(event_mask));
75 }
76