Sparse IO: Allow failover to network in debug builds

net/http/mock_http_cache.h

 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 // This is a mock of the http cache and related testing classes. To be fair, it
 // is not really a mock http cache given that it uses the real implementation of
 // the http cache, but it has fake implementations of all required components,
 // so it is useful for unit tests at the http layer.
 
 #ifndef NET_HTTP_MOCK_HTTP_CACHE_H_
 #define NET_HTTP_MOCK_HTTP_CACHE_H_
 
 #include "base/containers/hash_tables.h"
 #include "net/disk_cache/disk_cache.h"
 #include "net/http/http_cache.h"
 #include "net/http/http_transaction_unittest.h"
 
 //-----------------------------------------------------------------------------
 // Mock disk cache (a very basic memory cache implementation).
 
+class MockDiskCache;
+
 class MockDiskEntry : public disk_cache::Entry,
                       public base::RefCounted<MockDiskEntry> {
  public:
-  MockDiskEntry();
-  explicit MockDiskEntry(const std::string& key);
+  MockDiskEntry(MockDiskCache* backend, const std::string& key);
 
   bool is_doomed() const { return doomed_; }
 
   virtual void Doom() OVERRIDE;
   virtual void Close() OVERRIDE;
   virtual std::string GetKey() const OVERRIDE;
   virtual base::Time GetLastUsed() const OVERRIDE;
   virtual base::Time GetLastModified() const OVERRIDE;
   virtual int32 GetDataSize(int index) const OVERRIDE;
   virtual int ReadData(int index, int offset, net::IOBuffer* buf, int buf_len,
@@ skipping 10 matching lines @@
       int64 offset, int len, int64* start,
       const net::CompletionCallback& callback) OVERRIDE;
   virtual bool CouldBeSparse() const OVERRIDE;
   virtual void CancelSparseIO() OVERRIDE;
   virtual int ReadyForSparseIO(
       const net::CompletionCallback& completion_callback) OVERRIDE;
 
   // Fail most subsequent requests.
   void set_fail_requests() { fail_requests_ = true; }
 
+  void set_fail_sparse_requests() { fail_sparse_requests_ = true; }
+
   // If |value| is true, don't deliver any completion callbacks until called
   // again with |value| set to false.  Caution: remember to enable callbacks
   // again or all subsequent tests will fail.
   static void IgnoreCallbacks(bool value);
 
  private:
   friend class base::RefCounted<MockDiskEntry>;
   struct CallbackInfo;
 
   virtual ~MockDiskEntry();
@@ skipping 12 matching lines @@
                                        int result);
 
   static const int kNumCacheEntryDataIndices = 3;
 
   std::string key_;
   std::vector<char> data_[kNumCacheEntryDataIndices];
   int test_mode_;
   bool doomed_;
   bool sparse_;
   bool fail_requests_;
+  bool fail_sparse_requests_;
   bool busy_;
   bool delayed_;
   static bool cancel_;
   static bool ignore_callbacks_;
+  MockDiskCache* backend_;
 };
 
 class MockDiskCache : public disk_cache::Backend {
  public:
   MockDiskCache();
   virtual ~MockDiskCache();
 
   virtual net::CacheType GetCacheType() const OVERRIDE;
   virtual int32 GetEntryCount() const OVERRIDE;
   virtual int OpenEntry(const std::string& key, disk_cache::Entry** entry,
@@ skipping 16 matching lines @@
   virtual void GetStats(
       std::vector<std::pair<std::string, std::string> >* stats) OVERRIDE;
   virtual void OnExternalCacheHit(const std::string& key) OVERRIDE;
 
   // Returns number of times a cache entry was successfully opened.
   int open_count() const { return open_count_; }
 
   // Returns number of times a cache entry was successfully created.
   int create_count() const { return create_count_; }
 
+  // Returns the number of operations attempted that are specific to range
+  // requests.
+  int sparse_count() const { return sparse_count_; }
+
   // Fail any subsequent CreateEntry and OpenEntry.
   void set_fail_requests() { fail_requests_ = true; }
 
   // Return entries that fail some of their requests.
   void set_soft_failures(bool value) { soft_failures_ = value; }
 
   // Makes sure that CreateEntry is not called twice for a given key.
   void set_double_create_check(bool value) { double_create_check_ = value; }
 
+  // Makes all requests for data ranges to fail as not implemented.
+  void set_fail_sparse_requests() { fail_sparse_requests_ = true; }
+
+  void NotifySparseIO() { sparse_count_++; }
+
   void ReleaseAll();
 
  private:
   typedef base::hash_map<std::string, MockDiskEntry*> EntryMap;
 
   void CallbackLater(const net::CompletionCallback& callback, int result);
 
   EntryMap entries_;
   int open_count_;
   int create_count_;
+  int sparse_count_;
   bool fail_requests_;
   bool soft_failures_;
   bool double_create_check_;
+  bool fail_sparse_requests_;
 };
 
 class MockBackendFactory : public net::HttpCache::BackendFactory {
  public:
   virtual int CreateBackend(net::NetLog* net_log,
                             scoped_ptr<disk_cache::Backend>* backend,
                             const net::CompletionCallback& callback) OVERRIDE;
 };
 
 class MockHttpCache {
@@ skipping 72 matching lines @@
  private:
   int Result() { return fail_ ? net::ERR_FAILED : net::OK; }
 
   scoped_ptr<disk_cache::Backend>* backend_;
   net::CompletionCallback callback_;
   bool block_;
   bool fail_;
 };
 
 #endif  // NET_HTTP_MOCK_HTTP_CACHE_H_