frontend/utils/src/browser/index.test.ts - develop - vshmidt/streamlit

frontend/utils/src/browser/index.test.ts
95 строк3 KB
Ken McGrady
Update copyright header to 2026 (#13491)
02 янв 2026, 16:21
02 янв 2026, 16:21374540a
/**
 * Copyright (c) Streamlit Inc. (2018-2022) Snowflake Inc. (2022-2026)
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import { getCookie, localStorageAvailable } from "."

describe("browser", () => {
  describe("localStorageAvailable", () => {
    // NOTE: localStorage is weird, and calling .spyOn(window.localStorage, "setItem")
    // doesn't work. Accessing .__proto__ here isn't too bad of a crime since
    // it's test code.
    const breakLocalStorage = (): void => {
      vi
        // eslint-disable-next-line no-proto
        .spyOn(window.localStorage.__proto__, "setItem")
        .mockImplementation(() => {
          throw new Error("boom")
        })
    }

    afterEach(() => {
      vi.restoreAllMocks()
      window.localStorage.clear()
    })

    it("returns false if a localStorage function explodes", () => {
      breakLocalStorage()
      expect(localStorageAvailable()).toBe(false)
    })

    it("returns true if all localStorage functions work", () => {
      expect(localStorageAvailable()).toBe(true)
    })
  })

  describe("getCookie", () => {
    afterEach(() => {
      document.cookie.split(";").forEach(cookie => {
        const eqPos = cookie.indexOf("=")
        const name = eqPos > -1 ? cookie.substr(0, eqPos) : cookie
        document.cookie = `${name}=;expires=Thu, 01 Jan 1970 00:00:00 GMT`
      })
    })

    it("get existing cookie", () => {
      document.cookie = "flavor=chocolatechip"
      const cookie = getCookie("flavor")
      expect(cookie).toEqual("chocolatechip")
    })

    it("get missing cookie", () => {
      document.cookie = "sweetness=medium;"
      document.cookie = "flavor=chocolatechip;"
      document.cookie = "type=darkchocolate;"
      const cookie = getCookie("recipe")
      expect(cookie).toEqual(undefined)
    })

    it("find cookie in the front", () => {
      document.cookie = "flavor=chocolatechip;"
      document.cookie = "sweetness=medium;"
      document.cookie = "type=darkchocolate;"
      const cookie = getCookie("flavor")
      expect(cookie).toEqual("chocolatechip")
    })

    it("find cookie in the middle", () => {
      document.cookie = "sweetness=medium;"
      document.cookie = "flavor=chocolatechip;"
      document.cookie = "type=darkchocolate;"
      const cookie = getCookie("flavor")
      expect(cookie).toEqual("chocolatechip")
    })

    it("find cookie in the end", () => {
      document.cookie = "sweetness=medium;"
      document.cookie = "type=darkchocolate;"
      document.cookie = "flavor=chocolatechip;"
      const cookie = getCookie("flavor")
      expect(cookie).toEqual("chocolatechip")
    })
  })
})