| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "chrome/renderer/safe_browsing/phishing_classifier_delegate.h" | 5 #include "chrome/renderer/safe_browsing/phishing_classifier_delegate.h" |
| 6 | 6 |
| 7 #include "base/command_line.h" | 7 #include "base/command_line.h" |
| 8 #include "base/memory/scoped_ptr.h" | 8 #include "base/memory/scoped_ptr.h" |
| 9 #include "base/run_loop.h" | 9 #include "base/run_loop.h" |
| 10 #include "base/strings/string_number_conversions.h" | 10 #include "base/strings/string_number_conversions.h" |
| (...skipping 69 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 80 }; | 80 }; |
| 81 | 81 |
| 82 class InterceptingMessageFilter : public content::BrowserMessageFilter { | 82 class InterceptingMessageFilter : public content::BrowserMessageFilter { |
| 83 public: | 83 public: |
| 84 InterceptingMessageFilter() | 84 InterceptingMessageFilter() |
| 85 : BrowserMessageFilter(SafeBrowsingMsgStart), | 85 : BrowserMessageFilter(SafeBrowsingMsgStart), |
| 86 waiting_message_loop_(NULL) { | 86 waiting_message_loop_(NULL) { |
| 87 } | 87 } |
| 88 | 88 |
| 89 const ClientPhishingRequest* verdict() const { return verdict_.get(); } | 89 const ClientPhishingRequest* verdict() const { return verdict_.get(); } |
| 90 virtual bool OnMessageReceived(const IPC::Message& message, | 90 virtual bool OnMessageReceived(const IPC::Message& message) OVERRIDE { |
| 91 bool* message_was_ok) OVERRIDE { | |
| 92 bool handled = true; | 91 bool handled = true; |
| 93 IPC_BEGIN_MESSAGE_MAP(InterceptingMessageFilter, message) | 92 IPC_BEGIN_MESSAGE_MAP(InterceptingMessageFilter, message) |
| 94 IPC_MESSAGE_HANDLER(SafeBrowsingHostMsg_PhishingDetectionDone, | 93 IPC_MESSAGE_HANDLER(SafeBrowsingHostMsg_PhishingDetectionDone, |
| 95 OnPhishingDetectionDone) | 94 OnPhishingDetectionDone) |
| 96 IPC_MESSAGE_UNHANDLED(handled = false); | 95 IPC_MESSAGE_UNHANDLED(handled = false); |
| 97 IPC_END_MESSAGE_MAP() | 96 IPC_END_MESSAGE_MAP() |
| 98 return handled; | 97 return handled; |
| 99 } | 98 } |
| 100 | 99 |
| 101 void Reset() { | 100 void Reset() { |
| (...skipping 524 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 626 RunClassificationDone(verdict); | 625 RunClassificationDone(verdict); |
| 627 ASSERT_TRUE(intercepting_filter_->verdict()); | 626 ASSERT_TRUE(intercepting_filter_->verdict()); |
| 628 EXPECT_EQ(verdict.SerializeAsString(), | 627 EXPECT_EQ(verdict.SerializeAsString(), |
| 629 intercepting_filter_->verdict()->SerializeAsString()); | 628 intercepting_filter_->verdict()->SerializeAsString()); |
| 630 | 629 |
| 631 // The delegate will cancel pending classification on destruction. | 630 // The delegate will cancel pending classification on destruction. |
| 632 EXPECT_CALL(*classifier_, CancelPendingClassification()); | 631 EXPECT_CALL(*classifier_, CancelPendingClassification()); |
| 633 } | 632 } |
| 634 | 633 |
| 635 } // namespace safe_browsing | 634 } // namespace safe_browsing |
| OLD | NEW |