Close privet printer notifications when clicked.

chrome/browser/printing/cloud_print/privet_notifications.h

 // Copyright 2013 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.
 
 #ifndef CHROME_BROWSER_PRINTING_CLOUD_PRINT_PRIVET_NOTIFICATIONS_H_
 #define CHROME_BROWSER_PRINTING_CLOUD_PRINT_PRIVET_NOTIFICATIONS_H_
 
 #include <map>
 #include <memory>
 #include <string>
 
 #include "chrome/browser/notifications/notification_delegate.h"
 #include "chrome/browser/printing/cloud_print/privet_device_lister.h"
 #include "chrome/browser/printing/cloud_print/privet_http.h"
 #include "components/keyed_service/core/keyed_service.h"
 #include "components/prefs/pref_member.h"
 
 class NotificationUIManager;
+class Profile;
 
 namespace content {
 class BrowserContext;
 }
 
 namespace local_discovery {
 class ServiceDiscoverySharedClient;
 }
 
 namespace cloud_print {
 
 class PrivetDeviceLister;
 class PrivetHTTPAsynchronousFactory;
 class PrivetHTTPResolution;
+class PrivetNotificationDelegate;
 struct DeviceDescription;
 
 #if defined(ENABLE_MDNS)
 class PrivetTrafficDetector;
 #endif  // ENABLE_MDNS
 
 // Contains logic related to notifications not tied actually displaying them.
 class PrivetNotificationsListener  {
  public:
   class Delegate {
@@ skipping 69 matching lines @@
   void PrivetRemoveNotification() override;
 
   static bool IsEnabled();
   static bool IsForced();
 
  private:
   void Start();
   void OnNotificationsEnabledChanged();
   void StartLister();
 
+  // Virtual for testing. The returned delegate is refcounted.
+  virtual PrivetNotificationDelegate* CreateNotificationDelegate(
+      Profile* profile);
+
   content::BrowserContext* const profile_;
   std::unique_ptr<PrivetDeviceLister> device_lister_;
   scoped_refptr<local_discovery::ServiceDiscoverySharedClient>
       service_discovery_client_;
   std::unique_ptr<PrivetNotificationsListener> privet_notifications_listener_;
   BooleanPrefMember enable_privet_notification_member_;
 
 #if defined(ENABLE_MDNS)
   scoped_refptr<PrivetTrafficDetector> traffic_detector_;
 #endif  // ENABLE_MDNS
 };
 
 class PrivetNotificationDelegate : public NotificationDelegate {
  public:
-  explicit PrivetNotificationDelegate(content::BrowserContext* profile);
+  explicit PrivetNotificationDelegate(Profile* profile);
 
   // NotificationDelegate implementation.
   std::string id() const override;
   void ButtonClick(int button_index) override;
 
- private:
+ protected:
   // Refcounted.
   ~PrivetNotificationDelegate() override;
 
-  void OpenTab(const GURL& url);
-  void DisableNotifications();
+ private:
+  // ButtonClick() response handlers. Virtual for testing.
+  virtual void OpenTab(const GURL& url);
+  virtual void DisableNotifications();
 
-  content::BrowserContext* const profile_;
+  void CloseNotification();
+
+  Profile* const profile_;
 };
 
 }  // namespace cloud_print
 
 #endif  // CHROME_BROWSER_PRINTING_CLOUD_PRINT_PRIVET_NOTIFICATIONS_H_