import pytest

from pydantic import BaseModel
from datetime import datetime
from typing import Optional

from selfprivacy_api.utils.redis_model_storage import store_model_as_hash, hash_as_model
from selfprivacy_api.utils.redis_pool import RedisPool

TEST_KEY = "model_storage"
redis = RedisPool().get_connection()

@pytest.fixture()
def clean_redis():
    redis.delete(TEST_KEY)


class DummyModel(BaseModel):
    name: str
    date: Optional[datetime]

def test_store_retrieve():
    model = DummyModel(
        name= "test",
        date= datetime.now()
    )
    store_model_as_hash(redis, TEST_KEY, model)
    assert hash_as_model(redis, TEST_KEY, DummyModel) == model 

def test_store_retrieve_none():
    model = DummyModel(
        name= "test",
        date= None
    )
    store_model_as_hash(redis, TEST_KEY, model)
    assert hash_as_model(redis, TEST_KEY, DummyModel) == model