// Copyright 2017 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/conflicts/module_event_sink_impl_win.h"

#include <memory>

#include "base/memory/ptr_util.h"
#include "base/test/scoped_task_environment.h"
#include "base/threading/sequenced_task_runner_handle.h"
#include "chrome/browser/conflicts/module_database_win.h"
#include "chrome/common/conflicts/module_watcher_win.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace {

// The address of this module in memory. The linker will take care of defining
// this symbol.
extern "C" IMAGE_DOS_HEADER __ImageBase;

// An invalid load address.
const uint64_t kInvalidLoadAddress = 0xDEADBEEF;

}  // namespace

class ModuleEventSinkImplTest : public testing::Test {
 protected:
  ModuleEventSinkImplTest()
      : module_database_(base::MakeUnique<ModuleDatabase>(
            base::SequencedTaskRunnerHandle::Get())) {}

  void CreateModuleSinkImpl() {
    module_event_sink_impl_ = base::MakeUnique<ModuleEventSinkImpl>(
        ::GetCurrentProcess(), content::PROCESS_TYPE_BROWSER,
        module_database_.get());
  }

  const ModuleDatabase::ModuleMap& modules() {
    return module_database_->modules_;
  }

  const ModuleDatabase::ProcessMap& processes() {
    return module_database_->processes_;
  }

  uint32_t process_id() { return module_event_sink_impl_->process_id_; }

  // Must be before |module_database_|.
  base::test::ScopedTaskEnvironment scoped_task_environment_;
  std::unique_ptr<ModuleDatabase> module_database_;
  std::unique_ptr<ModuleEventSinkImpl> module_event_sink_impl_;

 private:
  DISALLOW_COPY_AND_ASSIGN(ModuleEventSinkImplTest);
};

TEST_F(ModuleEventSinkImplTest, CallsForwardedAsExpected) {
  const uintptr_t kValidLoadAddress = reinterpret_cast<uintptr_t>(&__ImageBase);

  EXPECT_EQ(0u, modules().size());
  EXPECT_EQ(0u, processes().size());

  // Construction should immediately fire off a call to OnProcessStarted and
  // create a process entry in the module database.
  CreateModuleSinkImpl();
  EXPECT_EQ(::GetCurrentProcessId(), process_id());
  EXPECT_EQ(0u, modules().size());
  EXPECT_EQ(1u, processes().size());

  // An invalid load event should not cause a module entry.
  module_event_sink_impl_->OnModuleEvent(
      mojom::ModuleEventType::MODULE_ALREADY_LOADED, kInvalidLoadAddress);
  EXPECT_EQ(0u, modules().size());
  EXPECT_EQ(1u, processes().size());

  // A valid load event should cause a module entry.
  module_event_sink_impl_->OnModuleEvent(mojom::ModuleEventType::MODULE_LOADED,
                                         kValidLoadAddress);
  EXPECT_EQ(1u, modules().size());
  EXPECT_EQ(1u, processes().size());
}