Handle automation server crashes.

chrome_frame/test/ie_event_sink.h

 // Copyright (c) 2010 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_FRAME_TEST_IE_EVENT_SINK_H_
 #define CHROME_FRAME_TEST_IE_EVENT_SINK_H_
 
 #include <atlbase.h>
 #include <atlwin.h>
 #include <exdispid.h>
@@ skipping 39 matching lines @@
   virtual void OnQuit() {}
 
   // CF callbacks
   virtual void OnLoad(const wchar_t* url) {}
   virtual void OnLoadError(const wchar_t* url) {}
   virtual void OnMessage(const wchar_t* message, const wchar_t* origin,
                          const wchar_t* source) {}
   virtual void OnNewBrowserWindow(IDispatch* new_window, const wchar_t* url) {}
 };
 
+// Listener for IPropertyNotifySink.
+class PropertyNotifySinkListener {
+ public:
+  virtual ~PropertyNotifySinkListener() {}
+  virtual void OnChanged(DISPID dispid) {}
+  virtual void OnRequestEdit(DISPID dispid) {}
+};
+
 // This class sets up event sinks to the IWebBrowser interface. It forwards
 // all events to its listener.
 // TODO(kkania): Delete WebBrowserEventSink and use this class instead for
 // the reliability tests.
 class IEEventSink
     : public CComObjectRootEx<CComSingleThreadModel>,
       public IDispEventSimpleImpl<0, IEEventSink,
                                   &DIID_DWebBrowserEvents2>,
       public IUnknown {
  public:
@@ skipping 152 matching lines @@
   static _ATL_FUNC_INFO kBeforeNavigate2Info;
   static _ATL_FUNC_INFO kNavigateComplete2Info;
   static _ATL_FUNC_INFO kNavigateErrorInfo;
   static _ATL_FUNC_INFO kNewWindow2Info;
   static _ATL_FUNC_INFO kNewWindow3Info;
   static _ATL_FUNC_INFO kVoidMethodInfo;
   static _ATL_FUNC_INFO kDocumentCompleteInfo;
   static _ATL_FUNC_INFO kFileDownloadInfo;
 };
 
+class PropertyNotifySinkImpl
+    : public CComObjectRootEx<CComSingleThreadModel>,
+      public IPropertyNotifySink {
+ public:
+  PropertyNotifySinkImpl() : listener_(NULL) {
+  }
+
+BEGIN_COM_MAP(PropertyNotifySinkImpl)
+  COM_INTERFACE_ENTRY(IPropertyNotifySink)
+END_COM_MAP()
+
+  STDMETHOD(OnChanged)(DISPID dispid) {
+    if (listener_)
+      listener_->OnChanged(dispid);
+    return S_OK;
+  }
+
+  STDMETHOD(OnRequestEdit)(DISPID dispid) {
+    if (listener_)
+      listener_->OnRequestEdit(dispid);
+    return S_OK;
+  }
+
+  void set_listener(PropertyNotifySinkListener* listener) {
+    DCHECK(listener_ == NULL || listener == NULL);
+    listener_ = listener;
+  }
+
+ protected:
+  PropertyNotifySinkListener* listener_;
+};
+
 }  // namespace chrome_frame_test
 
 #endif  // CHROME_FRAME_TEST_IE_EVENT_SINK_H_