webkit/browser/appcache/appcache_storage_unittest.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "base/message_loop/message_loop.h"
   6 #include "testing/gtest/include/gtest/gtest.h"
   7 #include "webkit/browser/appcache/appcache.h"
   8 #include "webkit/browser/appcache/appcache_group.h"
   9 #include "webkit/browser/appcache/appcache_response.h"
  10 #include "webkit/browser/appcache/appcache_storage.h"
  11 #include "webkit/browser/appcache/mock_appcache_service.h"
  12 #include "webkit/browser/quota/mock_quota_manager.h"
  13
  14 namespace appcache {
  15
  16 namespace {
  17 const quota::StorageType kTemp = quota::kStorageTypeTemporary;
  18 }
  19
  20 class AppCacheStorageTest : public testing::Test {
  21  public:
  22   class MockStorageDelegate : public AppCacheStorage::Delegate {
  23    public:
  24   };
  25 };
  26
  27 TEST_F(AppCacheStorageTest, AddRemoveCache) {
  28   MockAppCacheService service;
  29   scoped_refptr<AppCache> cache(new AppCache(service.storage(), 111));
  30
  31   EXPECT_EQ(cache.get(),
  32             service.storage()->working_set()->GetCache(111));
  33
  34   service.storage()->working_set()->RemoveCache(cache.get());
  35
  36   EXPECT_TRUE(!service.storage()->working_set()->GetCache(111));
  37
  38   // Removing non-existing cache from service should not fail.
  39   MockAppCacheService dummy;
  40   dummy.storage()->working_set()->RemoveCache(cache.get());
  41 }
  42
  43 TEST_F(AppCacheStorageTest, AddRemoveGroup) {
  44   MockAppCacheService service;
  45   scoped_refptr<AppCacheGroup> group(
  46       new AppCacheGroup(service.storage(), GURL(), 111));
  47
  48   EXPECT_EQ(group.get(), service.storage()->working_set()->GetGroup(GURL()));
  49
  50   service.storage()->working_set()->RemoveGroup(group.get());
  51
  52   EXPECT_TRUE(!service.storage()->working_set()->GetGroup(GURL()));
  53
  54   // Removing non-existing group from service should not fail.
  55   MockAppCacheService dummy;
  56   dummy.storage()->working_set()->RemoveGroup(group.get());
  57 }
  58
  59 TEST_F(AppCacheStorageTest, AddRemoveResponseInfo) {
  60   MockAppCacheService service;
  61   scoped_refptr<AppCacheResponseInfo> info(
  62       new AppCacheResponseInfo(&service, GURL(),
  63                                111, new net::HttpResponseInfo,
  64                                kUnkownResponseDataSize));
  65
  66   EXPECT_EQ(info.get(),
  67             service.storage()->working_set()->GetResponseInfo(111));
  68
  69   service.storage()->working_set()->RemoveResponseInfo(info.get());
  70
  71   EXPECT_TRUE(!service.storage()->working_set()->GetResponseInfo(111));
  72
  73   // Removing non-existing info from service should not fail.
  74   MockAppCacheService dummy;
  75   dummy.storage()->working_set()->RemoveResponseInfo(info.get());
  76 }
  77
  78 TEST_F(AppCacheStorageTest, DelegateReferences) {
  79   typedef scoped_refptr<AppCacheStorage::DelegateReference>
  80       ScopedDelegateReference;
  81   MockAppCacheService service;
  82   MockStorageDelegate delegate;
  83   ScopedDelegateReference delegate_reference1;
  84   ScopedDelegateReference delegate_reference2;
  85
  86   EXPECT_FALSE(service.storage()->GetDelegateReference(&delegate));
  87
  88   delegate_reference1 =
  89       service.storage()->GetOrCreateDelegateReference(&delegate);
  90   EXPECT_TRUE(delegate_reference1.get());
  91   EXPECT_TRUE(delegate_reference1->HasOneRef());
  92   EXPECT_TRUE(service.storage()->GetDelegateReference(&delegate));
  93   EXPECT_EQ(&delegate,
  94             service.storage()->GetDelegateReference(&delegate)->delegate);
  95   EXPECT_EQ(service.storage()->GetDelegateReference(&delegate),
  96             service.storage()->GetOrCreateDelegateReference(&delegate));
  97   delegate_reference1 = NULL;
  98   EXPECT_FALSE(service.storage()->GetDelegateReference(&delegate));
  99
 100   delegate_reference1 =
 101       service.storage()->GetOrCreateDelegateReference(&delegate);
 102   service.storage()->CancelDelegateCallbacks(&delegate);
 103   EXPECT_TRUE(delegate_reference1.get());
 104   EXPECT_TRUE(delegate_reference1->HasOneRef());
 105   EXPECT_FALSE(delegate_reference1->delegate);
 106   EXPECT_FALSE(service.storage()->GetDelegateReference(&delegate));
 107
 108   delegate_reference2 =
 109       service.storage()->GetOrCreateDelegateReference(&delegate);
 110   EXPECT_TRUE(delegate_reference2.get());
 111   EXPECT_TRUE(delegate_reference2->HasOneRef());
 112   EXPECT_EQ(&delegate, delegate_reference2->delegate);
 113   EXPECT_NE(delegate_reference1.get(), delegate_reference2.get());
 114 }
 115
 116 TEST_F(AppCacheStorageTest, UsageMap) {
 117   const GURL kOrigin("http://origin/");
 118   const GURL kOrigin2("http://origin2/");
 119
 120   MockAppCacheService service;
 121   scoped_refptr<quota::MockQuotaManagerProxy> mock_proxy(
 122       new quota::MockQuotaManagerProxy(NULL, NULL));
 123   service.set_quota_manager_proxy(mock_proxy.get());
 124
 125   service.storage()->UpdateUsageMapAndNotify(kOrigin, 0);
 126   EXPECT_EQ(0, mock_proxy->notify_storage_modified_count());
 127
 128   service.storage()->UpdateUsageMapAndNotify(kOrigin, 10);
 129   EXPECT_EQ(1, mock_proxy->notify_storage_modified_count());
 130   EXPECT_EQ(10, mock_proxy->last_notified_delta());
 131   EXPECT_EQ(kOrigin, mock_proxy->last_notified_origin());
 132   EXPECT_EQ(kTemp, mock_proxy->last_notified_type());
 133
 134   service.storage()->UpdateUsageMapAndNotify(kOrigin, 100);
 135   EXPECT_EQ(2, mock_proxy->notify_storage_modified_count());
 136   EXPECT_EQ(90, mock_proxy->last_notified_delta());
 137   EXPECT_EQ(kOrigin, mock_proxy->last_notified_origin());
 138   EXPECT_EQ(kTemp, mock_proxy->last_notified_type());
 139
 140   service.storage()->UpdateUsageMapAndNotify(kOrigin, 0);
 141   EXPECT_EQ(3, mock_proxy->notify_storage_modified_count());
 142   EXPECT_EQ(-100, mock_proxy->last_notified_delta());
 143   EXPECT_EQ(kOrigin, mock_proxy->last_notified_origin());
 144   EXPECT_EQ(kTemp, mock_proxy->last_notified_type());
 145
 146   service.storage()->NotifyStorageAccessed(kOrigin2);
 147   EXPECT_EQ(0, mock_proxy->notify_storage_accessed_count());
 148
 149   service.storage()->usage_map_[kOrigin2] = 1;
 150   service.storage()->NotifyStorageAccessed(kOrigin2);
 151   EXPECT_EQ(1, mock_proxy->notify_storage_accessed_count());
 152   EXPECT_EQ(kOrigin2, mock_proxy->last_notified_origin());
 153   EXPECT_EQ(kTemp, mock_proxy->last_notified_type());
 154
 155   service.storage()->usage_map_.clear();
 156   service.storage()->usage_map_[kOrigin] = 5000;
 157   service.storage()->ClearUsageMapAndNotify();
 158   EXPECT_EQ(4, mock_proxy->notify_storage_modified_count());
 159   EXPECT_EQ(-5000, mock_proxy->last_notified_delta());
 160   EXPECT_EQ(kOrigin, mock_proxy->last_notified_origin());
 161   EXPECT_EQ(kTemp, mock_proxy->last_notified_type());
 162   EXPECT_TRUE(service.storage()->usage_map_.empty());
 163 }
 164
 165 }  // namespace appcache