chrome/browser/content_settings/permission_queue_controller_unittest.cc

   1 // Copyright 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 #include "chrome/browser/content_settings/permission_queue_controller.h"
   6
   7 #include "base/synchronization/waitable_event.h"
   8 #include "chrome/browser/chrome_notification_types.h"
   9 #include "chrome/browser/content_settings/permission_request_id.h"
  10 #include "chrome/browser/infobars/infobar_service.h"
  11 #include "chrome/common/content_settings_types.h"
  12 #include "chrome/test/base/chrome_render_view_host_test_harness.h"
  13 #include "chrome/test/base/testing_profile.h"
  14 #include "content/public/browser/web_contents.h"
  15 #include "content/public/test/mock_render_process_host.h"
  16 #include "content/public/test/test_browser_thread.h"
  17 #include "testing/gtest/include/gtest/gtest.h"
  18
  19
  20 // PermissionQueueControllerTests ---------------------------------------------
  21
  22 class PermissionQueueControllerTests : public ChromeRenderViewHostTestHarness {
  23  protected:
  24   PermissionQueueControllerTests() {}
  25   virtual ~PermissionQueueControllerTests() {}
  26
  27   PermissionRequestID RequestID(int bridge_id) {
  28     return PermissionRequestID(
  29         web_contents()->GetRenderProcessHost()->GetID(),
  30         web_contents()->GetRenderViewHost()->GetRoutingID(),
  31         bridge_id,
  32         0);
  33   }
  34
  35  private:
  36   // ChromeRenderViewHostTestHarness:
  37   virtual void SetUp() OVERRIDE {
  38     ChromeRenderViewHostTestHarness::SetUp();
  39     InfoBarService::CreateForWebContents(web_contents());
  40   }
  41
  42   DISALLOW_COPY_AND_ASSIGN(PermissionQueueControllerTests);
  43 };
  44
  45
  46 // ObservationCountingQueueController -----------------------------------------
  47
  48 class ObservationCountingQueueController : public PermissionQueueController {
  49  public:
  50   explicit ObservationCountingQueueController(Profile* profile);
  51   virtual ~ObservationCountingQueueController();
  52
  53   int call_count() const { return call_count_; }
  54
  55  private:
  56   int call_count_;
  57
  58   // PermissionQueueController:
  59   virtual void Observe(int type,
  60                        const content::NotificationSource& source,
  61                        const content::NotificationDetails& details) OVERRIDE;
  62
  63   DISALLOW_COPY_AND_ASSIGN(ObservationCountingQueueController);
  64 };
  65
  66 ObservationCountingQueueController::ObservationCountingQueueController(
  67     Profile* profile)
  68     : PermissionQueueController(profile, CONTENT_SETTINGS_TYPE_GEOLOCATION),
  69       call_count_(0) {
  70 }
  71
  72 ObservationCountingQueueController::~ObservationCountingQueueController() {
  73 }
  74
  75 void ObservationCountingQueueController::Observe(
  76     int type,
  77     const content::NotificationSource& source,
  78     const content::NotificationDetails& details) {
  79   DCHECK_EQ(chrome::NOTIFICATION_TAB_CONTENTS_INFOBAR_REMOVED, type);
  80   ++call_count_;
  81   PermissionQueueController::Observe(type, source, details);
  82 }
  83
  84
  85 // Actual tests ---------------------------------------------------------------
  86
  87 TEST_F(PermissionQueueControllerTests, OneObservationPerInfoBarCancelled) {
  88   // When an infobar is cancelled, the infobar helper sends a notification to
  89   // the controller. If the controller has another infobar queued, it should
  90   // maintain its registration for notifications with the helper, but on the
  91   // last infobar cancellation it should unregister for notifications.
  92   //
  93   // What we don't want is for the controller to unregister and then re-register
  94   // for notifications, which can lead to getting notified multiple times.  This
  95   // test checks that in the case where the controller should remain registered
  96   // for notifications, it gets notified exactly once."
  97   ObservationCountingQueueController queue_controller(profile());
  98   GURL url("http://www.example.com/geolocation");
  99   base::Callback<void(bool)> callback;
 100   queue_controller.CreateInfoBarRequest(RequestID(0), url, url, callback);
 101   queue_controller.CreateInfoBarRequest(RequestID(1), url, url, callback);
 102   queue_controller.CancelInfoBarRequest(RequestID(0));
 103   EXPECT_EQ(1, queue_controller.call_count());
 104 };