| // 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 "base/strings/nullable_string16.h" |
| #include "base/strings/string16.h" |
| #include "base/strings/string_util.h" |
| #include "base/strings/utf_string_conversions.h" |
| #include "chrome/browser/webshare/share_target_pref_helper.h" |
| #include "chrome/common/pref_names.h" |
| #include "chrome/test/base/testing_profile.h" |
| #include "components/prefs/pref_registry_simple.h" |
| #include "components/prefs/testing_pref_service.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| class PrefRegistrySimple; |
| |
| namespace { |
| |
| class ShareTargetPrefHelperUnittest : public testing::Test { |
| protected: |
| ShareTargetPrefHelperUnittest() {} |
| ~ShareTargetPrefHelperUnittest() override {} |
| |
| void SetUp() override { |
| pref_service_.reset(new TestingPrefServiceSimple()); |
| pref_service_->registry()->RegisterDictionaryPref( |
| prefs::kWebShareVisitedTargets); |
| } |
| |
| PrefService* pref_service() { return pref_service_.get(); } |
| |
| private: |
| std::unique_ptr<TestingPrefServiceSimple> pref_service_; |
| }; |
| |
| constexpr char kNameKey[] = "name"; |
| constexpr char kUrlTemplateKey[] = "url_template"; |
| |
| TEST_F(ShareTargetPrefHelperUnittest, AddMultipleShareTargets) { |
| // Add a share target to prefs that wasn't previously stored. |
| GURL manifest_url("https://www.sharetarget.com/manifest.json"); |
| content::Manifest::ShareTarget share_target; |
| std::string url_template = "share/?title={title}"; |
| share_target.url_template = |
| base::NullableString16(base::ASCIIToUTF16(url_template), false); |
| content::Manifest manifest; |
| manifest.share_target = |
| base::Optional<content::Manifest::ShareTarget>(share_target); |
| |
| UpdateShareTargetInPrefs(manifest_url, manifest, pref_service()); |
| |
| const base::DictionaryValue* share_target_dict = |
| pref_service()->GetDictionary(prefs::kWebShareVisitedTargets); |
| EXPECT_EQ(1UL, share_target_dict->size()); |
| const base::DictionaryValue* share_target_info_dict = nullptr; |
| ASSERT_TRUE(share_target_dict->GetDictionaryWithoutPathExpansion( |
| manifest_url.spec(), &share_target_info_dict)); |
| EXPECT_EQ(1UL, share_target_info_dict->size()); |
| std::string url_template_in_dict; |
| EXPECT_TRUE(share_target_info_dict->GetString(kUrlTemplateKey, |
| &url_template_in_dict)); |
| EXPECT_EQ(url_template, url_template_in_dict); |
| |
| // Add second share target to prefs that wasn't previously stored. |
| manifest_url = GURL("https://www.sharetarget2.com/manifest.json"); |
| std::string name = "Share Target Name"; |
| manifest.name = base::NullableString16(base::ASCIIToUTF16(name), false); |
| |
| UpdateShareTargetInPrefs(manifest_url, manifest, pref_service()); |
| |
| share_target_dict = |
| pref_service()->GetDictionary(prefs::kWebShareVisitedTargets); |
| EXPECT_EQ(2UL, share_target_dict->size()); |
| ASSERT_TRUE(share_target_dict->GetDictionaryWithoutPathExpansion( |
| manifest_url.spec(), &share_target_info_dict)); |
| EXPECT_EQ(2UL, share_target_info_dict->size()); |
| EXPECT_TRUE(share_target_info_dict->GetString(kUrlTemplateKey, |
| &url_template_in_dict)); |
| EXPECT_EQ(url_template, url_template_in_dict); |
| std::string name_in_dict; |
| EXPECT_TRUE(share_target_info_dict->GetString(kNameKey, &name_in_dict)); |
| EXPECT_EQ(name, name_in_dict); |
| } |
| |
| TEST_F(ShareTargetPrefHelperUnittest, AddShareTargetTwice) { |
| const char kManifestUrl[] = "https://www.sharetarget.com/manifest.json"; |
| const char kUrlTemplate[] = "share/?title={title}"; |
| |
| // Add a share target to prefs that wasn't previously stored. |
| GURL manifest_url(kManifestUrl); |
| content::Manifest::ShareTarget share_target; |
| share_target.url_template = |
| base::NullableString16(base::ASCIIToUTF16(kUrlTemplate), false); |
| content::Manifest manifest; |
| manifest.share_target = |
| base::Optional<content::Manifest::ShareTarget>(share_target); |
| |
| UpdateShareTargetInPrefs(manifest_url, manifest, pref_service()); |
| |
| const base::DictionaryValue* share_target_dict = |
| pref_service()->GetDictionary(prefs::kWebShareVisitedTargets); |
| EXPECT_EQ(1UL, share_target_dict->size()); |
| const base::DictionaryValue* share_target_info_dict = nullptr; |
| ASSERT_TRUE(share_target_dict->GetDictionaryWithoutPathExpansion( |
| kManifestUrl, &share_target_info_dict)); |
| EXPECT_EQ(1UL, share_target_info_dict->size()); |
| std::string url_template_in_dict; |
| EXPECT_TRUE(share_target_info_dict->GetString(kUrlTemplateKey, |
| &url_template_in_dict)); |
| EXPECT_EQ(kUrlTemplate, url_template_in_dict); |
| |
| // Add same share target to prefs that was previously stored; shouldn't |
| // duplicate it. |
| UpdateShareTargetInPrefs(manifest_url, manifest, pref_service()); |
| |
| share_target_dict = |
| pref_service()->GetDictionary(prefs::kWebShareVisitedTargets); |
| EXPECT_EQ(1UL, share_target_dict->size()); |
| ASSERT_TRUE(share_target_dict->GetDictionaryWithoutPathExpansion( |
| kManifestUrl, &share_target_info_dict)); |
| EXPECT_EQ(1UL, share_target_info_dict->size()); |
| EXPECT_TRUE(share_target_info_dict->GetString(kUrlTemplateKey, |
| &url_template_in_dict)); |
| EXPECT_EQ(kUrlTemplate, url_template_in_dict); |
| } |
| |
| TEST_F(ShareTargetPrefHelperUnittest, UpdateShareTarget) { |
| // Add a share target to prefs that wasn't previously stored. |
| GURL manifest_url("https://www.sharetarget.com/manifest.json"); |
| content::Manifest::ShareTarget share_target; |
| std::string url_template = "share/?title={title}"; |
| share_target.url_template = |
| base::NullableString16(base::ASCIIToUTF16(url_template), false); |
| content::Manifest manifest; |
| manifest.share_target = |
| base::Optional<content::Manifest::ShareTarget>(share_target); |
| |
| UpdateShareTargetInPrefs(manifest_url, manifest, pref_service()); |
| |
| const base::DictionaryValue* share_target_dict = |
| pref_service()->GetDictionary(prefs::kWebShareVisitedTargets); |
| EXPECT_EQ(1UL, share_target_dict->size()); |
| const base::DictionaryValue* share_target_info_dict = nullptr; |
| ASSERT_TRUE(share_target_dict->GetDictionaryWithoutPathExpansion( |
| manifest_url.spec(), &share_target_info_dict)); |
| EXPECT_EQ(1UL, share_target_info_dict->size()); |
| std::string url_template_in_dict; |
| EXPECT_TRUE(share_target_info_dict->GetString(kUrlTemplateKey, |
| &url_template_in_dict)); |
| EXPECT_EQ(url_template, url_template_in_dict); |
| |
| // Add same share target to prefs that was previously stored, with new |
| // url_template_in_dict; should update the value. |
| url_template = "share/?title={title}&text={text}"; |
| manifest.share_target.value().url_template = |
| base::NullableString16(base::ASCIIToUTF16(url_template), false); |
| |
| UpdateShareTargetInPrefs(manifest_url, manifest, pref_service()); |
| |
| share_target_dict = |
| pref_service()->GetDictionary(prefs::kWebShareVisitedTargets); |
| EXPECT_EQ(1UL, share_target_dict->size()); |
| ASSERT_TRUE(share_target_dict->GetDictionaryWithoutPathExpansion( |
| manifest_url.spec(), &share_target_info_dict)); |
| EXPECT_EQ(1UL, share_target_info_dict->size()); |
| EXPECT_TRUE(share_target_info_dict->GetString(kUrlTemplateKey, |
| &url_template_in_dict)); |
| EXPECT_EQ(url_template, url_template_in_dict); |
| } |
| |
| TEST_F(ShareTargetPrefHelperUnittest, DontAddNonShareTarget) { |
| const char kManifestUrl[] = "https://www.dudsharetarget.com/manifest.json"; |
| const base::Optional<std::string> kUrlTemplate; |
| |
| // Don't add a site that has a null template. |
| UpdateShareTargetInPrefs(GURL(kManifestUrl), content::Manifest(), |
| pref_service()); |
| |
| const base::DictionaryValue* share_target_dict = |
| pref_service()->GetDictionary(prefs::kWebShareVisitedTargets); |
| EXPECT_EQ(0UL, share_target_dict->size()); |
| const base::DictionaryValue* share_target_info_dict = nullptr; |
| ASSERT_FALSE(share_target_dict->GetDictionaryWithoutPathExpansion( |
| kManifestUrl, &share_target_info_dict)); |
| } |
| |
| TEST_F(ShareTargetPrefHelperUnittest, RemoveShareTarget) { |
| // Add a share target to prefs that wasn't previously stored. |
| GURL manifest_url("https://www.sharetarget.com/manifest.json"); |
| content::Manifest::ShareTarget share_target; |
| std::string url_template = "share/?title={title}"; |
| share_target.url_template = |
| base::NullableString16(base::ASCIIToUTF16(url_template), false); |
| content::Manifest manifest; |
| manifest.share_target = |
| base::Optional<content::Manifest::ShareTarget>(share_target); |
| |
| UpdateShareTargetInPrefs(manifest_url, manifest, pref_service()); |
| |
| const base::DictionaryValue* share_target_dict = |
| pref_service()->GetDictionary(prefs::kWebShareVisitedTargets); |
| EXPECT_EQ(1UL, share_target_dict->size()); |
| const base::DictionaryValue* share_target_info_dict = nullptr; |
| ASSERT_TRUE(share_target_dict->GetDictionaryWithoutPathExpansion( |
| manifest_url.spec(), &share_target_info_dict)); |
| EXPECT_EQ(1UL, share_target_info_dict->size()); |
| std::string url_template_in_dict; |
| EXPECT_TRUE(share_target_info_dict->GetString(kUrlTemplateKey, |
| &url_template_in_dict)); |
| EXPECT_EQ(url_template, url_template_in_dict); |
| |
| // Share target already added now has null template. Remove from prefs. |
| manifest_url = GURL("https://www.sharetarget.com/manifest.json"); |
| |
| UpdateShareTargetInPrefs(manifest_url, content::Manifest(), pref_service()); |
| |
| share_target_dict = |
| pref_service()->GetDictionary(prefs::kWebShareVisitedTargets); |
| EXPECT_EQ(0UL, share_target_dict->size()); |
| } |
| |
| } // namespace |