from src.utils.database import invoke_redis_lambda

from dagster import (
    OpExecutionContext,
    asset,
)


@asset(
    name="test_redis",
    description="Test Redis",
    group_name="test",
    tags={"tech-alerts": "true"},
)
def test_redis(context: OpExecutionContext):
    test_set = invoke_redis_lambda(
        operation="set_json",
        keys=["test"],
        values=[{"name": "test", "id": 1}],
    )
    context.log.info(test_set)
    test_get = invoke_redis_lambda(
        operation="get",
        keys=["test"],
    )
    context.log.info(test_get)
    test_delete = invoke_redis_lambda(
        operation="delete",
        keys=["test"],
    )
    context.log.info(test_delete)
    test_get_django_model = invoke_redis_lambda(
        operation="get",
        keys=["song:1:5b2984c0-c33e-44ce-a7bc-829796429913"],
    )
    context.log.info(test_get_django_model)


test_asset = test_redis