OLD | NEW |
1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "core/fetch/MockResourceClients.h" | 5 #include "core/fetch/MockResourceClients.h" |
6 | 6 |
7 #include "core/fetch/ImageResource.h" | 7 #include "core/fetch/ImageResource.h" |
| 8 #include "core/fetch/ImageResourceContent.h" |
8 #include "testing/gtest/include/gtest/gtest.h" | 9 #include "testing/gtest/include/gtest/gtest.h" |
9 | 10 |
10 namespace blink { | 11 namespace blink { |
11 | 12 |
12 MockResourceClient::MockResourceClient(Resource* resource) | 13 MockResourceClient::MockResourceClient(Resource* resource) |
13 : m_resource(resource), | 14 : m_resource(resource), |
14 m_notifyFinishedCalled(false), | 15 m_notifyFinishedCalled(false), |
15 m_encodedSizeOnNotifyFinished(0) { | 16 m_encodedSizeOnNotifyFinished(0) { |
16 ThreadState::current()->registerPreFinalizer(this); | 17 ThreadState::current()->registerPreFinalizer(this); |
17 m_resource->addClient(this); | 18 m_resource->addClient(this); |
(...skipping 23 matching lines...) Expand all Loading... |
41 visitor->trace(m_resource); | 42 visitor->trace(m_resource); |
42 ResourceClient::trace(visitor); | 43 ResourceClient::trace(visitor); |
43 } | 44 } |
44 | 45 |
45 MockImageResourceClient::MockImageResourceClient(ImageResource* resource) | 46 MockImageResourceClient::MockImageResourceClient(ImageResource* resource) |
46 : MockResourceClient(resource), | 47 : MockResourceClient(resource), |
47 m_imageChangedCount(0), | 48 m_imageChangedCount(0), |
48 m_encodedSizeOnLastImageChanged(0), | 49 m_encodedSizeOnLastImageChanged(0), |
49 m_imageNotifyFinishedCount(0), | 50 m_imageNotifyFinishedCount(0), |
50 m_encodedSizeOnImageNotifyFinished(0) { | 51 m_encodedSizeOnImageNotifyFinished(0) { |
51 toImageResource(m_resource.get())->addObserver(this); | 52 toImageResource(m_resource)->getContent()->addObserver(this); |
52 } | 53 } |
53 | 54 |
54 MockImageResourceClient::~MockImageResourceClient() {} | 55 MockImageResourceClient::~MockImageResourceClient() {} |
55 | 56 |
56 void MockImageResourceClient::removeAsClient() { | 57 void MockImageResourceClient::removeAsClient() { |
57 toImageResource(m_resource.get())->removeObserver(this); | 58 toImageResource(m_resource)->getContent()->removeObserver(this); |
58 MockResourceClient::removeAsClient(); | 59 MockResourceClient::removeAsClient(); |
59 } | 60 } |
60 | 61 |
61 void MockImageResourceClient::dispose() { | 62 void MockImageResourceClient::dispose() { |
62 if (m_resource) | 63 if (m_resource) |
63 toImageResource(m_resource.get())->removeObserver(this); | 64 toImageResource(m_resource)->getContent()->removeObserver(this); |
64 MockResourceClient::dispose(); | 65 MockResourceClient::dispose(); |
65 } | 66 } |
66 | 67 |
67 void MockImageResourceClient::imageChanged(ImageResource* image, | 68 void MockImageResourceClient::imageChanged(ImageResourceContent* image, |
68 const IntRect*) { | 69 const IntRect*) { |
69 m_imageChangedCount++; | 70 m_imageChangedCount++; |
70 m_encodedSizeOnLastImageChanged = image->encodedSize(); | 71 m_encodedSizeOnLastImageChanged = m_resource->encodedSize(); |
71 } | 72 } |
72 | 73 |
73 void MockImageResourceClient::imageNotifyFinished(ImageResource* image) { | 74 void MockImageResourceClient::imageNotifyFinished(ImageResourceContent* image) { |
74 ASSERT_EQ(0, m_imageNotifyFinishedCount); | 75 ASSERT_EQ(0, m_imageNotifyFinishedCount); |
75 m_imageNotifyFinishedCount++; | 76 m_imageNotifyFinishedCount++; |
76 m_encodedSizeOnImageNotifyFinished = image->encodedSize(); | 77 m_encodedSizeOnImageNotifyFinished = m_resource->encodedSize(); |
77 } | 78 } |
78 | 79 |
79 bool MockImageResourceClient::notifyFinishedCalled() const { | 80 bool MockImageResourceClient::notifyFinishedCalled() const { |
80 return m_notifyFinishedCalled; | 81 return m_notifyFinishedCalled; |
81 } | 82 } |
82 | 83 |
83 } // namespace blink | 84 } // namespace blink |
OLD | NEW |