Index: third_party/WebKit/Source/core/fetch/MockResourceClients.cpp |
diff --git a/third_party/WebKit/Source/core/fetch/MockResourceClients.cpp b/third_party/WebKit/Source/core/fetch/MockResourceClients.cpp |
index 8c8c86191fc42c194af86a789a1f53146db68323..2fd676bc411bd23245e5221805d3c46eeb9daa38 100644 |
--- a/third_party/WebKit/Source/core/fetch/MockResourceClients.cpp |
+++ b/third_party/WebKit/Source/core/fetch/MockResourceClients.cpp |
@@ -48,33 +48,35 @@ MockImageResourceClient::MockImageResourceClient(ImageResource* 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); |
+ 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 { |