1 files changed, 17 insertions, 0 deletions
diff --git a/net/base/cookie_monster_unittest.cc b/net/base/cookie_monster_unittest.cc
index db76cdb..5ca3e76 100644
--- a/net/base/cookie_monster_unittest.cc
+++ b/net/base/cookie_monster_unittest.cc
@@ -928,4 +928,21 @@ TEST(CookieMonsterTest, TestDeleteSingleCookie) {
   EXPECT_EQ("A=B; E=F", cm.GetCookies(url_google));
 }
 
+TEST(CookieMonsterTest, SetCookieableSchemes) {
+  net::CookieMonster cm;
+  net::CookieMonster cm_foo;
+
+  // Only cm_foo should allow foo:// cookies.
+  const char* kSchemes[] = {"foo"};
+  cm_foo.SetCookieableSchemes(kSchemes, 1);
+
+  GURL foo_url("foo://host/path");
+  GURL http_url("http://host/path");
+
+  EXPECT_TRUE(cm.SetCookie(http_url, "x=1"));
+  EXPECT_FALSE(cm.SetCookie(foo_url, "x=1"));
+  EXPECT_TRUE(cm_foo.SetCookie(foo_url, "x=1"));
+  EXPECT_FALSE(cm_foo.SetCookie(http_url, "x=1"));
+}
+
 // TODO test overwrite cookie