1 files changed, 23 insertions, 9 deletions
diff --git a/net/base/cookie_monster_store_test.h b/net/base/cookie_monster_store_test.h
index 1ee757d..4bae6f9 100644
--- a/net/base/cookie_monster_store_test.h
+++ b/net/base/cookie_monster_store_test.h
@@ -7,6 +7,14 @@
 // that need to test out CookieMonster interactions with the backing store.
 // It should only be included by test code.
 
+#ifndef NET_BASE_COOKIE_MONSTER_STORE_TEST_H_
+#define NET_BASE_COOKIE_MONSTER_STORE_TEST_H_
+#pragma once
+
+#include <map>
+#include <string>
+#include <utility>
+#include <vector>
 #include "net/base/cookie_monster.h"
 
 namespace base {
@@ -51,21 +59,20 @@ class MockPersistentCookieStore
     return commands_;
   }
 
-  virtual bool Load(
-      std::vector<CookieMonster::CanonicalCookie*>* out_cookies);
+  virtual bool Load(const LoadedCallback& loaded_callback) OVERRIDE;
 
-  virtual void AddCookie(const CookieMonster::CanonicalCookie& cookie);
+  virtual void AddCookie(const CookieMonster::CanonicalCookie& cookie) OVERRIDE;
 
   virtual void UpdateCookieAccessTime(
-      const CookieMonster::CanonicalCookie& cookie);
+      const CookieMonster::CanonicalCookie& cookie) OVERRIDE;
 
   virtual void DeleteCookie(
-      const CookieMonster::CanonicalCookie& cookie);
+      const CookieMonster::CanonicalCookie& cookie) OVERRIDE;
 
-  virtual void Flush(Task* completion_task);
+  virtual void Flush(Task* completion_task) OVERRIDE;
 
   // No files are created so nothing to clear either
-  virtual void SetClearLocalStateOnExit(bool clear_local_state);
+  virtual void SetClearLocalStateOnExit(bool clear_local_state) OVERRIDE;
 
  private:
   CommandList commands_;
@@ -102,6 +109,12 @@ class MockCookieMonsterDelegate : public CookieMonster::Delegate {
   DISALLOW_COPY_AND_ASSIGN(MockCookieMonsterDelegate);
 };
 
+// Helper to build a single CanonicalCookie.
+CookieMonster::CanonicalCookie BuildCanonicalCookie(
+    const std::string& key,
+    const std::string& cookie_line,
+    const base::Time& creation_time);
+
 // Helper to build a list of CanonicalCookie*s.
 void AddCookieToList(
     const std::string& key,
@@ -117,8 +130,7 @@ class MockSimplePersistentCookieStore
   MockSimplePersistentCookieStore();
   virtual ~MockSimplePersistentCookieStore();
 
-  virtual bool Load(
-      std::vector<CookieMonster::CanonicalCookie*>* out_cookies);
+  virtual bool Load(const LoadedCallback& loaded_callback);
 
   virtual void AddCookie(
       const CookieMonster::CanonicalCookie& cookie);
@@ -154,3 +166,5 @@ CookieMonster* CreateMonsterFromStoreForGC(
     int days_old);
 
 }  // namespace net
+
+#endif  // NET_BASE_COOKIE_MONSTER_STORE_TEST_H_