Page Speed Optimization Libraries  1.6.29.3
net/instaweb/rewriter/public/mock_critical_images_finder.h
Go to the documentation of this file.
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_
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Defines