Split ImageResourceClient into ResourceClient and ImageResourceObserver [2/2]

third_party/WebKit/Source/core/fetch/MockResourceClients.cpp

 // Copyright 2015 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 "core/fetch/MockResourceClients.h"
 
 #include "core/fetch/ImageResource.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace blink {
@@ skipping 18 matching lines @@
 
 void MockResourceClient::removeAsClient()
 {
     m_resource->removeClient(this);
     m_resource = nullptr;
 }
 
 MockImageResourceClient::MockImageResourceClient(PassRefPtrWillBeRawPtr<ImageResource> resource)
     : MockResourceClient(resource)
     , m_imageChangedCount(0)
+    , m_imageNotifyFinishedCount(0)
 {
     toImageResource(m_resource.get())->addObserver(this);
 }
 
 MockImageResourceClient::~MockImageResourceClient()
 {
     if (m_resource)
         toImageResource(m_resource.get())->removeObserver(this);
 }
 
 void MockImageResourceClient::removeAsClient()
 {
     toImageResource(m_resource.get())->removeObserver(this);
     MockResourceClient::removeAsClient();
 }
 
 void MockImageResourceClient::imageChanged(ImageResource*, const IntRect*)
 {
     m_imageChangedCount++;
 }
 
+void MockImageResourceClient::imageNotifyFinished(ImageResource*)
+{
+    ASSERT_EQ(0, m_imageNotifyFinishedCount);
+    m_imageNotifyFinishedCount++;
+}
+
+bool MockImageResourceClient::notifyFinishedCalled() const
+{
+    EXPECT_EQ(m_notifyFinishedCalled ? 1 : 0, m_imageNotifyFinishedCount);
+
+    return m_notifyFinishedCalled;
+}
+
 } // namespace blink