text: // Copyright (c) 2010 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.
  
  #include "base/message_loop.h"
  #include "testing/gtest/include/gtest/gtest.h"
  #include "webkit/appcache/appcache.h"
  #include "webkit/appcache/appcache_group.h"
  #include "webkit/appcache/appcache_response.h"
  #include "webkit/appcache/appcache_storage.h"
  #include "webkit/appcache/mock_appcache_service.h"
  
  namespace appcache {
  
  class AppCacheStorageTest : public testing::Test {
   public:
    class MockStorageDelegate : public AppCacheStorage::Delegate {
     public:
    };
  
    class MockQuotaManagerProxy : public quota::QuotaManagerProxy {
     public:
      MockQuotaManagerProxy()
          : QuotaManagerProxy(NULL, NULL),
            notify_storage_accessed_count_(0),
            notify_storage_modified_count_(0),
            last_delta_(0) {}
  
      virtual void NotifyStorageAccessed(quota::QuotaClient::ID client_id,
                                         const GURL& origin,
                                         quota::StorageType type) {
        EXPECT_EQ(quota::QuotaClient::kAppcache, client_id);
        EXPECT_EQ(quota::kStorageTypeTemporary, type);
        ++notify_storage_accessed_count_;
        last_origin_ = origin;
      }
  
      virtual void NotifyStorageModified(quota::QuotaClient::ID client_id,
                                         const GURL& origin,
                                         quota::StorageType type,
                                         int64 delta) {
        EXPECT_EQ(quota::QuotaClient::kAppcache, client_id);
        EXPECT_EQ(quota::kStorageTypeTemporary, type);
        ++notify_storage_modified_count_;
        last_origin_ = origin;
        last_delta_ = delta;
      }
  
      // Not needed for our tests.
      virtual void RegisterClient(quota::QuotaClient* client) {}
      virtual void NotifyOriginInUse(const GURL& origin) {}
      virtual void NotifyOriginNoLongerInUse(const GURL& origin) {}
  
      int notify_storage_accessed_count_;
      int notify_storage_modified_count_;
      GURL last_origin_;
      int last_delta_;
    };
  };
  
  TEST_F(AppCacheStorageTest, AddRemoveCache) {
    MockAppCacheService service;
    scoped_refptr<AppCache> cache(new AppCache(&service, 111));
  
    EXPECT_EQ(cache.get(),
              service.storage()->working_set()->GetCache(111));
  
    service.storage()->working_set()->RemoveCache(cache);
  
    EXPECT_TRUE(!service.storage()->working_set()->GetCache(111));
  
    // Removing non-existing cache from service should not fail.
    MockAppCacheService dummy;
    dummy.storage()->working_set()->RemoveCache(cache);
  }
  
  TEST_F(AppCacheStorageTest, AddRemoveGroup) {
    MockAppCacheService service;
    scoped_refptr<AppCacheGroup> group(new AppCacheGroup(&service, GURL(), 111));
  
    EXPECT_EQ(group.get(), service.storage()->working_set()->GetGroup(GURL()));
  
    service.storage()->working_set()->RemoveGroup(group);
  
    EXPECT_TRUE(!service.storage()->working_set()->GetGroup(GURL()));
  
    // Removing non-existing group from service should not fail.
    MockAppCacheService dummy;
    dummy.storage()->working_set()->RemoveGroup(group);
  }
  
  TEST_F(AppCacheStorageTest, AddRemoveResponseInfo) {
    MockAppCacheService service;
    scoped_refptr<AppCacheResponseInfo> info(
        new AppCacheResponseInfo(&service, GURL(),
                                 111, new net::HttpResponseInfo,
                                 kUnkownResponseDataSize));
  
    EXPECT_EQ(info.get(),
              service.storage()->working_set()->GetResponseInfo(111));
  
    service.storage()->working_set()->RemoveResponseInfo(info);
  
    EXPECT_TRUE(!service.storage()->working_set()->GetResponseInfo(111));
  
    // Removing non-existing info from service should not fail.
    MockAppCacheService dummy;
    dummy.storage()->working_set()->RemoveResponseInfo(info);
  }
  
  TEST_F(AppCacheStorageTest, DelegateReferences) {
    typedef scoped_refptr<AppCacheStorage::DelegateReference>
        ScopedDelegateReference;
    MockAppCacheService service;
    MockStorageDelegate delegate;
    ScopedDelegateReference delegate_reference1;
    ScopedDelegateReference delegate_reference2;
  
    EXPECT_FALSE(service.storage()->GetDelegateReference(&delegate));
  
    delegate_reference1 =
        service.storage()->GetOrCreateDelegateReference(&delegate);
    EXPECT_TRUE(delegate_reference1.get());
    EXPECT_TRUE(delegate_reference1->HasOneRef());
    EXPECT_TRUE(service.storage()->GetDelegateReference(&delegate));
    EXPECT_EQ(&delegate,
              service.storage()->GetDelegateReference(&delegate)->delegate);
    EXPECT_EQ(service.storage()->GetDelegateReference(&delegate),
              service.storage()->GetOrCreateDelegateReference(&delegate));
    delegate_reference1 = NULL;
    EXPECT_FALSE(service.storage()->GetDelegateReference(&delegate));
  
    delegate_reference1 =
        service.storage()->GetOrCreateDelegateReference(&delegate);
    service.storage()->CancelDelegateCallbacks(&delegate);
    EXPECT_TRUE(delegate_reference1.get());
    EXPECT_TRUE(delegate_reference1->HasOneRef());
    EXPECT_FALSE(delegate_reference1->delegate);
    EXPECT_FALSE(service.storage()->GetDelegateReference(&delegate));
  
    delegate_reference2 =
        service.storage()->GetOrCreateDelegateReference(&delegate);
    EXPECT_TRUE(delegate_reference2.get());
    EXPECT_TRUE(delegate_reference2->HasOneRef());
    EXPECT_EQ(&delegate, delegate_reference2->delegate);
    EXPECT_NE(delegate_reference1.get(), delegate_reference2.get());
  }
  
  TEST_F(AppCacheStorageTest, UsageMap) {
    const GURL kOrigin("http://origin/");
    const GURL kOrigin2("http://origin2/");
  
    MockAppCacheService service;
    scoped_refptr<MockQuotaManagerProxy> mock_proxy(new MockQuotaManagerProxy);
    service.set_quota_manager_proxy(mock_proxy);
  
    service.storage()->UpdateUsageMapAndNotify(kOrigin, 0);
    EXPECT_EQ(0, mock_proxy->notify_storage_modified_count_);
  
    service.storage()->UpdateUsageMapAndNotify(kOrigin, 10);
    EXPECT_EQ(1, mock_proxy->notify_storage_modified_count_);
    EXPECT_EQ(10, mock_proxy->last_delta_);
    EXPECT_EQ(kOrigin, mock_proxy->last_origin_);
  
    service.storage()->UpdateUsageMapAndNotify(kOrigin, 100);
    EXPECT_EQ(2, mock_proxy->notify_storage_modified_count_);
    EXPECT_EQ(90, mock_proxy->last_delta_);
    EXPECT_EQ(kOrigin, mock_proxy->last_origin_);
  
    service.storage()->UpdateUsageMapAndNotify(kOrigin, 0);
    EXPECT_EQ(3, mock_proxy->notify_storage_modified_count_);
    EXPECT_EQ(-100, mock_proxy->last_delta_);
    EXPECT_EQ(kOrigin, mock_proxy->last_origin_);
  
    service.storage()->NotifyStorageAccessed(kOrigin2);
    EXPECT_EQ(0, mock_proxy->notify_storage_accessed_count_);
  
    service.storage()->usage_map_[kOrigin2] = 1;
    service.storage()->NotifyStorageAccessed(kOrigin2);
    EXPECT_EQ(1, mock_proxy->notify_storage_accessed_count_);
    EXPECT_EQ(kOrigin2, mock_proxy->last_origin_);
  
    service.storage()->usage_map_.clear();
    service.storage()->usage_map_[kOrigin] = 5000;
    service.storage()->ClearUsageMapAndNotify();
    EXPECT_EQ(4, mock_proxy->notify_storage_modified_count_);
    EXPECT_EQ(-5000, mock_proxy->last_delta_);
    EXPECT_EQ(kOrigin, mock_proxy->last_origin_);
    EXPECT_TRUE(service.storage()->usage_map_.empty());
  }
  
  }  // namespace appcache
  