1 // Copyright (c) 2011 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/basictypes.h"
6 #include "base/files/file_path.h"
7 #include "base/strings/string_util.h"
8 #include "net/disk_cache/disk_cache_test_base.h"
9 #include "net/disk_cache/disk_cache_test_util.h"
10 #include "net/disk_cache/file_block.h"
11 #include "net/disk_cache/mapped_file.h"
12 #include "testing/gtest/include/gtest/gtest.h"
13
14 namespace {
15
16 // Implementation of FileIOCallback for the tests.
17 class FileCallbackTest: public disk_cache::FileIOCallback {
18 public:
FileCallbackTest(int id,MessageLoopHelper * helper,int * max_id)19 FileCallbackTest(int id, MessageLoopHelper* helper, int* max_id)
20 : id_(id),
21 helper_(helper),
22 max_id_(max_id) {
23 }
~FileCallbackTest()24 virtual ~FileCallbackTest() {}
25
26 virtual void OnFileIOComplete(int bytes_copied) OVERRIDE;
27
28 private:
29 int id_;
30 MessageLoopHelper* helper_;
31 int* max_id_;
32 };
33
OnFileIOComplete(int bytes_copied)34 void FileCallbackTest::OnFileIOComplete(int bytes_copied) {
35 if (id_ > *max_id_) {
36 NOTREACHED();
37 helper_->set_callback_reused_error(true);
38 }
39
40 helper_->CallbackWasCalled();
41 }
42
43 class TestFileBlock : public disk_cache::FileBlock {
44 public:
TestFileBlock()45 TestFileBlock() {
46 CacheTestFillBuffer(buffer_, sizeof(buffer_), false);
47 }
~TestFileBlock()48 virtual ~TestFileBlock() {}
49
50 // FileBlock interface.
buffer() const51 virtual void* buffer() const OVERRIDE { return const_cast<char*>(buffer_); }
size() const52 virtual size_t size() const OVERRIDE { return sizeof(buffer_); }
offset() const53 virtual int offset() const OVERRIDE { return 1024; }
54
55 private:
56 char buffer_[20];
57 };
58
59 } // namespace
60
TEST_F(DiskCacheTest,MappedFile_SyncIO)61 TEST_F(DiskCacheTest, MappedFile_SyncIO) {
62 base::FilePath filename = cache_path_.AppendASCII("a_test");
63 scoped_refptr<disk_cache::MappedFile> file(new disk_cache::MappedFile);
64 ASSERT_TRUE(CreateCacheTestFile(filename));
65 ASSERT_TRUE(file->Init(filename, 8192));
66
67 char buffer1[20];
68 char buffer2[20];
69 CacheTestFillBuffer(buffer1, sizeof(buffer1), false);
70 base::strlcpy(buffer1, "the data", arraysize(buffer1));
71 EXPECT_TRUE(file->Write(buffer1, sizeof(buffer1), 8192));
72 EXPECT_TRUE(file->Read(buffer2, sizeof(buffer2), 8192));
73 EXPECT_STREQ(buffer1, buffer2);
74 }
75
TEST_F(DiskCacheTest,MappedFile_AsyncIO)76 TEST_F(DiskCacheTest, MappedFile_AsyncIO) {
77 base::FilePath filename = cache_path_.AppendASCII("a_test");
78 scoped_refptr<disk_cache::MappedFile> file(new disk_cache::MappedFile);
79 ASSERT_TRUE(CreateCacheTestFile(filename));
80 ASSERT_TRUE(file->Init(filename, 8192));
81
82 int max_id = 0;
83 MessageLoopHelper helper;
84 FileCallbackTest callback(1, &helper, &max_id);
85
86 char buffer1[20];
87 char buffer2[20];
88 CacheTestFillBuffer(buffer1, sizeof(buffer1), false);
89 base::strlcpy(buffer1, "the data", arraysize(buffer1));
90 bool completed;
91 EXPECT_TRUE(file->Write(buffer1, sizeof(buffer1), 1024 * 1024, &callback,
92 &completed));
93 int expected = completed ? 0 : 1;
94
95 max_id = 1;
96 helper.WaitUntilCacheIoFinished(expected);
97
98 EXPECT_TRUE(file->Read(buffer2, sizeof(buffer2), 1024 * 1024, &callback,
99 &completed));
100 if (!completed)
101 expected++;
102
103 helper.WaitUntilCacheIoFinished(expected);
104
105 EXPECT_EQ(expected, helper.callbacks_called());
106 EXPECT_FALSE(helper.callback_reused_error());
107 EXPECT_STREQ(buffer1, buffer2);
108 }
109
TEST_F(DiskCacheTest,MappedFile_AsyncLoadStore)110 TEST_F(DiskCacheTest, MappedFile_AsyncLoadStore) {
111 base::FilePath filename = cache_path_.AppendASCII("a_test");
112 scoped_refptr<disk_cache::MappedFile> file(new disk_cache::MappedFile);
113 ASSERT_TRUE(CreateCacheTestFile(filename));
114 ASSERT_TRUE(file->Init(filename, 8192));
115
116 int max_id = 0;
117 MessageLoopHelper helper;
118 FileCallbackTest callback(1, &helper, &max_id);
119
120 TestFileBlock file_block1;
121 TestFileBlock file_block2;
122 base::strlcpy(static_cast<char*>(file_block1.buffer()), "the data",
123 file_block1.size());
124 bool completed;
125 EXPECT_TRUE(file->Store(&file_block1, &callback, &completed));
126 int expected = completed ? 0 : 1;
127
128 max_id = 1;
129 helper.WaitUntilCacheIoFinished(expected);
130
131 EXPECT_TRUE(file->Load(&file_block2, &callback, &completed));
132 if (!completed)
133 expected++;
134
135 helper.WaitUntilCacheIoFinished(expected);
136
137 EXPECT_EQ(expected, helper.callbacks_called());
138 EXPECT_FALSE(helper.callback_reused_error());
139 EXPECT_STREQ(static_cast<char*>(file_block1.buffer()),
140 static_cast<char*>(file_block2.buffer()));
141 }
142