Mercurial > hg > orthanc-stone
annotate UnitTestsSources/TestMessageBroker.cpp @ 267:89d02de83c03 am-2
added declaretion of messages handled/emitted
| author | am@osimis.io |
|---|---|
| date | Wed, 22 Aug 2018 14:59:20 +0200 |
| parents | c9cf95b49a86 |
| children | 5bd4161bf11b |
| 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 | |
| 45 void HandleMessage(const OrthancStone::IObservable& from, const OrthancStone::IMessage& message) { | |
| 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 | |
|
266
c9cf95b49a86
removed OrthancSlicesLoader::ISliceLoaderObserver; now using standard messages instead
am@osimis.io
parents:
258
diff
changeset
|
73 void HandleMessage(const 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 } |