Track CTR of uncommon download warning.

content/public/test/download_test_observer.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "content/public/test/download_test_observer.h"
 
 #include <vector>
 
 #include "base/bind.h"
 #include "base/logging.h"
@@ skipping 181 matching lines @@
 size_t DownloadTestObserver::NumDownloadsSeenInState(
     DownloadItem::DownloadState state) const {
   StateMap::const_iterator it = states_observed_.find(state);
 
   if (it == states_observed_.end())
     return 0;
 
   return it->second;
 }
 
+DownloadManager* DownloadTestObserver::download_manager() {
+  return download_manager_;
+}
+
+DownloadTestObserver::DangerousDownloadAction
+DownloadTestObserver::dangerous_download_action() {
+  return dangerous_download_action_;
+}
+
+std::set<uint32_t>& DownloadTestObserver::dangerous_downloads_seen() {
+  return dangerous_downloads_seen_;
+}
+
 void DownloadTestObserver::DownloadInFinalState(DownloadItem* download) {
   if (finished_downloads_.find(download) != finished_downloads_.end()) {
     // We've already seen the final state on this download.
     return;
   }
 
   // Record the transition.
   finished_downloads_.insert(download);
 
   // Record the state.
   states_observed_[download->GetState()]++;  // Initializes to 0 the first time.
 
   SignalIfFinished();
 }
 
 void DownloadTestObserver::SignalIfFinished() {
   if (waiting_ && IsFinished())
     base::MessageLoopForUI::current()->QuitWhenIdle();
 }
 
 void DownloadTestObserver::AcceptDangerousDownload(uint32_t download_id) {
   // Download manager was shutdown before the UI thread could accept the
   // download.
   if (!download_manager_)
     return;
   DownloadItem* download = download_manager_->GetDownload(download_id);
-  if (download && !download->IsDone())
+  if (download && download->IsDangerous() && !download->IsDone())
     download->ValidateDangerousDownload();
 }
 
 void DownloadTestObserver::DenyDangerousDownload(uint32_t download_id) {
   // Download manager was shutdown before the UI thread could deny the
   // download.
   if (!download_manager_)
     return;
   DownloadItem* download = download_manager_->GetDownload(download_id);
   if (download && !download->IsDone())
@@ skipping 207 matching lines @@
     base::MessageLoopForUI::current()->QuitWhenIdle();
 }
 
 const DownloadUrlParameters::OnStartedCallback
     DownloadTestItemCreationObserver::callback() {
   return base::Bind(
       &DownloadTestItemCreationObserver::DownloadItemCreationCallback, this);
 }
 
 }  // namespace content