// Copyright 2013 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 "chrome/browser/install_verification/win/module_ids.h"

#include "base/strings/string_piece.h"
#include "chrome/browser/install_verification/win/module_verification_common.h"
#include "testing/gtest/include/gtest/gtest.h"

TEST(ModuleIDsTest, LoadModuleIDs) {
  ModuleIDs module_ids;
  LoadModuleIDs(&module_ids);
  ASSERT_EQ(1u, module_ids[CalculateModuleNameDigest(L"chrome.dll")]);
}

TEST(ModuleIDsTest, ParseAdditionalModuleIDs) {
  ModuleIDs module_ids;

  ParseAdditionalModuleIDs(base::StringPiece(), &module_ids);
  ASSERT_EQ(0u, module_ids.size());

  // Invalid input.
  ParseAdditionalModuleIDs("hello", &module_ids);
  ASSERT_EQ(0u, module_ids.size());
  ParseAdditionalModuleIDs("hello world", &module_ids);
  ASSERT_EQ(0u, module_ids.size());
  ParseAdditionalModuleIDs("hello world\nfoo bar", &module_ids);
  ASSERT_EQ(0u, module_ids.size());
  ParseAdditionalModuleIDs("123 world\nfoo bar", &module_ids);
  ASSERT_EQ(0u, module_ids.size());
  ParseAdditionalModuleIDs("hello 0123456789abcdef0123456789abcdef\nfoo bar",
                           &module_ids);
  ASSERT_EQ(0u, module_ids.size());

  // A single valid line followed by an invalid line.
  ParseAdditionalModuleIDs("123 0123456789abcdef0123456789abcdef\nfoo bar",
                         &module_ids);
  ASSERT_EQ(1u, module_ids.size());
  ASSERT_EQ(123u, module_ids.begin()->second);
  ASSERT_EQ("0123456789abcdef0123456789abcdef", module_ids.begin()->first);
  module_ids.clear();

  // The same, but with \r\n.
  ParseAdditionalModuleIDs("123 0123456789abcdef0123456789abcdef\r\nfoo bar",
                           &module_ids);
  ASSERT_EQ(1u, module_ids.size());
  ASSERT_EQ(123u, module_ids.begin()->second);
  ASSERT_EQ("0123456789abcdef0123456789abcdef", module_ids.begin()->first);
  module_ids.clear();

  // Several valid and invalid lines, with varying line terminations etc.
  ParseAdditionalModuleIDs("123 0123456789abcdef0123456789abcdef\r\n"
                           "456 DEADBEEFDEADBEEF\r"
                           "789 DEADBEEFDEADBEEFDEADBEEFDEADBEEF\n"
                           "\n"
                           "\n"
                           "321 BAADCAFEBAADCAFEBAADCAFEBAADCAFE\n",
                           &module_ids);
  ASSERT_EQ(3u, module_ids.size());
  ASSERT_EQ(123u, module_ids["0123456789abcdef0123456789abcdef"]);
  ASSERT_EQ(789u, module_ids["DEADBEEFDEADBEEFDEADBEEFDEADBEEF"]);
  ASSERT_EQ(321u, module_ids["BAADCAFEBAADCAFEBAADCAFEBAADCAFE"]);
  module_ids.clear();

  // Leading empty lines, no termination on final line.
  ParseAdditionalModuleIDs("\n"
                           "123 0123456789abcdef0123456789abcdef\r\n"
                           "321 BAADCAFEBAADCAFEBAADCAFEBAADCAFE",
                           &module_ids);
  ASSERT_EQ(2u, module_ids.size());
  ASSERT_EQ(123u, module_ids["0123456789abcdef0123456789abcdef"]);
  ASSERT_EQ(321u, module_ids["BAADCAFEBAADCAFEBAADCAFEBAADCAFE"]);
  module_ids.clear();
}