chromeos/components/quick_answers/utils/spell_checker_unittest.cc

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

#include "chromeos/components/quick_answers/utils/spell_checker.h"

#include <memory>
#include <string>

#include "base/files/file_path.h"
#include "base/files/file_util.h"
#include "base/functional/bind.h"
#include "base/path_service.h"
#include "base/test/scoped_feature_list.h"
#include "base/test/scoped_path_override.h"
#include "base/test/task_environment.h"
#include "chrome/common/chrome_paths.h"
#include "chromeos/components/quick_answers/test/quick_answers_test_base.h"
#include "chromeos/constants/chromeos_features.h"
#include "content/public/test/test_utils.h"
#include "services/network/public/cpp/weak_wrapper_shared_url_loader_factory.h"
#include "services/network/test/test_url_loader_factory.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace quick_answers {

class SpellCheckerTest : public QuickAnswersTestBase {
 public:
  SpellCheckerTest() = default;
  SpellCheckerTest(const SpellCheckerTest&) = delete;
  SpellCheckerTest& operator=(const SpellCheckerTest&) = delete;
  ~SpellCheckerTest() override = default;

  // QuickAnswersTestBase:
  void SetUp() override {
    QuickAnswersTestBase::SetUp();

    test_shared_loader_factory_ =
        base::MakeRefCounted<network::WeakWrapperSharedURLLoaderFactory>(
            &test_url_loader_factory_);
    spell_checker_ =
        std::make_unique<SpellChecker>(test_shared_loader_factory_);

    // Set up a temporary directory which will be used as the fake dictionary
    // dir.
    ASSERT_TRUE(fake_dict_dir_.CreateUniqueTempDir());
    path_override_ = std::make_unique<base::ScopedPathOverride>(
        chrome::DIR_APP_DICTIONARIES, fake_dict_dir_.GetPath());
  }

  void TearDown() override {
    spell_checker_.reset();
    QuickAnswersTestBase::TearDown();
  }

  SpellChecker* spell_checker() { return spell_checker_.get(); }

 private:
  base::test::TaskEnvironment task_environment_;

  network::TestURLLoaderFactory test_url_loader_factory_;
  scoped_refptr<network::SharedURLLoaderFactory> test_shared_loader_factory_;
  std::unique_ptr<SpellChecker> spell_checker_;

  base::ScopedTempDir fake_dict_dir_;
  std::unique_ptr<base::ScopedPathOverride> path_override_;
};

TEST_F(SpellCheckerTest, ShouldNotSetupBeforePrefsInitialized) {
  EXPECT_FALSE(fake_quick_answers_state()->prefs_initialized());
  EXPECT_FALSE(spell_checker()->GetSpellcheckLanguagesForTesting().size());

  fake_quick_answers_state()->SetSettingsEnabled(true);
  fake_quick_answers_state()->SetApplicationLocale("en");
  fake_quick_answers_state()->SetPreferredLanguages("en,es");

  EXPECT_FALSE(spell_checker()->GetSpellcheckLanguagesForTesting().size());
}

TEST_F(SpellCheckerTest, ShouldNotSetupIfFeatureDisabled) {
  EXPECT_FALSE(fake_quick_answers_state()->prefs_initialized());
  EXPECT_FALSE(spell_checker()->GetSpellcheckLanguagesForTesting().size());

  fake_quick_answers_state()->OnPrefsInitialized();
  fake_quick_answers_state()->AsyncSetConsentStatus(
      prefs::ConsentStatus::kAccepted);
  // `SetSettingsEnabled` must be after `AsyncSetConsentStatus` as
  // `AsyncSetConsentStatus` changes enabled state.
  fake_quick_answers_state()->SetSettingsEnabled(false);
  fake_quick_answers_state()->SetApplicationLocale("en");
  fake_quick_answers_state()->SetPreferredLanguages("en,es");

  EXPECT_FALSE(spell_checker()->GetSpellcheckLanguagesForTesting().size());
}

