Upstream Chrome on iOS source code [8/11].

ios/chrome/browser/ui/settings/translate_collection_view_controller_unittest.mm

Index: ios/chrome/browser/ui/settings/translate_collection_view_controller_unittest.mm
diff --git a/ios/chrome/browser/ui/settings/translate_collection_view_controller_unittest.mm b/ios/chrome/browser/ui/settings/translate_collection_view_controller_unittest.mm
new file mode 100644
index 0000000000000000000000000000000000000000..cde3e2778950eaa827abcb5a85c3f89b2e33855f
--- /dev/null
+++ b/ios/chrome/browser/ui/settings/translate_collection_view_controller_unittest.mm
@@ -0,0 +1,114 @@
+// Copyright 2015 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.
+
+#import "ios/chrome/browser/ui/settings/translate_collection_view_controller.h"
+
+#include <memory>
+
+#include "base/compiler_specific.h"
+#include "base/files/file_path.h"
+#include "base/message_loop/message_loop.h"
+#include "base/threading/thread_task_runner_handle.h"
+#include "components/pref_registry/pref_registry_syncable.h"
+#include "components/prefs/pref_member.h"
+#include "components/prefs/pref_service.h"
+#include "components/strings/grit/components_locale_settings.h"
+#include "components/sync_preferences/pref_service_mock_factory.h"
+#include "components/translate/core/browser/translate_prefs.h"
+#include "components/translate/core/common/translate_pref_names.h"
+#include "ios/chrome/browser/pref_names.h"
+#import "ios/chrome/browser/translate/chrome_ios_translate_client.h"
+#import "ios/chrome/browser/ui/collection_view/collection_view_controller_test.h"
+#include "ios/chrome/grit/ios_strings.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "testing/gtest_mac.h"
+#include "testing/platform_test.h"
+#include "ui/base/l10n/l10n_util.h"
+#include "ui/base/l10n/l10n_util_mac.h"
+
+using user_prefs::PrefRegistrySyncable;
+
+namespace {
+
+const char kBlacklistedSite[] = "http://blacklistedsite.com";
+const char kLanguage1[] = "klingon";
+const char kLanguage2[] = "pirate";
+
+class TranslateCollectionViewControllerTest
+    : public CollectionViewControllerTest {
+ protected:
+  void SetUp() override {
+    CollectionViewControllerTest::SetUp();
+    pref_service_ = CreateLocalState();
+  }
+
+  CollectionViewController* NewController() override NS_RETURNS_RETAINED {
+    return [[TranslateCollectionViewController alloc]
+        initWithPrefs:pref_service_.get()];
+  }
+
+  std::unique_ptr<PrefService> CreateLocalState() {
+    scoped_refptr<PrefRegistrySyncable> registry = new PrefRegistrySyncable();
+    registry->RegisterBooleanPref(prefs::kEnableTranslate, false,
+                                  PrefRegistrySyncable::SYNCABLE_PREF);
+    translate::TranslatePrefs::RegisterProfilePrefs(registry.get());
+    registry->RegisterStringPref(
+        prefs::kAcceptLanguages,
+        l10n_util::GetStringUTF8(IDS_ACCEPT_LANGUAGES));
+    base::FilePath path("TranslateCollectionViewControllerTest.pref");
+    sync_preferences::PrefServiceMockFactory factory;
+    factory.SetUserPrefsFile(path, base::ThreadTaskRunnerHandle::Get().get());
+    return factory.Create(registry.get());
+  }
+
+  base::MessageLoopForUI message_loop_;
+  std::unique_ptr<PrefService> pref_service_;
+};
+
+TEST_F(TranslateCollectionViewControllerTest, TestModelTranslateOff) {
+  CreateController();
+  CheckController();
+  EXPECT_EQ(2, NumberOfSections());
+  EXPECT_EQ(2, NumberOfItemsInSection(0));
+  CheckSwitchCellStateAndTitleWithId(NO, IDS_IOS_TRANSLATE_SETTING, 0, 0);
+  CheckTextCellTitleWithId(IDS_IOS_TRANSLATE_SETTING_RESET, 0, 1);
+}
+
+TEST_F(TranslateCollectionViewControllerTest, TestModelTranslateOn) {
+  BooleanPrefMember translateEnabled;
+  translateEnabled.Init(prefs::kEnableTranslate, pref_service_.get());
+  translateEnabled.SetValue(true);
+  CreateController();
+  EXPECT_EQ(2, NumberOfSections());
+  EXPECT_EQ(2, NumberOfItemsInSection(0));
+  CheckSwitchCellStateAndTitleWithId(YES, IDS_IOS_TRANSLATE_SETTING, 0, 0);
+  CheckTextCellTitleWithId(IDS_IOS_TRANSLATE_SETTING_RESET, 0, 1);
+}
+
+TEST_F(TranslateCollectionViewControllerTest, TestClearPreferences) {
+  // Set some preferences.
+  std::unique_ptr<translate::TranslatePrefs> translate_prefs(
+      ChromeIOSTranslateClient::CreateTranslatePrefs(pref_service_.get()));
+  translate_prefs->BlacklistSite(kBlacklistedSite);
+  ASSERT_TRUE(translate_prefs->IsSiteBlacklisted(kBlacklistedSite));
+  translate_prefs->BlockLanguage(kLanguage1);
+  ASSERT_TRUE(translate_prefs->IsBlockedLanguage(kLanguage1));
+  translate_prefs->WhitelistLanguagePair(kLanguage1, kLanguage2);
+  ASSERT_TRUE(
+      translate_prefs->IsLanguagePairWhitelisted(kLanguage1, kLanguage2));
+  // Reset the preferences through the UI.
+  CreateController();
+  TranslateCollectionViewController* controller =
+      static_cast<TranslateCollectionViewController*>(this->controller());
+  // Simulate a tap on the "reset" item.
+  [controller collectionView:[controller collectionView]
+      didSelectItemAtIndexPath:[NSIndexPath indexPathForItem:1 inSection:0]];
+  // Check that preferences are gone.
+  EXPECT_FALSE(translate_prefs->IsSiteBlacklisted(kBlacklistedSite));
+  EXPECT_FALSE(translate_prefs->IsBlockedLanguage(kLanguage1));
+  EXPECT_FALSE(
+      translate_prefs->IsLanguagePairWhitelisted(kLanguage1, kLanguage2));
+}
+
+}  // namespace