Page Speed Optimization Libraries  1.7.30.1
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 "net/instaweb/rewriter/public/critical_images_finder_test_base.h"
00022 #include "net/instaweb/rewriter/rendered_image.pb.h"
00023 #include "net/instaweb/util/public/property_cache.h"
00024 #include "net/instaweb/util/public/string_util.h"
00025 #include "pagespeed/kernel/base/basictypes.h"
00026 #include "pagespeed/kernel/base/scoped_ptr.h"
00027 
00028 namespace net_instaweb {
00029 
00030 class RewriteDriver;
00031 class Statistics;
00032 
00035 class MockCriticalImagesFinder : public TestCriticalImagesFinder {
00036  public:
00037   explicit MockCriticalImagesFinder(Statistics* stats)
00038       : TestCriticalImagesFinder(NULL, stats), compute_calls_(0) {}
00039 
00040   MockCriticalImagesFinder(const PropertyCache::Cohort* cohort,
00041                            Statistics* stats)
00042       : TestCriticalImagesFinder(cohort, stats), compute_calls_(0) {}
00043 
00044   ~MockCriticalImagesFinder();
00045 
00046   virtual bool IsMeaningful(const RewriteDriver* driver) const { return true; }
00047 
00048   virtual void UpdateCriticalImagesSetInDriver(RewriteDriver* driver);
00049 
00051   virtual RenderedImages* ExtractRenderedImageDimensionsFromCache(
00052       RewriteDriver* driver);
00053 
00054   virtual void ComputeCriticalImages(RewriteDriver* driver) {
00055     ++compute_calls_;
00056   }
00057 
00058   int num_compute_calls() { return compute_calls_; }
00059 
00060   void set_critical_images(StringSet* critical_images) {
00061     critical_images_.reset(critical_images);
00062   }
00063 
00064   void set_css_critical_images(StringSet* css_critical_images) {
00065     css_critical_images_.reset(css_critical_images);
00066   }
00067 
00068   void set_rendered_images(RenderedImages* rendered_images) {
00069     rendered_images_.reset(rendered_images);
00070   }
00071 
00072   virtual bool IsCriticalImageInfoPresent(RewriteDriver* driver) {
00073     return true;
00074   }
00075 
00076  private:
00077   int compute_calls_;
00078   scoped_ptr<StringSet> critical_images_;
00079   scoped_ptr<StringSet> css_critical_images_;
00080   scoped_ptr<RenderedImages> rendered_images_;
00081   DISALLOW_COPY_AND_ASSIGN(MockCriticalImagesFinder);
00082 };
00083 
00084 }  
00085 
00086 #endif  ///< NET_INSTAWEB_REWRITER_PUBLIC_MOCK_CRITICAL_IMAGES_FINDER_H_
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Defines