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

test/diffuser/generator_test.exs

@@ -0,0 +1,92 @@
+defmodule Diffuser.GeneratorTest do
+  use Diffuser.DataCase
+
+  alias Diffuser.Generator
+  alias Diffuser.Test.FileHelper
+  alias Diffuser.Repo
+
+  describe "prompt_requests" do
+    alias Diffuser.Generator.PromptRequest
+
+    import Diffuser.GeneratorFixtures
+
+    @invalid_attrs %{prompt: nil}
+
+    test "list_prompt_requests/0 returns all prompt_requests" do
+      prompt_request = prompt_request_fixture()
+      assert Generator.list_prompt_requests() == [prompt_request]
+    end
+
+    test "get_prompt_request!/1 returns the prompt_request with given id" do
+      prompt_request = prompt_request_fixture()
+      assert Generator.get_prompt_request!(prompt_request.id) == prompt_request
+    end
+
+    test "create_prompt_request/1 with valid data creates a prompt_request" do
+      valid_attrs = %{prompt: "some prompt"}
+
+      assert {:ok, %PromptRequest{} = prompt_request} =
+               Generator.create_prompt_request(valid_attrs)
+
+      assert prompt_request.prompt == "some prompt"
+    end
+
+    test "create_prompt_request/1 with invalid data returns error changeset" do
+      assert {:error, %Ecto.Changeset{}} = Generator.create_prompt_request(@invalid_attrs)
+    end
+
+    test "update_prompt_request/2 with valid data updates the prompt_request" do
+      prompt_request = prompt_request_fixture()
+
+      update_attrs = %{
+        prompt: "some updated prompt"
+      }
+
+      assert {:ok, %PromptRequest{} = prompt_request} =
+               Generator.update_prompt_request(prompt_request, update_attrs)
+
+      assert prompt_request.prompt == "some updated prompt"
+    end
+
+    test "update_prompt_request/2 with invalid data returns error changeset" do
+      prompt_request = prompt_request_fixture()
+
+      assert {:error, %Ecto.Changeset{}} =
+               Generator.update_prompt_request(prompt_request, @invalid_attrs)
+
+      assert prompt_request == Generator.get_prompt_request!(prompt_request.id)
+    end
+
+    test "delete_prompt_request/1 deletes the prompt_request" do
+      prompt_request = prompt_request_fixture()
+      assert {:ok, %PromptRequest{}} = Generator.delete_prompt_request(prompt_request)
+      assert_raise Ecto.NoResultsError, fn -> Generator.get_prompt_request!(prompt_request.id) end
+    end
+
+    test "change_prompt_request/1 returns a prompt_request changeset" do
+      prompt_request = prompt_request_fixture()
+      assert %Ecto.Changeset{} = Generator.change_prompt_request(prompt_request)
+    end
+
+    test "begin_prompt_request/1 returns pid of a prompt request generator" do
+      prompt_request = prompt_request_fixture()
+      assert {:ok, _pid} = Generator.begin_prompt_request(prompt_request)
+    end
+
+    test "create_prompt_request_results/2 properly associates images to prompt request" do
+      %{id: prompt_request_id} = prompt_request_fixture()
+
+      images = [
+        %{file_name: "cat.jpg", binary: FileHelper.binary_image()},
+        %{file_name: "dog.jpg", binary: FileHelper.binary_image()},
+        %{file_name: "horse.jpg", binary: FileHelper.binary_image()}
+      ]
+
+      Generator.create_prompt_request_results(prompt_request_id, images)
+
+      %PromptRequest{images: images} = Generator.get_prompt_request!(prompt_request_id)
+      assert images |> length() > 0
+      # assert images |> Enum.all?
+    end
+  end
+end