2
* Copyright 2013 Google Inc.
4
* Licensed under the Apache License, Version 2.0 (the "License");
5
* you may not use this file except in compliance with the License.
6
* You may obtain a copy of the License at
8
* http://www.apache.org/licenses/LICENSE-2.0
10
* Unless required by applicable law or agreed to in writing, software
11
* distributed under the License is distributed on an "AS IS" BASIS,
12
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13
* See the License for the specific language governing permissions and
14
* limitations under the License.
16
// Author: nikhilmadan@google.com (Nikhil Madan)
18
#ifndef NET_INSTAWEB_REWRITER_PUBLIC_MOCK_CRITICAL_IMAGES_FINDER_H_
19
#define NET_INSTAWEB_REWRITER_PUBLIC_MOCK_CRITICAL_IMAGES_FINDER_H_
21
#include "net/instaweb/rewriter/public/critical_images_finder_test_base.h"
22
#include "net/instaweb/rewriter/public/critical_images_finder.h"
23
#include "net/instaweb/rewriter/rendered_image.pb.h"
24
#include "net/instaweb/util/public/property_cache.h"
25
#include "net/instaweb/util/public/string_util.h"
26
#include "pagespeed/kernel/base/basictypes.h"
27
#include "pagespeed/kernel/base/scoped_ptr.h"
29
namespace net_instaweb {
34
// Mock implementation of CriticalImagesFinder that can store and retrieve
35
// critical images. Note that this doesn't use property cache.
36
class MockCriticalImagesFinder : public TestCriticalImagesFinder {
38
explicit MockCriticalImagesFinder(Statistics* stats)
39
: TestCriticalImagesFinder(NULL, stats), compute_calls_(0) {}
41
MockCriticalImagesFinder(const PropertyCache::Cohort* cohort,
43
: TestCriticalImagesFinder(cohort, stats), compute_calls_(0) {}
45
~MockCriticalImagesFinder();
47
virtual Availability Available(RewriteDriver* driver) {
51
virtual void UpdateCriticalImagesSetInDriver(RewriteDriver* driver);
53
// Extracts rendered image dimensions from property cache.
54
virtual RenderedImages* ExtractRenderedImageDimensionsFromCache(
55
RewriteDriver* driver);
57
virtual void ComputeCriticalImages(RewriteDriver* driver) {
61
int num_compute_calls() { return compute_calls_; }
63
void set_critical_images(StringSet* critical_images) {
64
critical_images_.reset(critical_images);
67
void set_css_critical_images(StringSet* css_critical_images) {
68
css_critical_images_.reset(css_critical_images);
71
void set_rendered_images(RenderedImages* rendered_images) {
72
rendered_images_.reset(rendered_images);
75
virtual bool IsCriticalImageInfoPresent(RewriteDriver* driver) {
81
scoped_ptr<StringSet> critical_images_;
82
scoped_ptr<StringSet> css_critical_images_;
83
scoped_ptr<RenderedImages> rendered_images_;
84
DISALLOW_COPY_AND_ASSIGN(MockCriticalImagesFinder);
87
} // namespace net_instaweb
89
#endif // NET_INSTAWEB_REWRITER_PUBLIC_MOCK_CRITICAL_IMAGES_FINDER_H_