import uuid

from temporalio.testing import WorkflowEnvironment
from temporalio.worker import Worker

from hello.hello_query import GreetingWorkflow


async def test_query_workflow():
    task_queue_name = str(uuid.uuid4())
    # start manual time skipping
    async with await WorkflowEnvironment.start_time_skipping() as env:
        async with Worker(
            env.client, task_queue=task_queue_name, workflows=[GreetingWorkflow]
        ):
            handle = await env.client.start_workflow(
                GreetingWorkflow.run,
                "World",
                id=str(uuid.uuid4()),
                task_queue=task_queue_name,
            )

            assert "Hello, World!" == await handle.query(GreetingWorkflow.greeting)
            # manually skip 3 seconds. This will allow the workflow execution to move forward
            await env.sleep(3)
            assert "Goodbye, World!" == await handle.query(GreetingWorkflow.greeting)