1 // 2 // Copyright (C) 2015 The Android Open Source Project 3 // 4 // Licensed under the Apache License, Version 2.0 (the "License"); 5 // you may not use this file except in compliance with the License. 6 // You may obtain a copy of the License at 7 // 8 // http://www.apache.org/licenses/LICENSE-2.0 9 // 10 // Unless required by applicable law or agreed to in writing, software 11 // distributed under the License is distributed on an "AS IS" BASIS, 12 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 // See the License for the specific language governing permissions and 14 // limitations under the License. 15 // 16 17 #ifndef SHILL_TEST_EVENT_DISPATCHER_H_ 18 #define SHILL_TEST_EVENT_DISPATCHER_H_ 19 20 #include <base/macros.h> 21 #include <base/message_loop/message_loop.h> 22 #include <brillo/message_loops/base_message_loop.h> 23 24 #include "shill/event_dispatcher.h" 25 26 namespace shill { 27 28 // Event dispatcher with message loop for testing. 29 class EventDispatcherForTest : public EventDispatcher { 30 public: EventDispatcherForTest()31 EventDispatcherForTest() { 32 chromeos_message_loop_.SetAsCurrent(); 33 } ~EventDispatcherForTest()34 ~EventDispatcherForTest() override {} 35 36 private: 37 // Message loop for testing. 38 base::MessageLoopForIO message_loop_; 39 // The chromeos wrapper for the main message loop. 40 brillo::BaseMessageLoop chromeos_message_loop_{&message_loop_}; 41 42 DISALLOW_COPY_AND_ASSIGN(EventDispatcherForTest); 43 }; 44 45 } // namespace shill 46 47 #endif // SHILL_TEST_EVENT_DISPATCHER_H_ 48