TEST_F(SpellCheckerTest, ShouldResetOnFeatureDisabled) {
  EXPECT_FALSE(fake_quick_answers_state()->prefs_initialized());
  EXPECT_FALSE(spell_checker()->GetSpellcheckLanguagesForTesting().size());

  fake_quick_answers_state()->OnPrefsInitialized();
  fake_quick_answers_state()->SetSettingsEnabled(true);
  fake_quick_answers_state()->AsyncSetConsentStatus(
      prefs::ConsentStatus::kAccepted);
  fake_quick_answers_state()->SetApplicationLocale("en");
  fake_quick_answers_state()->SetPreferredLanguages("en,es");

  EXPECT_TRUE(spell_checker()->GetSpellcheckLanguagesForTesting().size());

  fake_quick_answers_state()->SetSettingsEnabled(false);

  EXPECT_FALSE(spell_checker()->GetSpellcheckLanguagesForTesting().size());
}

TEST_F(SpellCheckerTest, ShouldSetupIfShouldShowUserConsent) {
  EXPECT_FALSE(fake_quick_answers_state()->prefs_initialized());
  EXPECT_FALSE(spell_checker()->GetSpellcheckLanguagesForTesting().size());

  fake_quick_answers_state()->OnPrefsInitialized();
  fake_quick_answers_state()->SetSettingsEnabled(false);
  // We should show user consent UI if consent status is kUnknown.
  fake_quick_answers_state()->AsyncSetConsentStatus(
      prefs::ConsentStatus::kUnknown);
  fake_quick_answers_state()->SetApplicationLocale("en");
  fake_quick_answers_state()->SetPreferredLanguages("en,es");

  EXPECT_TRUE(spell_checker()->GetSpellcheckLanguagesForTesting().size());
}

TEST_F(SpellCheckerTest, ShouldResetOnUserConsentRejected) {
  EXPECT_FALSE(fake_quick_answers_state()->prefs_initialized());
  EXPECT_FALSE(spell_checker()->GetSpellcheckLanguagesForTesting().size());

  fake_quick_answers_state()->OnPrefsInitialized();
  fake_quick_answers_state()->SetSettingsEnabled(false);
  // We should show user consent UI if consent status is kUnknown.
  fake_quick_answers_state()->AsyncSetConsentStatus(
      prefs::ConsentStatus::kUnknown);
  fake_quick_answers_state()->SetApplicationLocale("en");
  fake_quick_answers_state()->SetPreferredLanguages("en,es");

  EXPECT_TRUE(spell_checker()->GetSpellcheckLanguagesForTesting().size());

  fake_quick_answers_state()->AsyncSetConsentStatus(
      prefs::ConsentStatus::kRejected);

  EXPECT_FALSE(spell_checker()->GetSpellcheckLanguagesForTesting().size());
}

TEST_F(SpellCheckerTest, ShouldNotSetupWithUnsupportedApplicationLocale) {
  EXPECT_FALSE(fake_quick_answers_state()->prefs_initialized());
  EXPECT_FALSE(spell_checker()->GetSpellcheckLanguagesForTesting().size());

  fake_quick_answers_state()->OnPrefsInitialized();
  fake_quick_answers_state()->SetSettingsEnabled(true);
  fake_quick_answers_state()->AsyncSetConsentStatus(
      prefs::ConsentStatus::kAccepted);
  fake_quick_answers_state()->SetApplicationLocale("zh");
  fake_quick_answers_state()->SetPreferredLanguages("zh,en");

  EXPECT_FALSE(spell_checker()->GetSpellcheckLanguagesForTesting().size());
}

TEST_F(SpellCheckerTest, ShouldSetupWithSupportedApplicationLocale) {
  EXPECT_FALSE(fake_quick_answers_state()->prefs_initialized());
  EXPECT_FALSE(spell_checker()->GetSpellcheckLanguagesForTesting().size());

  fake_quick_answers_state()->OnPrefsInitialized();
  fake_quick_answers_state()->SetSettingsEnabled(true);
  fake_quick_answers_state()->AsyncSetConsentStatus(
      prefs::ConsentStatus::kAccepted);
  fake_quick_answers_state()->SetApplicationLocale("en");
  fake_quick_answers_state()->SetPreferredLanguages("en,es");

  EXPECT_TRUE(spell_checker()->GetSpellcheckLanguagesForTesting().size());
}

