Page Speed Optimization Libraries
1.6.29.3
|
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 "base/logging.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 CriticalImagesFinder { 00037 public: 00038 explicit MockCriticalImagesFinder(Statistics* stats) 00039 : CriticalImagesFinder(stats), 00040 compute_calls_(0) {} 00041 00042 ~MockCriticalImagesFinder(); 00043 00044 virtual bool IsMeaningful(const RewriteDriver* driver) const { return true; } 00045 00046 virtual void UpdateCriticalImagesSetInDriver(RewriteDriver* driver); 00047 00049 virtual RenderedImages* ExtractRenderedImageDimensionsFromCache( 00050 RewriteDriver* driver); 00051 00052 virtual void ComputeCriticalImages(RewriteDriver* driver) { 00053 ++compute_calls_; 00054 } 00055 00056 virtual const PropertyCache::Cohort* GetCriticalImagesCohort() const { 00059 LOG(DFATAL) << "Unexpected function call!!!"; 00060 return NULL; 00061 } 00062 00063 int num_compute_calls() { return compute_calls_; } 00064 00065 void set_critical_images(StringSet* critical_images) { 00066 critical_images_.reset(critical_images); 00067 } 00068 00069 void set_css_critical_images(StringSet* css_critical_images) { 00070 css_critical_images_.reset(css_critical_images); 00071 } 00072 00073 void set_rendered_images(RenderedImages* rendered_images) { 00074 rendered_images_.reset(rendered_images); 00075 } 00076 00077 virtual bool IsCriticalImageInfoPresent(RewriteDriver* driver) { 00078 return true; 00079 } 00080 00081 private: 00082 int compute_calls_; 00083 scoped_ptr<StringSet> critical_images_; 00084 scoped_ptr<StringSet> css_critical_images_; 00085 scoped_ptr<RenderedImages> rendered_images_; 00086 DISALLOW_COPY_AND_ASSIGN(MockCriticalImagesFinder); 00087 }; 00088 00092 class ForwardingMockCriticalImagesFinder : public MockCriticalImagesFinder { 00093 public: 00094 explicit ForwardingMockCriticalImagesFinder(Statistics* stats) : 00095 MockCriticalImagesFinder(stats) {} 00096 virtual void UpdateCriticalImagesSetInDriver(RewriteDriver* driver) { 00097 CriticalImagesFinder::UpdateCriticalImagesSetInDriver(driver); 00098 } 00099 virtual const PropertyCache::Cohort* GetCriticalImagesCohort() const { 00100 return NULL; 00101 } 00102 virtual bool IsCriticalImageInfoPresent(RewriteDriver* driver) { 00103 return CriticalImagesFinder::IsCriticalImageInfoPresent(driver); 00104 } 00105 }; 00106 00107 } 00108 00109 #endif ///< NET_INSTAWEB_REWRITER_PUBLIC_MOCK_CRITICAL_IMAGES_FINDER_H_