from pydantic_factories import ModelFactory

from pynyaata.cache import client
from pynyaata.types import RemoteFile


class RemoteFileFactory(ModelFactory[RemoteFile]):
    __model__ = RemoteFile


def test_cache_data():
    remote = RemoteFileFactory.build()
    client.set("test", remote)
    assert client.get("test") == remote