TEST_F(SpellCheckerTest, ShouldFilterCountryCodeOfApplicationLocale) {
  EXPECT_FALSE(fake_quick_answers_state()->prefs_initialized());
  EXPECT_FALSE(spell_checker()->GetSpellcheckLanguagesForTesting().size());

  fake_quick_answers_state()->OnPrefsInitialized();
  fake_quick_answers_state()->SetSettingsEnabled(true);
  fake_quick_answers_state()->AsyncSetConsentStatus(
      prefs::ConsentStatus::kAccepted);
  fake_quick_answers_state()->SetApplicationLocale("en-US");
  fake_quick_answers_state()->SetPreferredLanguages("en-US,en-GB");

  EXPECT_EQ(spell_checker()->GetSpellcheckLanguagesForTesting().size(), 1u);
  EXPECT_EQ(spell_checker()->GetSpellcheckLanguagesForTesting()[0]->language(),
            "en");
}

TEST_F(SpellCheckerTest, ShouldSetupWithPreferredLanguages) {
  EXPECT_FALSE(fake_quick_answers_state()->prefs_initialized());
  EXPECT_FALSE(spell_checker()->GetSpellcheckLanguagesForTesting().size());

  fake_quick_answers_state()->OnPrefsInitialized();
  fake_quick_answers_state()->SetSettingsEnabled(true);
  fake_quick_answers_state()->AsyncSetConsentStatus(
      prefs::ConsentStatus::kAccepted);
  fake_quick_answers_state()->SetApplicationLocale("en");
  fake_quick_answers_state()->SetPreferredLanguages("en,es");

  EXPECT_EQ(spell_checker()->GetSpellcheckLanguagesForTesting().size(), 2u);
  EXPECT_EQ(spell_checker()->GetSpellcheckLanguagesForTesting()[0]->language(),
            "en");
  EXPECT_EQ(spell_checker()->GetSpellcheckLanguagesForTesting()[1]->language(),
            "es");

  fake_quick_answers_state()->SetPreferredLanguages("en,es,it");

  EXPECT_EQ(spell_checker()->GetSpellcheckLanguagesForTesting().size(), 3u);
  EXPECT_EQ(spell_checker()->GetSpellcheckLanguagesForTesting()[0]->language(),
            "en");
  EXPECT_EQ(spell_checker()->GetSpellcheckLanguagesForTesting()[1]->language(),
            "es");
  EXPECT_EQ(spell_checker()->GetSpellcheckLanguagesForTesting()[2]->language(),
            "it");
}

TEST_F(SpellCheckerTest, ShouldFilterUnsupportedPreferredLanguages) {
  EXPECT_FALSE(fake_quick_answers_state()->prefs_initialized());
  EXPECT_FALSE(spell_checker()->GetSpellcheckLanguagesForTesting().size());

  fake_quick_answers_state()->OnPrefsInitialized();
  fake_quick_answers_state()->SetSettingsEnabled(true);
  fake_quick_answers_state()->AsyncSetConsentStatus(
      prefs::ConsentStatus::kAccepted);
  fake_quick_answers_state()->SetApplicationLocale("en");
  fake_quick_answers_state()->SetPreferredLanguages("en,zh,es");

  EXPECT_EQ(spell_checker()->GetSpellcheckLanguagesForTesting().size(), 2u);
  EXPECT_EQ(spell_checker()->GetSpellcheckLanguagesForTesting()[0]->language(),
            "en");
  EXPECT_EQ(spell_checker()->GetSpellcheckLanguagesForTesting()[1]->language(),
            "es");
}

TEST_F(SpellCheckerTest, ShouldUseQuickAnswersDictionaryDirectory) {
  EXPECT_FALSE(fake_quick_answers_state()->prefs_initialized());
  EXPECT_FALSE(spell_checker()->GetSpellcheckLanguagesForTesting().size());

  base::FilePath quick_answers_dict_dir;
  ASSERT_TRUE(base::PathService::Get(chrome::DIR_APP_DICTIONARIES,
                                     &quick_answers_dict_dir));
  quick_answers_dict_dir = quick_answers_dict_dir.AppendASCII("quick_answers");

  EXPECT_FALSE(base::PathExists(quick_answers_dict_dir));

  fake_quick_answers_state()->OnPrefsInitialized();
  fake_quick_answers_state()->SetSettingsEnabled(true);
  fake_quick_answers_state()->AsyncSetConsentStatus(
      prefs::ConsentStatus::kAccepted);
  fake_quick_answers_state()->SetApplicationLocale("en");
  fake_quick_answers_state()->SetPreferredLanguages("en,es,it");

  // Wait for the dictionary directory to be created.
  content::RunAllTasksUntilIdle();

  EXPECT_TRUE(base::PathExists(quick_answers_dict_dir));
}

}  // namespace quick_answers