// Copyright 2013 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "base/version.h"
#include "extensions/common/extension.h"
#include "extensions/common/extension_id.h"
#include "extensions/common/manifest_constants.h"
#include "extensions/common/manifest_handlers/shared_module_info.h"
#include "extensions/common/manifest_test.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace {

const char kValidImportPath[] =
    "_modules/abcdefghijklmnopabcdefghijklmnop/foo/bar.html";
const char kValidImportPathID[] = "abcdefghijklmnopabcdefghijklmnop";
const char kValidImportPathRelative[] = "foo/bar.html";
const char kInvalidImportPath[] = "_modules/abc/foo.html";
const char kImportId1[] = "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa";
const char kImportId2[] = "bbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbb";
const char kNoImport[] = "cccccccccccccccccccccccccccccccc";

}  // namespace

namespace extensions {

class SharedModuleManifestTest : public ManifestTest {
};

TEST_F(SharedModuleManifestTest, ExportsAll) {
  ManifestData manifest("shared_module_export.json");

  scoped_refptr<Extension> extension = LoadAndExpectSuccess(manifest);

  EXPECT_TRUE(SharedModuleInfo::IsSharedModule(extension.get()))
      << manifest.name();
  EXPECT_FALSE(SharedModuleInfo::ImportsModules(extension.get()))
      << manifest.name();

  EXPECT_TRUE(
      SharedModuleInfo::IsExportAllowedByAllowlist(extension.get(), kImportId1))
      << manifest.name();
  EXPECT_TRUE(
      SharedModuleInfo::IsExportAllowedByAllowlist(extension.get(), kImportId2))
      << manifest.name();
  EXPECT_FALSE(
      SharedModuleInfo::IsExportAllowedByAllowlist(extension.get(), kNoImport))
      << manifest.name();
}

TEST_F(SharedModuleManifestTest, ExportAllowlistAll) {
  ManifestData manifest("shared_module_export_no_allowlist.json");

  scoped_refptr<Extension> extension = LoadAndExpectSuccess(manifest);

  EXPECT_TRUE(
      SharedModuleInfo::IsExportAllowedByAllowlist(extension.get(), kImportId1))
      << manifest.name();
  EXPECT_TRUE(
      SharedModuleInfo::IsExportAllowedByAllowlist(extension.get(), kImportId2))
      << manifest.name();
  EXPECT_TRUE(
      SharedModuleInfo::IsExportAllowedByAllowlist(extension.get(), kNoImport))
      << manifest.name();
}

TEST_F(SharedModuleManifestTest, ExportAllowlistEmpty) {
  scoped_refptr<Extension> extension =
      LoadAndExpectWarning("shared_module_export_allowlist_empty.json",
                           manifest_errors::kInvalidExportAllowlistEmpty);

  EXPECT_TRUE(
      SharedModuleInfo::IsExportAllowedByAllowlist(extension.get(), kImportId1))
      << extension.get()->name();
  EXPECT_TRUE(
      SharedModuleInfo::IsExportAllowedByAllowlist(extension.get(), kImportId2))
      << extension.get()->name();
  EXPECT_TRUE(
      SharedModuleInfo::IsExportAllowedByAllowlist(extension.get(), kNoImport))
      << extension.get()->name();
}

TEST_F(SharedModuleManifestTest, ExportParseErrors) {
  const Testcase testcases[] = {
      Testcase("shared_module_export_and_import.json",
               "Simultaneous 'import' and 'export' are not allowed."),
      Testcase("shared_module_export_not_dict.json",
               "Error at key 'export'. Type is invalid. Expected dictionary, "
               "found list."),
      Testcase("shared_module_export_allowlist_item_not_id.json",
               "Invalid value for 'export.allowlist[0]'."),
      Testcase("shared_module_export_allowlist_item_not_string.json",
               "Error at key 'export.allowlist'. Parsing array failed at index "
               "0: expected string, got dictionary"),
      Testcase("shared_module_export_allowlist_not_list.json",
               "Error at key 'export.allowlist'. Type is invalid. Expected "
               "list, found string."),
  };
  RunTestcases(testcases, ExpectType::kError);
}

TEST_F(SharedModuleManifestTest, SharedModuleStaticFunctions) {
  EXPECT_TRUE(SharedModuleInfo::IsImportedPath(kValidImportPath));
  EXPECT_FALSE(SharedModuleInfo::IsImportedPath(kInvalidImportPath));

  ExtensionId id;
  std::string relative;
  SharedModuleInfo::ParseImportedPath(kValidImportPath, &id, &relative);
  EXPECT_EQ(id, kValidImportPathID);
  EXPECT_EQ(relative, kValidImportPathRelative);
}

TEST_F(SharedModuleManifestTest, Import) {
  ManifestData manifest("shared_module_import.json");

  scoped_refptr<Extension> extension = LoadAndExpectSuccess(manifest);

  EXPECT_FALSE(SharedModuleInfo::IsSharedModule(extension.get()))
      << manifest.name();
  EXPECT_TRUE(SharedModuleInfo::ImportsModules(extension.get()))
      << manifest.name();
  const std::vector<SharedModuleInfo::ImportInfo>& imports =
      SharedModuleInfo::GetImports(extension.get());
  ASSERT_EQ(2U, imports.size());
  EXPECT_EQ(imports[0].extension_id, kImportId1);
  EXPECT_EQ(imports[0].minimum_version, "");
  EXPECT_EQ(imports[1].extension_id, kImportId2);
  EXPECT_TRUE(base::Version(imports[1].minimum_version).IsValid());
  EXPECT_TRUE(
      SharedModuleInfo::ImportsExtensionById(extension.get(), kImportId1));
  EXPECT_TRUE(
      SharedModuleInfo::ImportsExtensionById(extension.get(), kImportId2));
  EXPECT_FALSE(
      SharedModuleInfo::ImportsExtensionById(extension.get(), kNoImport));
}

TEST_F(SharedModuleManifestTest, ImportRepeats) {
  scoped_refptr<Extension> extension =
      LoadAndExpectWarning("shared_module_import_repeats.json",
                           manifest_errors::kInvalidImportRepeatedImport);

  EXPECT_FALSE(SharedModuleInfo::IsSharedModule(extension.get()))
      << extension.get()->name();
  EXPECT_TRUE(SharedModuleInfo::ImportsModules(extension.get()))
      << extension.get()->name();
  const std::vector<SharedModuleInfo::ImportInfo>& imports =
      SharedModuleInfo::GetImports(extension.get());
  ASSERT_EQ(4u, imports.size());
  EXPECT_EQ(imports[0].extension_id, kImportId1);
  EXPECT_EQ(imports[0].minimum_version, "");
  EXPECT_EQ(imports[1].extension_id, kImportId2);
  EXPECT_EQ(imports[1].minimum_version, "");
  EXPECT_EQ(imports[2].extension_id, kImportId1);
  EXPECT_EQ(imports[2].minimum_version, "1.0");
  EXPECT_EQ(imports[3].extension_id, kImportId1);
  EXPECT_EQ(imports[3].minimum_version, "1.1");
  EXPECT_TRUE(base::Version(imports[2].minimum_version).IsValid());
  EXPECT_TRUE(base::Version(imports[3].minimum_version).IsValid());
  EXPECT_TRUE(
      SharedModuleInfo::ImportsExtensionById(extension.get(), kImportId1));
  EXPECT_TRUE(
      SharedModuleInfo::ImportsExtensionById(extension.get(), kImportId2));
  EXPECT_FALSE(
      SharedModuleInfo::ImportsExtensionById(extension.get(), kNoImport));
}

TEST_F(SharedModuleManifestTest, ImportParseErrors) {
  const Testcase testcases[] = {
      Testcase("shared_module_import_not_list.json",
               "Error at key 'import'. Type is invalid. Expected list, found "
               "dictionary."),
      Testcase("shared_module_import_invalid_id.json",
               "Invalid value for 'import[0].id'."),
      Testcase("shared_module_import_invalid_version.json",
               "Invalid value for 'import[0].minimum_version'."),
  };
  RunTestcases(testcases, ExpectType::kError);
}

}  // namespace extensions