Page Speed Optimization Libraries  1.8.31.3
net/instaweb/rewriter/public/critical_images_finder_test_base.h
Go to the documentation of this file.
00001 /*
00002  * Copyright 2012 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_CRITICAL_IMAGES_FINDER_TEST_BASE_H_
00019 #define NET_INSTAWEB_REWRITER_PUBLIC_CRITICAL_IMAGES_FINDER_TEST_BASE_H_
00020 
00021 #include "net/instaweb/rewriter/public/critical_images_finder.h"
00022 #include "net/instaweb/rewriter/public/rewrite_test_base.h"
00023 #include "net/instaweb/util/public/null_statistics.h"
00024 #include "net/instaweb/util/public/property_cache.h"
00025 #include "net/instaweb/util/public/string.h"
00026 #include "net/instaweb/util/public/string_util.h"
00027 
00028 namespace net_instaweb {
00029 
00030 class RewriteDriver;
00031 class Statistics;
00032 
00035 class TestCriticalImagesFinder : public CriticalImagesFinder {
00036  public:
00037   TestCriticalImagesFinder(const PropertyCache::Cohort* cohort,
00038                            Statistics* stats)
00039       : CriticalImagesFinder(cohort, stats),
00040         available_(kAvailable) {}
00041   virtual ~TestCriticalImagesFinder();
00042   virtual Availability Available(RewriteDriver* driver) {
00043     return available_;
00044   }
00045   void set_available(Availability available) {
00046     available_ = available;
00047   }
00048   virtual void ComputeCriticalImages(RewriteDriver* driver) {}
00049  private:
00050   Availability available_;
00051 };
00052 
00053 class CriticalImagesFinderTestBase : public RewriteTestBase {
00054  public:
00055   virtual CriticalImagesFinder* finder() = 0;
00056 
00057   virtual bool UpdateCriticalImagesCacheEntry(
00058       const StringSet* critical_images_set,
00059       const StringSet* css_critical_images_set) {
00060     return finder()->UpdateCriticalImagesCacheEntryFromDriver(
00061         critical_images_set, css_critical_images_set, rewrite_driver());
00062   }
00063 
00064   void CheckCriticalImageFinderStats(int hits, int expiries, int not_found);
00065 
00066   bool IsHtmlCriticalImage(const GoogleString& url);
00067   bool IsCssCriticalImage(const GoogleString& url);
00068 
00069  protected:
00070   NullStatistics stats_;
00071 
00073   void ResetDriver();
00074 
00075   const PropertyValue* GetCriticalImagesUpdatedValue();
00076 
00077  private:
00078   static const char kRequestUrl[];
00079 };
00080 
00081 }  
00082 
00083 #endif  ///< NET_INSTAWEB_REWRITER_PUBLIC_CRITICAL_IMAGES_FINDER_TEST_BASE_H_
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Defines