Mercurial > hg > orthanc-stone
annotate UnitTestsSources/TestMessageBroker.cpp @ 286:6b3d91857b96 am-2
fix button titles
author | am@osimis.io |
---|---|
date | Tue, 28 Aug 2018 15:34:20 +0200 |
parents | 5bd4161bf11b |
children | 3897f9f28cfa |
rev | line source |
---|---|
251 | 1 /** |
2 * Stone of Orthanc | |
3 * Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics | |
4 * Department, University Hospital of Liege, Belgium | |
5 * Copyright (C) 2017-2018 Osimis S.A., Belgium | |
6 * | |
7 * This program is free software: you can redistribute it and/or | |
8 * modify it under the terms of the GNU Affero General Public License | |
9 * as published by the Free Software Foundation, either version 3 of | |
10 * the License, or (at your option) any later version. | |
11 * | |
12 * This program is distributed in the hope that it will be useful, but | |
13 * WITHOUT ANY WARRANTY; without even the implied warranty of | |
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
15 * Affero General Public License for more details. | |
16 * | |
17 * You should have received a copy of the GNU Affero General Public License | |
18 * along with this program. If not, see <http://www.gnu.org/licenses/>. | |
19 **/ | |
20 | |
21 | |
22 #include "gtest/gtest.h" | |
23 | |
24 #include "../Framework/Messages/MessageBroker.h" | |
25 #include "../Framework/Messages/IMessage.h" | |
26 #include "../Framework/Messages/IObservable.h" | |
27 #include "../Framework/Messages/IObserver.h" | |
28 #include "../Framework/StoneEnumerations.h" | |
29 | |
30 | |
267 | 31 static int test1Counter = 0; |
32 static int test2Counter = 0; | |
33 class MyFullObserver : public OrthancStone::IObserver | |
34 { | |
35 | |
36 public: | |
37 MyFullObserver(OrthancStone::MessageBroker& broker) | |
38 : OrthancStone::IObserver(broker) | |
39 { | |
40 DeclareHandledMessage(OrthancStone::MessageType_Test1); | |
41 DeclareIgnoredMessage(OrthancStone::MessageType_Test2); | |
42 } | |
43 | |
44 | |
268
5bd4161bf11b
removed constness of the observable when emitting a message
am@osimis.io
parents:
267
diff
changeset
|
45 void HandleMessage(OrthancStone::IObservable& from, const OrthancStone::IMessage& message) { |
267 | 46 switch (message.GetType()) |
47 { | |
48 case OrthancStone::MessageType_Test1: | |
49 test1Counter++; | |
50 break; | |
51 case OrthancStone::MessageType_Test2: | |
52 test2Counter++; | |
53 break; | |
54 default: | |
55 throw OrthancStone::MessageNotDeclaredException(message.GetType()); | |
56 } | |
57 } | |
58 | |
59 }; | |
60 | |
61 class MyPartialObserver : public OrthancStone::IObserver | |
251 | 62 { |
63 | |
64 public: | |
267 | 65 MyPartialObserver(OrthancStone::MessageBroker& broker) |
251 | 66 : OrthancStone::IObserver(broker) |
267 | 67 { |
68 DeclareHandledMessage(OrthancStone::MessageType_Test1); | |
69 // don't declare Test2 on purpose | |
70 } | |
251 | 71 |
72 | |
268
5bd4161bf11b
removed constness of the observable when emitting a message
am@osimis.io
parents:
267
diff
changeset
|
73 void HandleMessage(OrthancStone::IObservable& from, const OrthancStone::IMessage& message) { |
267 | 74 switch (message.GetType()) |
75 { | |
76 case OrthancStone::MessageType_Test1: | |
77 test1Counter++; | |
78 break; | |
79 case OrthancStone::MessageType_Test2: | |
80 test2Counter++; | |
81 break; | |
82 default: | |
83 throw OrthancStone::MessageNotDeclaredException(message.GetType()); | |
251 | 84 } |
267 | 85 } |
251 | 86 |
267 | 87 }; |
88 | |
89 | |
90 class MyObservable : public OrthancStone::IObservable | |
91 { | |
92 | |
93 public: | |
94 MyObservable(OrthancStone::MessageBroker& broker) | |
95 : OrthancStone::IObservable(broker) | |
96 { | |
97 DeclareEmittableMessage(OrthancStone::MessageType_Test1); | |
98 DeclareEmittableMessage(OrthancStone::MessageType_Test2); | |
251 | 99 } |
100 | |
101 }; | |
102 | |
103 | |
104 TEST(MessageBroker, NormalUsage) | |
105 { | |
106 OrthancStone::MessageBroker broker; | |
267 | 107 MyObservable observable(broker); |
251 | 108 |
267 | 109 test1Counter = 0; |
251 | 110 |
111 // no observers have been registered -> nothing shall happen | |
267 | 112 observable.EmitMessage(OrthancStone::IMessage(OrthancStone::MessageType_Test1)); |
251 | 113 |
267 | 114 ASSERT_EQ(0, test1Counter); |
251 | 115 |
116 // register an observer, check it is called | |
267 | 117 MyFullObserver fullObserver(broker); |
118 ASSERT_NO_THROW(observable.RegisterObserver(fullObserver)); | |
251 | 119 |
267 | 120 observable.EmitMessage(OrthancStone::IMessage(OrthancStone::MessageType_Test1)); |
251 | 121 |
267 | 122 ASSERT_EQ(1, test1Counter); |
251 | 123 |
267 | 124 // register an invalid observer, check it raises an exception |
125 MyPartialObserver partialObserver(broker); | |
126 ASSERT_THROW(observable.RegisterObserver(partialObserver), OrthancStone::MessageNotDeclaredException); | |
251 | 127 |
267 | 128 // check an exception is thrown when the observable emits an undeclared message |
129 ASSERT_THROW(observable.EmitMessage(OrthancStone::IMessage(OrthancStone::MessageType_LayerSource_GeometryReady)), OrthancStone::MessageNotDeclaredException); | |
251 | 130 |
131 // unregister the observer, make sure nothing happens afterwards | |
267 | 132 observable.UnregisterObserver(fullObserver); |
133 observable.EmitMessage(OrthancStone::IMessage(OrthancStone::MessageType_Test1)); | |
134 ASSERT_EQ(1, test1Counter); | |
251 | 135 } |
136 | |
137 TEST(MessageBroker, DeleteObserverWhileRegistered) | |
138 { | |
139 OrthancStone::MessageBroker broker; | |
267 | 140 MyObservable observable(broker); |
251 | 141 |
267 | 142 test1Counter = 0; |
251 | 143 |
144 { | |
145 // register an observer, check it is called | |
267 | 146 MyFullObserver observer(broker); |
251 | 147 observable.RegisterObserver(observer); |
148 | |
267 | 149 observable.EmitMessage(OrthancStone::IMessage(OrthancStone::MessageType_Test1)); |
251 | 150 |
267 | 151 ASSERT_EQ(1, test1Counter); |
251 | 152 } |
153 | |
154 // at this point, the observer has been deleted, the handle shall not be called again (and it shall not crash !) | |
267 | 155 observable.EmitMessage(OrthancStone::IMessage(OrthancStone::MessageType_Test1)); |
251 | 156 |
267 | 157 ASSERT_EQ(1, test1Counter); |
251 | 158 } |