Phase II Step 3: Reload LoFi/placeholder images via new ImageResource

third_party/WebKit/Source/core/loader/resource/MockImageResourceClient.cpp

 // Copyright 2016 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/loader/resource/MockImageResourceClient.h"
 
 #include "core/loader/resource/ImageResource.h"
 #include "core/loader/resource/ImageResourceContent.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace blink {
 
 MockImageResourceClient::MockImageResourceClient(ImageResource* resource)
     : MockResourceClient(resource),
       m_imageChangedCount(0),
       m_encodedSizeOnLastImageChanged(0),
       m_imageNotifyFinishedCount(0),
-      m_encodedSizeOnImageNotifyFinished(0) {
-  toImageResource(m_resource)->getContent()->addObserver(this);
+      m_encodedSizeOnImageNotifyFinished(0),
+      m_content(resource->getContent()) {
+  m_content->addObserver(this);
 }
 
 MockImageResourceClient::~MockImageResourceClient() {}
 
-void MockImageResourceClient::removeAsClient() {
-  toImageResource(m_resource)->getContent()->removeObserver(this);
+void MockImageResourceClient::removeAsClientAndObserver() {
+  m_content->removeObserver(this);
   MockResourceClient::removeAsClient();
 }
 
 void MockImageResourceClient::dispose() {
-  if (m_resource)
-    toImageResource(m_resource)->getContent()->removeObserver(this);
+  if (m_content)
+    m_content->removeObserver(this);
   MockResourceClient::dispose();
 }
 
 void MockImageResourceClient::imageChanged(ImageResourceContent* image,
                                            const IntRect*) {
   m_imageChangedCount++;
-  m_encodedSizeOnLastImageChanged = m_resource->encodedSize();
+  m_encodedSizeOnLastImageChanged = m_content->resourceForTest()->encodedSize();
 }
 
 void MockImageResourceClient::imageNotifyFinished(ImageResourceContent* image) {
   ASSERT_EQ(0, m_imageNotifyFinishedCount);
   m_imageNotifyFinishedCount++;
-  m_encodedSizeOnImageNotifyFinished = m_resource->encodedSize();
+  m_encodedSizeOnImageNotifyFinished =
+      m_content->resourceForTest()->encodedSize();
 }
 
 bool MockImageResourceClient::notifyFinishedCalled() const {
   return m_notifyFinishedCalled;
 }
 
 }  // namespace blink