net/disk_cache/mapped_file_unittest.cc

   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:
  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
  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:
  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
  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
  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
 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 }