Add chrome to the repository.

git-svn-id: svn://svn.chromium.org/chrome/trunk/src@15 0039d316-1c4b-4281-b951-d872f2087c98

1 files changed, 221 insertions, 0 deletions

diff --git a/chrome/browser/session_backend_unittest.cc b/chrome/browser/session_backend_unittest.cc
new file mode 100644
index 0000000..024dd9d
--- /dev/null
+++ b/chrome/browser/session_backend_unittest.cc
@@ -0,0 +1,221 @@
+// Copyright 2008, Google Inc.
+// All rights reserved.
+//
+// Redistribution and use in source and binary forms, with or without
+// modification, are permitted provided that the following conditions are
+// met:
+//
+//    * Redistributions of source code must retain the above copyright
+// notice, this list of conditions and the following disclaimer.
+//    * Redistributions in binary form must reproduce the above
+// copyright notice, this list of conditions and the following disclaimer
+// in the documentation and/or other materials provided with the
+// distribution.
+//    * Neither the name of Google Inc. nor the names of its
+// contributors may be used to endorse or promote products derived from
+// this software without specific prior written permission.
+//
+// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
+// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
+// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+
+#include "base/file_util.h"
+#include "base/path_service.h"
+#include "base/time.h"
+#include "chrome/browser/session_backend.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace {
+
+typedef std::vector<SessionCommand*> SessionCommands;
+
+struct TestData {
+  SessionCommand::id_type command_id;
+  std::string data;
+};
+
+SessionCommand* CreateCommandFromData(const TestData& data) {
+  SessionCommand* command =
+      new SessionCommand(
+          data.command_id,
+          static_cast<SessionCommand::size_type>(data.data.size()));
+  if (!data.data.empty())
+    memcpy(command->contents(), data.data.c_str(), data.data.size());
+  return command;
+}
+
+}  // namespace
+
+class SessionBackendTest : public testing::Test {
+ protected:
+  virtual void SetUp() {
+    wchar_t b[32];
+    _itow_s(static_cast<int>(GetTickCount()), b, arraysize(b), 10);
+
+    PathService::Get(base::DIR_TEMP, &path_);
+    file_util::AppendToPath(&path_, L"SessionTestDirs");
+    file_util::CreateDirectory(path_);
+    file_util::AppendToPath(&path_, b);
+  }
+
+  virtual void TearDown() {
+    file_util::Delete(path_, true);
+  }
+
+  void AssertCommandEqualsData(const TestData& data, SessionCommand* command) {
+    EXPECT_EQ(data.command_id, command->id());
+    EXPECT_EQ(data.data.size(), command->size());
+    EXPECT_TRUE(
+        memcmp(command->contents(), data.data.c_str(), command->size()) == 0);
+  }
+
+  // Path used in testing.
+  std::wstring path_;
+};
+
+TEST_F(SessionBackendTest, SimpleReadWrite) {
+  scoped_refptr<SessionBackend> backend(new SessionBackend(path_));
+  struct TestData data = { 1,  "a" };
+  std::vector<SessionCommand*> commands;
+  commands.push_back(CreateCommandFromData(data));
+  backend->AppendCommands(new SessionCommands(commands), false);
+  commands.clear();
+
+  // Read it back in.
+  backend = NULL;
+  backend = new SessionBackend(path_);
+  backend->ReadSessionImpl(false, &commands);
+
+  ASSERT_EQ(1, commands.size());
+  AssertCommandEqualsData(data, commands[0]);
+
+  STLDeleteElements(&commands);
+
+  backend = NULL;
+  backend = new SessionBackend(path_);
+  backend->ReadSessionImpl(false, &commands);
+
+  ASSERT_EQ(0, commands.size());
+
+  // Make sure we can delete.
+  backend->DeleteSession(false);
+  backend->ReadSessionImpl(false, &commands);
+  ASSERT_EQ(0, commands.size());
+}
+
+TEST_F(SessionBackendTest, RandomData) {
+  struct TestData data[] = {
+    { 1,  "a" },
+    { 2,  "ab" },
+    { 3,  "abc" },
+    { 4,  "abcd" },
+    { 5,  "abcde" },
+    { 6,  "abcdef" },
+    { 7,  "abcdefg" },
+    { 8,  "abcdefgh" },
+    { 9,  "abcdefghi" },
+    { 10, "abcdefghij" },
+    { 11, "abcdefghijk" },
+    { 12, "abcdefghijkl" },
+    { 13, "abcdefghijklm" },
+  };
+
+  for (int i = 0; i < arraysize(data); ++i) {
+    scoped_refptr<SessionBackend> backend(new SessionBackend(path_));
+    std::vector<SessionCommand*> commands;
+    if (i != 0) {
+      // Read previous data.
+      backend->ReadSessionImpl(false, &commands);
+      ASSERT_EQ(i, commands.size());
+      for (std::vector<SessionCommand*>::iterator j = commands.begin();
+           j != commands.end(); ++j) {
+        AssertCommandEqualsData(data[j - commands.begin()], *j);
+      }
+      backend->AppendCommands(new SessionCommands(commands), false);
+      commands.clear();
+    }
+    commands.push_back(CreateCommandFromData(data[i]));
+    backend->AppendCommands(new SessionCommands(commands), false);
+  }
+}
+
+TEST_F(SessionBackendTest, BigData) {
+  struct TestData data[] = {
+    { 1,  "a" },
+    { 2,  "ab" },
+  };
+
+  scoped_refptr<SessionBackend> backend(new SessionBackend(path_));
+  std::vector<SessionCommand*> commands;
+  commands.push_back(CreateCommandFromData(data[0]));
+  const SessionCommand::size_type big_size = SessionBackend::kFileReadBufferSize + 100;
+  const SessionCommand::id_type big_id = 50;
+  SessionCommand* big_command = new SessionCommand(big_id, big_size);
+  reinterpret_cast<char*>(big_command->contents())[0] = 'a';
+  reinterpret_cast<char*>(big_command->contents())[big_size - 1] = 'z';
+  commands.push_back(big_command);
+  commands.push_back(CreateCommandFromData(data[1]));
+  backend->AppendCommands(new SessionCommands(commands), false);
+  commands.clear();
+
+  backend = NULL;
+  backend = new SessionBackend(path_);
+  commands.clear();
+  backend->ReadSessionImpl(false, &commands);
+  ASSERT_EQ(3, commands.size());
+  AssertCommandEqualsData(data[0], commands[0]);
+  AssertCommandEqualsData(data[1], commands[2]);
+
+  EXPECT_EQ(big_id, commands[1]->id());
+  ASSERT_EQ(big_size, commands[1]->size());
+  EXPECT_EQ('a', reinterpret_cast<char*>(commands[1]->contents())[0]);
+  EXPECT_EQ('z',
+            reinterpret_cast<char*>(commands[1]->contents())[big_size - 1]);
+  STLDeleteElements(&commands);
+}
+
+TEST_F(SessionBackendTest, SaveSession) {
+  struct TestData data[] = {
+    { 1,  "a" },
+    { 2,  "ab" },
+  };
+
+  scoped_refptr<SessionBackend> backend(new SessionBackend(path_));
+  std::vector<SessionCommand*> commands;
+  for (int i = 0; i < arraysize(data); ++i) {
+    commands.push_back(CreateCommandFromData(data[i]));
+  }
+  backend->SaveSession(commands);
+
+  commands.clear();
+
+  backend->ReadSessionImpl(true, &commands);
+  ASSERT_EQ(arraysize(data), commands.size());
+  for (int i = 0; i < arraysize(data); ++i)
+    AssertCommandEqualsData(data[i], commands[i]);
+  STLDeleteElements(&commands);
+}
+
+TEST_F(SessionBackendTest, EmptyCommand) {
+  TestData empty_command;
+  empty_command.command_id = 1;
+  scoped_refptr<SessionBackend> backend(new SessionBackend(path_));
+  std::vector<SessionCommand*> commands;
+  commands.push_back(CreateCommandFromData(empty_command));
+  backend->SaveSession(commands);
+
+  commands.clear();
+
+  backend->ReadSessionImpl(true, &commands);
+  ASSERT_EQ(1, commands.size());
+  AssertCommandEqualsData(empty_command, commands[0]);
+  STLDeleteElements(&commands);
+}