| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2011 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 #include "chrome/browser/remoting/firewall_traversal_observer.h" | |
| 6 | |
| 7 #include "chrome/browser/prefs/pref_service.h" | |
| 8 #include "content/browser/renderer_host/mock_render_process_host.h" | |
| 9 #include "chrome/browser/ui/tab_contents/test_tab_contents_wrapper.h" | |
| 10 #include "chrome/common/chrome_notification_types.h" | |
| 11 #include "chrome/common/pref_names.h" | |
| 12 #include "chrome/test/base/testing_profile.h" | |
| 13 #include "content/browser/browser_thread.h" | |
| 14 #include "content/browser/tab_contents/test_tab_contents.h" | |
| 15 #include "content/common/view_messages.h" | |
| 16 #include "ipc/ipc_message.h" | |
| 17 | |
| 18 class FirewallTraversalTabHelperTest : public TabContentsWrapperTestHarness { | |
| 19 public: | |
| 20 FirewallTraversalTabHelperTest() | |
| 21 : browser_thread_(BrowserThread::UI, &message_loop_) {} | |
| 22 virtual ~FirewallTraversalTabHelperTest() {} | |
| 23 | |
| 24 private: | |
| 25 BrowserThread browser_thread_; | |
| 26 }; | |
| 27 | |
| 28 // Test receiving message. | |
| 29 TEST_F(FirewallTraversalTabHelperTest, TestBasicMessage) { | |
| 30 IPC::TestSink& sink = process()->sink(); | |
| 31 sink.ClearMessages(); | |
| 32 IPC::Message msg(MSG_ROUTING_NONE, | |
| 33 ViewHostMsg_RequestRemoteAccessClientFirewallTraversal::ID, | |
| 34 IPC::Message::PRIORITY_NORMAL); | |
| 35 ASSERT_TRUE(active_rvh()->TestOnMessageReceived(msg)); | |
| 36 const IPC::Message* expect_msg = sink.GetUniqueMessageMatching( | |
| 37 ViewMsg_UpdateRemoteAccessClientFirewallTraversal::ID); | |
| 38 ASSERT_TRUE(expect_msg); | |
| 39 } | |
| 40 | |
| 41 // Test changing preference. | |
| 42 TEST_F(FirewallTraversalTabHelperTest, TestPrefChange) { | |
| 43 IPC::TestSink& sink = process()->sink(); | |
| 44 sink.ClearMessages(); | |
| 45 contents_wrapper()->profile()->GetPrefs()->SetBoolean( | |
| 46 prefs::kRemoteAccessClientFirewallTraversal, | |
| 47 false); | |
| 48 const IPC::Message* expect_msg = sink.GetUniqueMessageMatching( | |
| 49 ViewMsg_UpdateRemoteAccessClientFirewallTraversal::ID); | |
| 50 ASSERT_TRUE(expect_msg); | |
| 51 } | |
| OLD | NEW |