Store Media gallery permissions in extension prefs.

chrome/browser/extensions/extension_prefs_unittest.cc

 // Copyright (c) 2012 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 "extension_prefs_unittest.h"
 
+#include "base/basictypes.h"
 #include "base/path_service.h"
 #include "base/scoped_temp_dir.h"
 #include "base/stl_util.h"
 #include "base/string_number_conversions.h"
 #include "base/stringprintf.h"
 #include "base/values.h"
 #include "chrome/browser/extensions/extension_prefs.h"
 #include "chrome/browser/extensions/extension_pref_value_map.h"
 #include "chrome/browser/prefs/pref_change_registrar.h"
 #include "chrome/browser/prefs/scoped_user_pref_update.h"
@@ skipping 77 matching lines @@
     EXPECT_TRUE(result == blacklist_time_);
   }
 
  private:
   Time extension_time_;
   Time blacklist_time_;
   std::string extension_id_;
 };
 TEST_F(ExtensionPrefsLastPingDay, LastPingDay) {}
 
+namespace {
+
+void AddGalleryPermission(MediaGalleryPrefId gallery, bool has_access,
+                          std::vector<MediaGalleryPermission>* vector) {
+  MediaGalleryPermission permission;
+  permission.pref_id = gallery;
+  permission.has_permission = has_access;
+  vector->push_back(permission);
+}
+
+}  // namspace
+
+// Test the MediaGalleries permissions functions.
+class MediaGalleriesPermissions : public ExtensionPrefsTest {
+ public:
+  virtual void Initialize() {
+    extension1_id_ = prefs_.AddExtensionAndReturnId("test1");
+    extension2_id_ = prefs_.AddExtensionAndReturnId("test2");
+    extension3_id_ = prefs_.AddExtensionAndReturnId("test3");
+    // Id4 isn't used to ensure that an empty permission list is ok.
+    extension4_id_ = prefs_.AddExtensionAndReturnId("test4");
+    Verify();
+
+    prefs()->SetMediaGalleryPermission(extension1_id_, 1, false);
+    AddGalleryPermission(1, false, &extension1_expectation_);
+    Verify();
+
+    prefs()->SetMediaGalleryPermission(extension1_id_, 2, true);
+    AddGalleryPermission(2, true, &extension1_expectation_);
+    Verify();
+
+    prefs()->SetMediaGalleryPermission(extension1_id_, 2, false);
+    extension1_expectation_[1].has_permission = false;
+    Verify();
+
+    prefs()->SetMediaGalleryPermission(extension2_id_, 1, true);
+    prefs()->SetMediaGalleryPermission(extension2_id_, 3, true);
+    prefs()->SetMediaGalleryPermission(extension2_id_, 4, true);
+    AddGalleryPermission(1, true, &extension2_expectation_);
+    AddGalleryPermission(3, true, &extension2_expectation_);
+    AddGalleryPermission(4, true, &extension2_expectation_);
+    Verify();
+
+    prefs()->SetMediaGalleryPermission(extension3_id_, 3, true);
+    AddGalleryPermission(3, true, &extension3_expectation_);
+    Verify();
+
+    prefs()->RemoveMediaGalleryPermissions(3);
+    extension2_expectation_.erase(extension2_expectation_.begin() + 1);
+    extension3_expectation_.erase(extension3_expectation_.begin());
+    Verify();
+  }
+
+  virtual void Verify() {
+    struct TestData {
+      std::string* id;
+      std::vector<MediaGalleryPermission>* expectation;
+    };
+
+    const TestData test_data[] = {{&extension1_id_, &extension1_expectation_},
+                                  {&extension2_id_, &extension2_expectation_},
+                                  {&extension3_id_, &extension3_expectation_},
+                                  {&extension4_id_, &extension4_expectation_}};
+    for (size_t i = 0; i < ARRAYSIZE_UNSAFE(test_data); i++) {
+      std::vector<MediaGalleryPermission> actual =
+          prefs()->GetMediaGalleryPermissions(*test_data[i].id);
+      EXPECT_EQ(test_data[i].expectation->size(), actual.size());
+      for (size_t permission_entry = 0;
+           permission_entry < test_data[i].expectation->size() &&
+               permission_entry < actual.size();
+           permission_entry++) {
+        EXPECT_EQ(test_data[i].expectation->at(permission_entry).pref_id,
+                  actual[permission_entry].pref_id);
+        EXPECT_EQ(test_data[i].expectation->at(permission_entry).has_permission,
+                  actual[permission_entry].has_permission);
+      }
+    }
+  }
+
+ private:
+  std::string extension1_id_;
+  std::string extension2_id_;
+  std::string extension3_id_;
+  std::string extension4_id_;
+
+  std::vector<MediaGalleryPermission> extension1_expectation_;
+  std::vector<MediaGalleryPermission> extension2_expectation_;
+  std::vector<MediaGalleryPermission> extension3_expectation_;
+  std::vector<MediaGalleryPermission> extension4_expectation_;
+};
+TEST_F(MediaGalleriesPermissions, MediaGalleries) {}
 
 // Tests the GetToolbarOrder/SetToolbarOrder functions.
 class ExtensionPrefsToolbarOrder : public ExtensionPrefsTest {
  public:
   virtual void Initialize() {
     list_.push_back(prefs_.AddExtensionAndReturnId("1"));
     list_.push_back(prefs_.AddExtensionAndReturnId("2"));
     list_.push_back(prefs_.AddExtensionAndReturnId("3"));
     std::vector<std::string> before_list = prefs()->GetToolbarOrder();
     EXPECT_TRUE(before_list.empty());
@@ skipping 1047 matching lines @@
     EXPECT_EQ(string16(), error16);
     EXPECT_TRUE(prefs()->UserMayModifySettings(extension_.get(), &error16));
     EXPECT_EQ(string16(), error16);
     EXPECT_FALSE(prefs()->MustRemainEnabled(extension_.get(), &error16));
     EXPECT_EQ(string16(), error16);
   }
 };
 TEST_F(ExtensionPrefsNotRequiredExtension, NotRequiredExtension) {}
 
 }  // namespace extensions