blob: 4244ceb193430b234599a7f68b7271bc2df81e76 [file] [log] [blame]
// Copyright 2016 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 <utility>
#include "base/bind.h"
#include "base/files/file_util.h"
#include "base/files/scoped_temp_dir.h"
#include "base/hash/hash.h"
#include "base/logging.h"
#include "base/memory/ptr_util.h"
#include "base/run_loop.h"
#include "base/single_thread_task_runner.h"
#include "base/test/scoped_task_environment.h"
#include "base/threading/thread_task_runner_handle.h"
#include "chromeos/printing/ppd_cache.h"
#include "net/url_request/test_url_request_interceptor.h"
#include "net/url_request/url_request_test_util.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace chromeos {
namespace {
// This fixture just points the cache at a temporary directory for the life of
// the test.
class PpdCacheTest : public ::testing::Test {
public:
PpdCacheTest()
: task_environment_(base::test::TaskEnvironment::MainThreadType::IO) {}
void SetUp() override {
ASSERT_TRUE(ppd_cache_temp_dir_.CreateUniqueTempDir());
}
// Make and return a cache for the test that uses a temporary directory
// which is cleaned up at the end of the test. Note that we pass
// a (nonexistant) subdirectory of temp_dir_ to the cache to exercise
// the lazy-creation-of-the-cache-directory code.
scoped_refptr<PpdCache> CreateTestCache() {
return PpdCache::CreateForTesting(
ppd_cache_temp_dir_.GetPath().Append("Cache"),
task_environment_.GetMainThreadTaskRunner());
}
void CaptureFindResult(const PpdCache::FindResult& result) {
++captured_find_results_;
find_result_ = result;
}
protected:
// Environment for task schedulers.
base::test::TaskEnvironment task_environment_;
// Number of find results we've captured.
int captured_find_results_ = 0;
// Most recent captured result.
PpdCache::FindResult find_result_;
// Overrider for DIR_CHROMEOS_PPD_CACHE that points it at a temporary
// directory for the life of the test.
base::ScopedTempDir ppd_cache_temp_dir_;
};
// Test that we miss on an empty cache.
TEST_F(PpdCacheTest, SimpleMiss) {
auto cache = CreateTestCache();
cache->Find("foo", base::BindOnce(&PpdCacheTest::CaptureFindResult,
base::Unretained(this)));
task_environment_.RunUntilIdle();
EXPECT_EQ(captured_find_results_, 1);
EXPECT_FALSE(find_result_.success);
}
TEST_F(PpdCacheTest, MissThenHit) {
auto cache = CreateTestCache();
const char kTestKey[] = "My totally awesome key";
const char kTestKey2[] = "A different key";
const char kTestContents[] = "Like, totally awesome contents";
cache->Find(kTestKey, base::BindOnce(&PpdCacheTest::CaptureFindResult,
base::Unretained(this)));
task_environment_.RunUntilIdle();
EXPECT_EQ(captured_find_results_, 1);
EXPECT_FALSE(find_result_.success);
cache->Store(kTestKey, kTestContents);
task_environment_.RunUntilIdle();
cache->Find(kTestKey, base::BindOnce(&PpdCacheTest::CaptureFindResult,
base::Unretained(this)));
task_environment_.RunUntilIdle();
EXPECT_EQ(captured_find_results_, 2);
EXPECT_TRUE(find_result_.success);
EXPECT_EQ(find_result_.contents, kTestContents);
EXPECT_LT(find_result_.age, base::TimeDelta::FromMinutes(5));
cache->Find(kTestKey2, base::BindOnce(&PpdCacheTest::CaptureFindResult,
base::Unretained(this)));
task_environment_.RunUntilIdle();
EXPECT_EQ(captured_find_results_, 3);
EXPECT_FALSE(find_result_.success);
}
// Test that we fill in the age field with something plausible.
TEST_F(PpdCacheTest, HitAge) {
auto cache = CreateTestCache();
const char kTestKey[] = "My totally awesome key";
const char kTestContents[] = "Like, totally awesome contents";
cache->Store(kTestKey, kTestContents);
task_environment_.RunUntilIdle();
cache->Find(kTestKey, base::BindOnce(&PpdCacheTest::CaptureFindResult,
base::Unretained(this)));
task_environment_.RunUntilIdle();
EXPECT_EQ(captured_find_results_, 1);
// The age should be well under a second, but accept anything under an hour.
EXPECT_LT(find_result_.age, TimeDelta::FromHours(1));
}
} // namespace
} // namespace chromeos