1// Copyright (c) 2013 The Chromium Authors. All rights reserved. 2// Use of this source code is governed by a BSD-style license that can be 3// found in the LICENSE file. 4 5// ATL headers have to go first. 6#include <atlbase.h> 7#include <atlhost.h> 8#include <string> 9 10#include "base/basictypes.h" 11#include "base/bind.h" 12#include "base/bind_helpers.h" 13#include "base/guid.h" 14#include "base/message_loop/message_loop.h" 15#include "base/run_loop.h" 16#include "base/win/scoped_com_initializer.h" 17#include "remoting/base/auto_thread_task_runner.h" 18#include "remoting/host/win/rdp_client.h" 19#include "remoting/host/win/wts_terminal_monitor.h" 20#include "testing/gmock/include/gmock/gmock.h" 21#include "testing/gmock_mutant.h" 22#include "testing/gtest/include/gtest/gtest.h" 23#include "third_party/webrtc/modules/desktop_capture/desktop_geometry.h" 24 25using testing::_; 26using testing::AtMost; 27using testing::InvokeWithoutArgs; 28 29namespace remoting { 30 31namespace { 32 33// Default width and hight of the RDP client window. 34const long kDefaultWidth = 1024; 35const long kDefaultHeight = 768; 36 37class MockRdpClientEventHandler : public RdpClient::EventHandler { 38 public: 39 MockRdpClientEventHandler() {} 40 virtual ~MockRdpClientEventHandler() {} 41 42 MOCK_METHOD0(OnRdpConnected, void()); 43 MOCK_METHOD0(OnRdpClosed, void()); 44 45 private: 46 DISALLOW_COPY_AND_ASSIGN(MockRdpClientEventHandler); 47}; 48 49// a14498c6-7f3b-4e42-9605-6c4a20d53c87 50static GUID RdpClientModuleLibid = { 51 0xa14498c6, 52 0x7f3b, 53 0x4e42, 54 { 0x96, 0x05, 0x6c, 0x4a, 0x20, 0xd5, 0x3c, 0x87 } 55}; 56 57class RdpClientModule : public ATL::CAtlModuleT<RdpClientModule> { 58 public: 59 RdpClientModule(); 60 virtual ~RdpClientModule(); 61 62 DECLARE_LIBID(RdpClientModuleLibid) 63 64 private: 65 base::win::ScopedCOMInitializer com_initializer_; 66}; 67 68RdpClientModule::RdpClientModule() { 69 AtlAxWinInit(); 70} 71 72RdpClientModule::~RdpClientModule() { 73 AtlAxWinTerm(); 74 ATL::_pAtlModule = NULL; 75} 76 77} // namespace 78 79class RdpClientTest : public testing::Test { 80 public: 81 RdpClientTest(); 82 virtual ~RdpClientTest(); 83 84 virtual void SetUp() OVERRIDE; 85 virtual void TearDown() OVERRIDE; 86 87 // Caaled when an RDP connection is established. 88 void OnRdpConnected(); 89 90 // Tears down |rdp_client_|. 91 void CloseRdpClient(); 92 93 protected: 94 // The ATL module instance required by the ATL code. 95 scoped_ptr<RdpClientModule> module_; 96 97 // The UI message loop used by RdpClient. The loop is stopped once there is no 98 // more references to |task_runner_|. 99 base::MessageLoop message_loop_; 100 base::RunLoop run_loop_; 101 scoped_refptr<AutoThreadTaskRunner> task_runner_; 102 103 // Mocks RdpClient::EventHandler for testing. 104 MockRdpClientEventHandler event_handler_; 105 106 // Points to the object being tested. 107 scoped_ptr<RdpClient> rdp_client_; 108 109 // Unique terminal identifier passed to RdpClient. 110 std::string terminal_id_; 111}; 112 113RdpClientTest::RdpClientTest() 114 : message_loop_(base::MessageLoop::TYPE_UI) { 115} 116 117RdpClientTest::~RdpClientTest() { 118} 119 120void RdpClientTest::SetUp() { 121 // Arrange to run |message_loop_| until no components depend on it. 122 task_runner_ = new AutoThreadTaskRunner( 123 message_loop_.message_loop_proxy(), run_loop_.QuitClosure()); 124 125 module_.reset(new RdpClientModule()); 126} 127 128void RdpClientTest::TearDown() { 129 EXPECT_TRUE(!rdp_client_); 130 131 module_.reset(); 132} 133 134void RdpClientTest::OnRdpConnected() { 135 uint32 session_id = WtsTerminalMonitor::LookupSessionId(terminal_id_); 136 137 std::string id; 138 EXPECT_TRUE(WtsTerminalMonitor::LookupTerminalId(session_id, &id)); 139 EXPECT_EQ(id, terminal_id_); 140 141 message_loop_.PostTask(FROM_HERE, base::Bind(&RdpClientTest::CloseRdpClient, 142 base::Unretained(this))); 143} 144 145void RdpClientTest::CloseRdpClient() { 146 EXPECT_TRUE(rdp_client_); 147 148 rdp_client_.reset(); 149} 150 151// Creates a loopback RDP connection. 152TEST_F(RdpClientTest, Basic) { 153 terminal_id_ = base::GenerateGUID(); 154 155 // An ability to establish a loopback RDP connection depends on many factors 156 // including OS SKU and having RDP enabled. Accept both successful connection 157 // and a connection error as a successful outcome. 158 EXPECT_CALL(event_handler_, OnRdpConnected()) 159 .Times(AtMost(1)) 160 .WillOnce(Invoke(this, &RdpClientTest::OnRdpConnected)); 161 EXPECT_CALL(event_handler_, OnRdpClosed()) 162 .Times(AtMost(1)) 163 .WillOnce(InvokeWithoutArgs(this, &RdpClientTest::CloseRdpClient)); 164 165 rdp_client_.reset(new RdpClient( 166 task_runner_, task_runner_, 167 webrtc::DesktopSize(kDefaultWidth, kDefaultHeight), 168 terminal_id_, &event_handler_)); 169 task_runner_ = NULL; 170 171 run_loop_.Run(); 172} 173 174} // namespace remoting 175