spec/cookie_spec: Add tests for :remove()

Author: daurnimator

spec/cookie_spec.lua

@@ -198,6 +198,96 @@ describe("cookie module", function()
 			assert.same("", s:lookup("example.com", "/", true, true, false, "other.com", false))
 		end)
 	end)
+	it("cleans up", function()
+		local s = http_cookie.new_store()
+		assert.truthy(s:store("example.com", "/", true, true, nil, http_cookie.parse_setcookie("foo=foo; Expires=Wed, 09 Jun 2021 10:18:14 GMT")))
+		assert.same("foo", s:get("example.com", "/", "foo"))
+		s.time = function() return 9876543210 end -- set time to something after the expiry
+		s:clean()
+		assert.same(nil, s:get("example.com", "/", "foo"))
+	end)
+	describe(":remove()", function()
+		it("can remove cookies by domain", function()
+			local s = http_cookie.new_store()
+			-- Try remove on empty store
+			s:remove("example.com")
+
+			assert.truthy(s:store("example.com", "/", true, true, nil, http_cookie.parse_setcookie("foo=foo")))
+			assert.truthy(s:store("example.com", "/", true, true, nil, http_cookie.parse_setcookie("foo=other; Path=/subpath")))
+			assert.truthy(s:store("other.com", "/", true, true, nil, http_cookie.parse_setcookie("bar=bar")))
+			assert.same("foo", s:get("example.com", "/", "foo"))
+			assert.same("other", s:get("example.com", "/subpath", "foo"))
+			assert.same("bar", s:get("other.com", "/", "bar"))
+
+			s:remove("example.com")
+			assert.same(nil, s:get("example.com", "/", "foo"))
+			assert.same(nil, s:get("example.com", "/subpath", "foo"))
+			assert.same("bar", s:get("other.com", "/", "bar"))
+		end)
+		it("can remove cookies by path", function()
+			local s = http_cookie.new_store()
+			-- Try remove on empty store
+			s:remove("example.com", "/")
+
+			assert.truthy(s:store("example.com", "/", true, true, nil, http_cookie.parse_setcookie("foo=foo")))
+			assert.truthy(s:store("example.com", "/", true, true, nil, http_cookie.parse_setcookie("foo=other; Path=/subpath")))
+			assert.truthy(s:store("other.com", "/", true, true, nil, http_cookie.parse_setcookie("bar=bar")))
+			assert.truthy(s:store("example.com", "/", true, true, nil, http_cookie.parse_setcookie("qux=qux")))
+			assert.same("foo", s:get("example.com", "/", "foo"))
+			assert.same("other", s:get("example.com", "/subpath", "foo"))
+			assert.same("bar", s:get("other.com", "/", "bar"))
+			assert.same("qux", s:get("example.com", "/", "qux"))
+
+			-- Remove all names under "/" path
+			s:remove("example.com", "/")
+			assert.same(nil, s:get("example.com", "/", "foo"))
+			assert.same("other", s:get("example.com", "/subpath", "foo"))
+			assert.same("bar", s:get("other.com", "/", "bar"))
+			assert.same(nil, s:get("example.com", "/", "qux"))
+
+			-- Remove last path in domain (making domain empty)
+			s:remove("example.com", "/subpath")
+			assert.same(nil, s:get("example.com", "/", "foo"))
+			assert.same(nil, s:get("example.com", "/subpath", "foo"))
+			assert.same("bar", s:get("other.com", "/", "bar"))
+			assert.same(nil, s:get("example.com", "/", "qux"))
+		end)
+		it("can remove cookies by name", function()
+			local s = http_cookie.new_store()
+			-- Try remove on empty store
+			s:remove("example.com", "/", "foo")
+
+			assert.truthy(s:store("example.com", "/", true, true, nil, http_cookie.parse_setcookie("foo=foo")))
+			assert.truthy(s:store("example.com", "/", true, true, nil, http_cookie.parse_setcookie("foo=other; Path=/subpath")))
+			assert.truthy(s:store("other.com", "/", true, true, nil, http_cookie.parse_setcookie("bar=bar")))
+			assert.truthy(s:store("example.com", "/", true, true, nil, http_cookie.parse_setcookie("qux=qux")))
+			assert.same("foo", s:get("example.com", "/", "foo"))
+			assert.same("other", s:get("example.com", "/subpath", "foo"))
+			assert.same("bar", s:get("other.com", "/", "bar"))
+			assert.same("qux", s:get("example.com", "/", "qux"))
+
+			-- Remove just one name
+			s:remove("example.com", "/", "foo")
+			assert.same(nil, s:get("example.com", "/", "foo"))
+			assert.same("other", s:get("example.com", "/subpath", "foo"))
+			assert.same("bar", s:get("other.com", "/", "bar"))
+			assert.same("qux", s:get("example.com", "/", "qux"))
+
+			-- Remove last name in path (making path empty)
+			s:remove("example.com", "/", "qux")
+			assert.same(nil, s:get("example.com", "/", "foo"))
+			assert.same("other", s:get("example.com", "/subpath", "foo"))
+			assert.same("bar", s:get("other.com", "/", "bar"))
+			assert.same(nil, s:get("example.com", "/", "qux"))
+
+			-- Remove last name in domain (making domain empty)
+			s:remove("example.com", "/subpath", "foo")
+			assert.same(nil, s:get("example.com", "/", "foo"))
+			assert.same(nil, s:get("example.com", "/subpath", "foo"))
+			assert.same("bar", s:get("other.com", "/", "bar"))
+			assert.same(nil, s:get("example.com", "/", "qux"))
+		end)
+	end)
 	it("can bake cookies", function()
 		assert.same("foo=bar", http_cookie.bake("foo", "bar"))
 		assert.same("foo=bar; Max-Age=0", http_cookie.bake("foo", "bar", -math.huge))