blob: 98bf68aa8ec33766637ce6e561586885181dccd2 [file] [log] [blame]
// Copyright (c) 2012 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/bind.h"
#include "base/file_util.h"
#include "base/files/scoped_temp_dir.h"
#include "base/memory/ref_counted.h"
#include "base/message_loop/message_loop.h"
#include "content/browser/appcache/appcache_database.h"
#include "content/browser/appcache/appcache_storage_impl.h"
#include "content/browser/appcache/chrome_appcache_service.h"
#include "content/browser/browser_thread_impl.h"
#include "content/public/browser/resource_context.h"
#include "content/public/test/mock_special_storage_policy.h"
#include "content/public/test/test_browser_context.h"
#include "content/test/appcache_test_helper.h"
#include "net/url_request/url_request_context_getter.h"
#include "testing/gtest/include/gtest/gtest.h"
#include <set>
namespace content {
namespace {
const base::FilePath::CharType kTestingAppCacheDirname[] =
FILE_PATH_LITERAL("Application Cache");
// Examples of a protected and an unprotected origin, to be used througout the
// test.
const char kProtectedManifest[] = "http://www.protected.com/cache.manifest";
const char kNormalManifest[] = "http://www.normal.com/cache.manifest";
const char kSessionOnlyManifest[] = "http://www.sessiononly.com/cache.manifest";
class MockURLRequestContextGetter : public net::URLRequestContextGetter {
public:
MockURLRequestContextGetter(
net::URLRequestContext* context,
base::MessageLoopProxy* message_loop_proxy)
: context_(context), message_loop_proxy_(message_loop_proxy) {
}
virtual net::URLRequestContext* GetURLRequestContext() OVERRIDE {
return context_;
}
virtual scoped_refptr<base::SingleThreadTaskRunner>
GetNetworkTaskRunner() const OVERRIDE {
return message_loop_proxy_;
}
protected:
virtual ~MockURLRequestContextGetter() {}
private:
net::URLRequestContext* context_;
scoped_refptr<base::SingleThreadTaskRunner> message_loop_proxy_;
};
} // namespace
class ChromeAppCacheServiceTest : public testing::Test {
public:
ChromeAppCacheServiceTest()
: kProtectedManifestURL(kProtectedManifest),
kNormalManifestURL(kNormalManifest),
kSessionOnlyManifestURL(kSessionOnlyManifest),
file_thread_(BrowserThread::FILE, &message_loop_),
file_user_blocking_thread_(BrowserThread::FILE_USER_BLOCKING,
&message_loop_),
cache_thread_(BrowserThread::CACHE, &message_loop_),
io_thread_(BrowserThread::IO, &message_loop_) {}
protected:
scoped_refptr<ChromeAppCacheService> CreateAppCacheServiceImpl(
const base::FilePath& appcache_path,
bool init_storage);
void InsertDataIntoAppCache(ChromeAppCacheService* appcache_service);
base::MessageLoopForIO message_loop_;
base::ScopedTempDir temp_dir_;
const GURL kProtectedManifestURL;
const GURL kNormalManifestURL;
const GURL kSessionOnlyManifestURL;
private:
BrowserThreadImpl file_thread_;
BrowserThreadImpl file_user_blocking_thread_;
BrowserThreadImpl cache_thread_;
BrowserThreadImpl io_thread_;
TestBrowserContext browser_context_;
};
scoped_refptr<ChromeAppCacheService>
ChromeAppCacheServiceTest::CreateAppCacheServiceImpl(
const base::FilePath& appcache_path,
bool init_storage) {
scoped_refptr<ChromeAppCacheService> appcache_service =
new ChromeAppCacheService(NULL);
scoped_refptr<MockSpecialStoragePolicy> mock_policy =
new MockSpecialStoragePolicy;
mock_policy->AddProtected(kProtectedManifestURL.GetOrigin());
mock_policy->AddSessionOnly(kSessionOnlyManifestURL.GetOrigin());
scoped_refptr<MockURLRequestContextGetter> mock_request_context_getter =
new MockURLRequestContextGetter(
browser_context_.GetResourceContext()->GetRequestContext(),
message_loop_.message_loop_proxy().get());
BrowserThread::PostTask(
BrowserThread::IO,
FROM_HERE,
base::Bind(&ChromeAppCacheService::InitializeOnIOThread,
appcache_service.get(),
appcache_path,
browser_context_.GetResourceContext(),
mock_request_context_getter,
mock_policy));
// Steps needed to initialize the storage of AppCache data.
message_loop_.RunUntilIdle();
if (init_storage) {
AppCacheStorageImpl* storage =
static_cast<AppCacheStorageImpl*>(
appcache_service->storage());
storage->database_->db_connection();
storage->disk_cache();
message_loop_.RunUntilIdle();
}
return appcache_service;
}
void ChromeAppCacheServiceTest::InsertDataIntoAppCache(
ChromeAppCacheService* appcache_service) {
AppCacheTestHelper appcache_helper;
appcache_helper.AddGroupAndCache(appcache_service, kNormalManifestURL);
appcache_helper.AddGroupAndCache(appcache_service, kProtectedManifestURL);
appcache_helper.AddGroupAndCache(appcache_service, kSessionOnlyManifestURL);
// Verify that adding the data succeeded
std::set<GURL> origins;
appcache_helper.GetOriginsWithCaches(appcache_service, &origins);
ASSERT_EQ(3UL, origins.size());
ASSERT_TRUE(origins.find(kProtectedManifestURL.GetOrigin()) != origins.end());
ASSERT_TRUE(origins.find(kNormalManifestURL.GetOrigin()) != origins.end());
ASSERT_TRUE(origins.find(kSessionOnlyManifestURL.GetOrigin()) !=
origins.end());
}
TEST_F(ChromeAppCacheServiceTest, KeepOnDestruction) {
ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
base::FilePath appcache_path =
temp_dir_.path().Append(kTestingAppCacheDirname);
// Create a ChromeAppCacheService and insert data into it
scoped_refptr<ChromeAppCacheService> appcache_service =
CreateAppCacheServiceImpl(appcache_path, true);
ASSERT_TRUE(base::PathExists(appcache_path));
ASSERT_TRUE(base::PathExists(appcache_path.AppendASCII("Index")));
InsertDataIntoAppCache(appcache_service.get());
// Test: delete the ChromeAppCacheService
appcache_service = NULL;
message_loop_.RunUntilIdle();
// Recreate the appcache (for reading the data back)
appcache_service = CreateAppCacheServiceImpl(appcache_path, false);
// The directory is still there
ASSERT_TRUE(base::PathExists(appcache_path));
// The appcache data is also there, except the session-only origin.
AppCacheTestHelper appcache_helper;
std::set<GURL> origins;
appcache_helper.GetOriginsWithCaches(appcache_service.get(), &origins);
EXPECT_EQ(2UL, origins.size());
EXPECT_TRUE(origins.find(kProtectedManifestURL.GetOrigin()) != origins.end());
EXPECT_TRUE(origins.find(kNormalManifestURL.GetOrigin()) != origins.end());
EXPECT_TRUE(origins.find(kSessionOnlyManifestURL.GetOrigin()) ==
origins.end());
// Delete and let cleanup tasks run prior to returning.
appcache_service = NULL;
message_loop_.RunUntilIdle();
}
TEST_F(ChromeAppCacheServiceTest, SaveSessionState) {
ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
base::FilePath appcache_path =
temp_dir_.path().Append(kTestingAppCacheDirname);
// Create a ChromeAppCacheService and insert data into it
scoped_refptr<ChromeAppCacheService> appcache_service =
CreateAppCacheServiceImpl(appcache_path, true);
ASSERT_TRUE(base::PathExists(appcache_path));
ASSERT_TRUE(base::PathExists(appcache_path.AppendASCII("Index")));
InsertDataIntoAppCache(appcache_service.get());
// Save session state. This should bypass the destruction-time deletion.
appcache_service->set_force_keep_session_state();
// Test: delete the ChromeAppCacheService
appcache_service = NULL;
message_loop_.RunUntilIdle();
// Recreate the appcache (for reading the data back)
appcache_service = CreateAppCacheServiceImpl(appcache_path, false);
// The directory is still there
ASSERT_TRUE(base::PathExists(appcache_path));
// No appcache data was deleted.
AppCacheTestHelper appcache_helper;
std::set<GURL> origins;
appcache_helper.GetOriginsWithCaches(appcache_service.get(), &origins);
EXPECT_EQ(3UL, origins.size());
EXPECT_TRUE(origins.find(kProtectedManifestURL.GetOrigin()) != origins.end());
EXPECT_TRUE(origins.find(kNormalManifestURL.GetOrigin()) != origins.end());
EXPECT_TRUE(origins.find(kSessionOnlyManifestURL.GetOrigin()) !=
origins.end());
// Delete and let cleanup tasks run prior to returning.
appcache_service = NULL;
message_loop_.RunUntilIdle();
}
} // namespace content