// Copyright (c) 2006-2008 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 <windows.h>

#include <fstream>
#include <iostream>

#include "base/base_paths.h"
#include "base/file_util.h"
#include "base/logging.h"
#include "base/path_service.h"
#include "base/process_util.h"
#include "base/scoped_ptr.h"
#include "base/string_util.h"
#include "chrome/installer/util/helper.h"
#include "chrome/installer/util/work_item.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace {
  class SetupHelperTest : public testing::Test {
   protected:
    virtual void SetUp() {
      // Name a subdirectory of the user temp directory.
      ASSERT_TRUE(PathService::Get(base::DIR_TEMP, &test_dir_));
      file_util::AppendToPath(&test_dir_, L"SetupHelperTest");

      // Create a fresh, empty copy of this test directory.
      file_util::Delete(test_dir_, true);
      CreateDirectory(test_dir_.c_str(), NULL);
      ASSERT_TRUE(file_util::PathExists(test_dir_));
    }

    virtual void TearDown() {
      logging::CloseLogFile();
      // Clean up test directory
      ASSERT_TRUE(file_util::Delete(test_dir_, false));
      ASSERT_FALSE(file_util::PathExists(test_dir_));
    }

    // the path to temporary directory used to contain the test operations
    std::wstring test_dir_;
  };

  // Simple function to dump some text into a new file.
  void CreateTextFile(const std::wstring& filename,
                      const std::wstring& contents) {
    std::ofstream file;
    file.open(filename.c_str());
    ASSERT_TRUE(file.is_open());
    file << contents;
    file.close();
  }

  wchar_t text_content_1[] = L"delete me";
  wchar_t text_content_2[] = L"delete me as well";
};

// Delete version directories. Everything lower than the given version
// should be deleted.
TEST_F(SetupHelperTest, Delete) {
  // Create a Chrome dir
  std::wstring chrome_dir(test_dir_);
  file_util::AppendToPath(&chrome_dir, L"chrome");
  CreateDirectory(chrome_dir.c_str(), NULL);
  ASSERT_TRUE(file_util::PathExists(chrome_dir));

  std::wstring chrome_dir_1(chrome_dir);
  file_util::AppendToPath(&chrome_dir_1, L"1.0.1.0");
  CreateDirectory(chrome_dir_1.c_str(), NULL);
  ASSERT_TRUE(file_util::PathExists(chrome_dir_1));

  std::wstring chrome_dir_2(chrome_dir);
  file_util::AppendToPath(&chrome_dir_2, L"1.0.2.0");
  CreateDirectory(chrome_dir_2.c_str(), NULL);
  ASSERT_TRUE(file_util::PathExists(chrome_dir_2));

  std::wstring chrome_dir_3(chrome_dir);
  file_util::AppendToPath(&chrome_dir_3, L"1.0.3.0");
  CreateDirectory(chrome_dir_3.c_str(), NULL);
  ASSERT_TRUE(file_util::PathExists(chrome_dir_3));

  std::wstring chrome_dir_4(chrome_dir);
  file_util::AppendToPath(&chrome_dir_4, L"1.0.4.0");
  CreateDirectory(chrome_dir_4.c_str(), NULL);
  ASSERT_TRUE(file_util::PathExists(chrome_dir_4));

  std::wstring chrome_dll_1(chrome_dir_1);
  file_util::AppendToPath(&chrome_dll_1, L"chrome.dll");
  CreateTextFile(chrome_dll_1, text_content_1);
  ASSERT_TRUE(file_util::PathExists(chrome_dll_1));

  std::wstring chrome_dll_2(chrome_dir_2);
  file_util::AppendToPath(&chrome_dll_2, L"chrome.dll");
  CreateTextFile(chrome_dll_2, text_content_1);
  ASSERT_TRUE(file_util::PathExists(chrome_dll_2));

  std::wstring chrome_dll_3(chrome_dir_3);
  file_util::AppendToPath(&chrome_dll_3, L"chrome.dll");
  CreateTextFile(chrome_dll_3, text_content_1);
  ASSERT_TRUE(file_util::PathExists(chrome_dll_3));

  std::wstring chrome_dll_4(chrome_dir_4);
  file_util::AppendToPath(&chrome_dll_4, L"chrome.dll");
  CreateTextFile(chrome_dll_4, text_content_1);
  ASSERT_TRUE(file_util::PathExists(chrome_dll_4));

  std::wstring latest_version(L"1.0.4.0");
  installer::RemoveOldVersionDirs(chrome_dir, latest_version);

  // old versions should be gone
  EXPECT_FALSE(file_util::PathExists(chrome_dir_1));
  EXPECT_FALSE(file_util::PathExists(chrome_dir_2));
  EXPECT_FALSE(file_util::PathExists(chrome_dir_3));
  // the latest version should stay
  EXPECT_TRUE(file_util::PathExists(chrome_dll_4));
}

