chrome/browser/webshare/share_target_pref_helper_unittest.cc

// 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