// Copyright 2016 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 "tools/gn/analyzer.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "tools/gn/build_settings.h"
#include "tools/gn/builder.h"
#include "tools/gn/loader.h"
#include "tools/gn/settings.h"
#include "tools/gn/source_file.h"

namespace {

class MockLoader : public Loader {
 public:
  MockLoader() {}

  void Load(const SourceFile& file,
            const LocationRange& origin,
            const Label& toolchain_name) override {}
  void ToolchainLoaded(const Toolchain* toolchain) override {}
  Label GetDefaultToolchain() const override {
    return Label(SourceDir("//tc/"), "default");
  }
  const Settings* GetToolchainSettings(const Label& label) const override {
    return nullptr;
  }

 private:
  ~MockLoader() override {}
};

class AnalyzerTest : public testing::Test {
 public:
  AnalyzerTest()
      : loader_(new MockLoader),
        builder_(loader_.get()),
        settings_(&build_settings_, std::string()) {
    build_settings_.SetBuildDir(SourceDir("//out/"));
    settings_.set_toolchain_label(Label(SourceDir("//tc/"), "default"));
    settings_.set_default_toolchain_label(settings_.toolchain_label());
    tc_dir_ = settings_.toolchain_label().dir();
    tc_name_ = settings_.toolchain_label().name();
  }

  Target* MakeTarget(const std::string dir,
                     const std::string name,
                     Target::OutputType type,
                     const std::vector<std::string>& sources,
                     const std::vector<Target*>& deps) {
    Label lbl(SourceDir(dir), name, tc_dir_, tc_name_);
    Target* target = new Target(&settings_, lbl);
    target->set_output_type(type);
    for (const auto& s : sources)
      target->sources().push_back(SourceFile(s));
    for (const auto* d : deps)
      target->public_deps().push_back(LabelTargetPair(d));
    builder_.ItemDefined(std::unique_ptr<Item>(target));
    return target;
  }

  void AddSource(Target* a, std::string path) {}

  void AddDep(Target* a, Target* b) {}

  void SetUpABasicBuildGraph() {
    std::vector<std::string> no_sources;
    std::vector<Target*> no_deps;

    // All of the targets below are owned by the builder, so none of them
    // get leaked.

    // Ignore the returned target since nothing depends on it.
    MakeTarget("//", "a", Target::EXECUTABLE, {"//a.cc"}, no_deps);

    Target* b =
        MakeTarget("//d", "b", Target::SOURCE_SET, {"//d/b.cc"}, no_deps);

    Target* b_unittests = MakeTarget("//d", "b_unittests", Target::EXECUTABLE,
                                     {"//d/b_unittest.cc"}, {b});

    Target* c = MakeTarget("//d", "c", Target::EXECUTABLE, {"//d/c.cc"}, {b});

    Target* b_unittests_and_c =
        MakeTarget("//d", "b_unittests_and_c", Target::GROUP, no_sources,
                   {b_unittests, c});

    Target* e =
        MakeTarget("//d", "e", Target::EXECUTABLE, {"//d/e.cc"}, no_deps);

    // Also ignore this returned target since nothing depends on it.
    MakeTarget("//d", "d", Target::GROUP, no_sources, {b_unittests_and_c, e});
  }

  void RunBasicTest(const std::string& input,
                    const std::string& expected_output) {
    SetUpABasicBuildGraph();
    Err err;
    std::string actual_output = Analyzer(builder_).Analyze(input, &err);
    EXPECT_EQ(err.has_error(), false);
    EXPECT_EQ(expected_output, actual_output);
  }

 protected:
  scoped_refptr<MockLoader> loader_;
  Builder builder_;
  BuildSettings build_settings_;
  Settings settings_;
  SourceDir tc_dir_;
  std::string tc_name_;
};

}  // namespace

TEST_F(AnalyzerTest, AllWasPruned) {
  RunBasicTest(
      R"({
        "files": [ "//d/b.cc" ],
        "additional_compile_targets": [ "all" ],
        "test_targets": [ ]
      })",
      "{"
      R"("compile_targets":["//d:b_unittests","//d:c"],)"
      R"("status":"Found dependency",)"
      R"("test_targets":[])"
      "}");
}

TEST_F(AnalyzerTest, NoDependency) {
  RunBasicTest(
      R"({
        "files":[ "//missing.cc" ],
        "additional_compile_targets": [ "all" ],
        "test_targets": [ "//:a" ]
      })",
      "{"
      R"("compile_targets":[],)"
      R"("status":"No dependency",)"
      R"("test_targets":[])"
      "}");
}

TEST_F(AnalyzerTest, NoFilesNoTargets) {
  RunBasicTest(
      R"({
        "files": [],
        "additional_compile_targets": [],
        "test_targets": []
      })",
      "{"
      R"("compile_targets":[],)"
      R"("status":"No dependency",)"
      R"("test_targets":[])"
      "}");
}

TEST_F(AnalyzerTest, OneTestTargetModified) {
  RunBasicTest(
      R"({
        "files": [ "//a.cc" ],
        "additional_compile_targets": [],
        "test_targets": [ "//:a" ]
      })",
      "{"
      R"("compile_targets":[],)"
      R"("status":"Found dependency",)"
      R"("test_targets":["//:a"])"
      "}");
}

TEST_F(AnalyzerTest, FilesArentSourceAbsolute) {
  RunBasicTest(
      R"({
        "files": [ "a.cc" ],
        "additional_compile_targets": [],
        "test_targets": [ "//:a" ]
      })",
      "{"
      R"("error":)"
      R"("\"a.cc\" is not a source-absolute or absolute path.",)"
      R"("invalid_targets":[])"
      "}");
}

TEST_F(AnalyzerTest, WrongInputFields) {
  RunBasicTest(
      R"({
        "files": [ "//a.cc" ],
        "compile_targets": [],
        "test_targets": [ "//:a" ]
      })",
      "{"
      R"("error":)"
      R"("Input does not have a key named )"
      R"(\"additional_compile_targets\" with a list value.",)"
      R"("invalid_targets":[])"
      "}");
}

TEST_F(AnalyzerTest, BuildFilesWereModified) {
  // This tests that if a build file is modified, we bail out early with
  // "Found dependency (all)" error since we can't handle changes to
  // build files yet (crbug.com/555273).
  RunBasicTest(
      R"({
        "files": [ "//a.cc", "//BUILD.gn" ],
        "additional_compile_targets": [],
        "test_targets": [ "//:a" ]
      })",
      "{"
      R"("compile_targets":["//:a"],)"
      R"/("status":"Found dependency (all)",)/"
      R"("test_targets":["//:a"])"
      "}");
}

TEST_F(AnalyzerTest, BuildFilesWereModifiedAndCompilingAll) {
  // This tests that if a build file is modified, we bail out early with
  // "Found dependency (all)" error since we can't handle changes to
  // build files yet (crbug.com/555273).
  RunBasicTest(
      R"({
        "files": [ "//a.cc", "//BUILD.gn" ],
        "additional_compile_targets": [ "all" ],
        "test_targets": [ "//:a" ]
      })",
      "{"
      R"("compile_targets":["all"],)"
      R"/("status":"Found dependency (all)",)/"
      R"("test_targets":["//:a"])"
      "}");
}