mapped_file_unittest.cc revision a1401311d1ab56c4ed0a474bd38c108f75cb0cd9 
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/blockfile/file_block.h"
9#include "net/disk_cache/blockfile/mapped_file.h"
10#include "net/disk_cache/disk_cache_test_base.h"
11#include "net/disk_cache/disk_cache_test_util.h"
12#include "testing/gtest/include/gtest/gtest.h"
13
14namespace {
15
16// Implementation of FileIOCallback for the tests.
17class FileCallbackTest: public disk_cache::FileIOCallback {
18 public:
19  FileCallbackTest(int id, MessageLoopHelper* helper, int* max_id)
20      : id_(id),
21        helper_(helper),
22        max_id_(max_id) {
23  }
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
34void 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
43class TestFileBlock : public disk_cache::FileBlock {
44 public:
45  TestFileBlock() {
46    CacheTestFillBuffer(buffer_, sizeof(buffer_), false);
47  }
48  virtual ~TestFileBlock() {}
49
50  // FileBlock interface.
51  virtual void* buffer() const OVERRIDE { return const_cast<char*>(buffer_); }
52  virtual size_t size() const OVERRIDE { return sizeof(buffer_); }
53  virtual int offset() const OVERRIDE { return 1024; }
54
55 private:
56  char buffer_[20];
57};
58
59}  // namespace
60
61TEST_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
76TEST_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
110TEST_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