Page Speed Optimization Libraries  1.8.31.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 "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_
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Defines