Page Speed Optimization Libraries
1.8.31.2
|
00001 /* 00002 * Copyright 2013 Google Inc. 00003 * 00004 * Licensed under the Apache License, Version 2.0 (the "License"); 00005 * you may not use this file except in compliance with the License. 00006 * You may obtain a copy of the License at 00007 * 00008 * http:///www.apache.org/licenses/LICENSE-2.0 00009 * 00010 * Unless required by applicable law or agreed to in writing, software 00011 * distributed under the License is distributed on an "AS IS" BASIS, 00012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 00013 * See the License for the specific language governing permissions and 00014 * limitations under the License. 00015 */ 00017 00018 #ifndef NET_INSTAWEB_REWRITER_PUBLIC_MOCK_CRITICAL_IMAGES_FINDER_H_ 00019 #define NET_INSTAWEB_REWRITER_PUBLIC_MOCK_CRITICAL_IMAGES_FINDER_H_ 00020 00021 #include "net/instaweb/rewriter/public/critical_images_finder_test_base.h" 00022 #include "net/instaweb/rewriter/public/critical_images_finder.h" 00023 #include "net/instaweb/rewriter/rendered_image.pb.h" 00024 #include "net/instaweb/util/public/property_cache.h" 00025 #include "net/instaweb/util/public/string_util.h" 00026 #include "pagespeed/kernel/base/basictypes.h" 00027 #include "pagespeed/kernel/base/scoped_ptr.h" 00028 00029 namespace net_instaweb { 00030 00031 class RewriteDriver; 00032 class Statistics; 00033 00036 class MockCriticalImagesFinder : public TestCriticalImagesFinder { 00037 public: 00038 explicit MockCriticalImagesFinder(Statistics* stats) 00039 : TestCriticalImagesFinder(NULL, stats), compute_calls_(0) {} 00040 00041 MockCriticalImagesFinder(const PropertyCache::Cohort* cohort, 00042 Statistics* stats) 00043 : TestCriticalImagesFinder(cohort, stats), compute_calls_(0) {} 00044 00045 ~MockCriticalImagesFinder(); 00046 00047 virtual Availability Available(RewriteDriver* driver) { 00048 return kAvailable; 00049 } 00050 00051 virtual void UpdateCriticalImagesSetInDriver(RewriteDriver* driver); 00052 00054 virtual RenderedImages* ExtractRenderedImageDimensionsFromCache( 00055 RewriteDriver* driver); 00056 00057 virtual void ComputeCriticalImages(RewriteDriver* driver) { 00058 ++compute_calls_; 00059 } 00060 00061 int num_compute_calls() { return compute_calls_; } 00062 00063 void set_critical_images(StringSet* critical_images) { 00064 critical_images_.reset(critical_images); 00065 } 00066 00067 void set_css_critical_images(StringSet* css_critical_images) { 00068 css_critical_images_.reset(css_critical_images); 00069 } 00070 00071 void set_rendered_images(RenderedImages* rendered_images) { 00072 rendered_images_.reset(rendered_images); 00073 } 00074 00075 virtual bool IsCriticalImageInfoPresent(RewriteDriver* driver) { 00076 return true; 00077 } 00078 00079 private: 00080 int compute_calls_; 00081 scoped_ptr<StringSet> critical_images_; 00082 scoped_ptr<StringSet> css_critical_images_; 00083 scoped_ptr<RenderedImages> rendered_images_; 00084 DISALLOW_COPY_AND_ASSIGN(MockCriticalImagesFinder); 00085 }; 00086 00087 } 00088 00089 #endif ///< NET_INSTAWEB_REWRITER_PUBLIC_MOCK_CRITICAL_IMAGES_FINDER_H_