// Delete older version directories, keeping the one in used intact.
TEST_F(SetupHelperTest, DeleteInUsed) {
  // Create a Chrome dir
  std::wstring chrome_dir(test_dir_);
  file_util::AppendToPath(&chrome_dir, L"chrome");
  CreateDirectory(chrome_dir.c_str(), NULL);
  ASSERT_TRUE(file_util::PathExists(chrome_dir));

  std::wstring chrome_dir_1(chrome_dir);
  file_util::AppendToPath(&chrome_dir_1, L"1.0.1.0");
  CreateDirectory(chrome_dir_1.c_str(), NULL);
  ASSERT_TRUE(file_util::PathExists(chrome_dir_1));

  std::wstring chrome_dir_2(chrome_dir);
  file_util::AppendToPath(&chrome_dir_2, L"1.0.2.0");
  CreateDirectory(chrome_dir_2.c_str(), NULL);
  ASSERT_TRUE(file_util::PathExists(chrome_dir_2));

  std::wstring chrome_dir_3(chrome_dir);
  file_util::AppendToPath(&chrome_dir_3, L"1.0.3.0");
  CreateDirectory(chrome_dir_3.c_str(), NULL);
  ASSERT_TRUE(file_util::PathExists(chrome_dir_3));

  std::wstring chrome_dir_4(chrome_dir);
  file_util::AppendToPath(&chrome_dir_4, L"1.0.4.0");
  CreateDirectory(chrome_dir_4.c_str(), NULL);
  ASSERT_TRUE(file_util::PathExists(chrome_dir_4));

  std::wstring chrome_dll_1(chrome_dir_1);
  file_util::AppendToPath(&chrome_dll_1, L"chrome.dll");
  CreateTextFile(chrome_dll_1, text_content_1);
  ASSERT_TRUE(file_util::PathExists(chrome_dll_1));

  std::wstring chrome_dll_2(chrome_dir_2);
  file_util::AppendToPath(&chrome_dll_2, L"chrome.dll");
  CreateTextFile(chrome_dll_2, text_content_1);
  ASSERT_TRUE(file_util::PathExists(chrome_dll_2));

  // Open the file to make it in use.
  std::ofstream file;
  file.open(chrome_dll_2.c_str());

  std::wstring chrome_othera_2(chrome_dir_2);
  file_util::AppendToPath(&chrome_othera_2, L"othera.dll");
  CreateTextFile(chrome_othera_2, text_content_2);
  ASSERT_TRUE(file_util::PathExists(chrome_othera_2));

  std::wstring chrome_otherb_2(chrome_dir_2);
  file_util::AppendToPath(&chrome_otherb_2, L"otherb.dll");
  CreateTextFile(chrome_otherb_2, text_content_2);
  ASSERT_TRUE(file_util::PathExists(chrome_otherb_2));

  std::wstring chrome_dll_3(chrome_dir_3);
  file_util::AppendToPath(&chrome_dll_3, L"chrome.dll");
  CreateTextFile(chrome_dll_3, text_content_1);
  ASSERT_TRUE(file_util::PathExists(chrome_dll_3));

  std::wstring chrome_dll_4(chrome_dir_4);
  file_util::AppendToPath(&chrome_dll_4, L"chrome.dll");
  CreateTextFile(chrome_dll_4, text_content_1);
  ASSERT_TRUE(file_util::PathExists(chrome_dll_4));

  std::wstring latest_version(L"1.0.4.0");
  installer::RemoveOldVersionDirs(chrome_dir, latest_version);

  // old versions not in used should be gone
  EXPECT_FALSE(file_util::PathExists(chrome_dir_1));
  EXPECT_FALSE(file_util::PathExists(chrome_dir_3));
  // every thing under in used version should stay
  EXPECT_TRUE(file_util::PathExists(chrome_dir_2));
  EXPECT_TRUE(file_util::PathExists(chrome_dll_2));
  EXPECT_TRUE(file_util::PathExists(chrome_othera_2));
  EXPECT_TRUE(file_util::PathExists(chrome_otherb_2));
  // the latest version should stay
  EXPECT_TRUE(file_util::PathExists(chrome_dll_4));
}