How to Mock a Redis Session Store Cookie SID in Python

import redis
import uuid
import time


class RedisSessionStore:
    def __init__(self, host="localhost", port=6379, db=0, prefix="session:"):
        self.client = redis.Redis(host=host, port=port, db=db)
        self.prefix = prefix

    def create_session(self, timeout_seconds=3600):
        session_id = uuid.uuid4().hex
        key = f"{self.prefix}{session_id}"
        self.client.set(key, str(time.time()), ex=timeout_seconds)
        return session_id

    def get_session_data(self, session_id):
        key = f"{self.prefix}{session_id}"
        value = self.client.get(key)
        return value.decode("utf-8") if value else None

    def delete_session(self, session_id):
        key = f"{self.prefix}{session_id}"
        return self.client.delete(key)


class CookieMock:
    def __init__(self, session_store):
        self.session_store = session_store
        self.cookies = {}

    def set_cookie(self, session_id):
        self.cookies["sid"] = session_id
        return f"sid={session_id}; HttpOnly; Path=/"

    def get_cookie(self):
        session_id = self.cookies.get("sid")
        if not session_id:
            return None
        return self.session_store.get_session_data(session_id)

    def clear_cookie(self):
        session_id = self.cookies.pop("sid", None)
        if session_id:
            self.session_store.delete_session(session_id)


if __name__ == "__main__":
    store = RedisSessionStore()
    cookie_mock = CookieMock(store)

    sid = store.create_session(timeout_seconds=120)
    print("Session created:", cookie_mock.set_cookie(sid))
    print("Retrieved from cookie:", cookie_mock.get_cookie())
    cookie_mock.clear_cookie()
    print("After clearing:", cookie_mock.get_cookie())