init commit

test/diffuser/generator/prompt_request_genserver_test.exs

@@ -0,0 +1,48 @@
+defmodule Diffuser.Generator.PromptRequestTest do
+  use Diffuser.DataCase
+  import Diffuser.GeneratorFixtures
+  alias Diffuser.Generator
+  alias Diffuser.Generator.PromptRequest
+  alias DiffuserWeb.Endpoint
+
+  import Phoenix.ChannelTest
+
+  # The default endpoint for testing
+  # @endpoint DiffuserWeb.Endpoint
+
+  # TODO: Write mock that calls python script that only sleeps for 300 milliseconds
+  test "starts a new prompt and can query status while it's running" do
+    prompt_request = prompt_request_fixture()
+
+    assert {:ok, _pid} =
+             Diffuser.Generator.PromptRequestGenserver.new(%{prompt_request: prompt_request})
+
+    :timer.sleep(1000)
+
+    assert %{status: "in_progress"} = Generator.get_prompt_request!(prompt_request.id)
+
+    :timer.sleep(3000)
+
+    assert %{status: "finished"} = Diffuser.Generator.get_prompt_request!(prompt_request.id)
+  end
+
+  test "properly updates when prompt is finished via websocket" do
+    %{id: id} = prompt_request = prompt_request_fixture()
+
+    Endpoint.subscribe("request:#{id}")
+
+    Diffuser.Generator.PromptRequestGenserver.new(%{prompt_request: prompt_request})
+
+    :timer.sleep(1000)
+
+    assert_broadcast("request", %{
+      prompt_request: %PromptRequest{id: ^id}
+    })
+
+    :timer.sleep(3000)
+
+    assert_broadcast("request", %{
+      prompt_request: %PromptRequest{id: ^id}
+    })
+  end
+end