init commit

test/support/conn_case.ex

@@ -0,0 +1,38 @@
+defmodule DiffuserWeb.ConnCase do
+  @moduledoc """
+  This module defines the test case to be used by
+  tests that require setting up a connection.
+
+  Such tests rely on `Phoenix.ConnTest` and also
+  import other functionality to make it easier
+  to build common data structures and query the data layer.
+
+  Finally, if the test case interacts with the database,
+  we enable the SQL sandbox, so changes done to the database
+  are reverted at the end of every test. If you are using
+  PostgreSQL, you can even run database tests asynchronously
+  by setting `use DiffuserWeb.ConnCase, async: true`, although
+  this option is not recommended for other databases.
+  """
+
+  use ExUnit.CaseTemplate
+
+  using do
+    quote do
+      # Import conveniences for testing with connections
+      import Plug.Conn
+      import Phoenix.ConnTest
+      import DiffuserWeb.ConnCase
+
+      alias DiffuserWeb.Router.Helpers, as: Routes
+
+      # The default endpoint for testing
+      @endpoint DiffuserWeb.Endpoint
+    end
+  end
+
+  setup tags do
+    Diffuser.DataCase.setup_sandbox(tags)
+    {:ok, conn: Phoenix.ConnTest.build_conn()}
+  end
+end

test/support/data_case.ex

@@ -0,0 +1,58 @@
+defmodule Diffuser.DataCase do
+  @moduledoc """
+  This module defines the setup for tests requiring
+  access to the application's data layer.
+
+  You may define functions here to be used as helpers in
+  your tests.
+
+  Finally, if the test case interacts with the database,
+  we enable the SQL sandbox, so changes done to the database
+  are reverted at the end of every test. If you are using
+  PostgreSQL, you can even run database tests asynchronously
+  by setting `use Diffuser.DataCase, async: true`, although
+  this option is not recommended for other databases.
+  """
+
+  use ExUnit.CaseTemplate
+
+  using do
+    quote do
+      alias Diffuser.Repo
+
+      import Ecto
+      import Ecto.Changeset
+      import Ecto.Query
+      import Diffuser.DataCase
+    end
+  end
+
+  setup tags do
+    Diffuser.DataCase.setup_sandbox(tags)
+    :ok
+  end
+
+  @doc """
+  Sets up the sandbox based on the test tags.
+  """
+  def setup_sandbox(tags) do
+    pid = Ecto.Adapters.SQL.Sandbox.start_owner!(Diffuser.Repo, shared: not tags[:async])
+    on_exit(fn -> Ecto.Adapters.SQL.Sandbox.stop_owner(pid) end)
+  end
+
+  @doc """
+  A helper that transforms changeset errors into a map of messages.
+
+      assert {:error, changeset} = Accounts.create_user(%{password: "short"})
+      assert "password is too short" in errors_on(changeset).password
+      assert %{password: ["password is too short"]} = errors_on(changeset)
+
+  """
+  def errors_on(changeset) do
+    Ecto.Changeset.traverse_errors(changeset, fn {message, opts} ->
+      Regex.replace(~r"%{(\w+)}", message, fn _, key ->
+        opts |> Keyword.get(String.to_existing_atom(key), key) |> to_string()
+      end)
+    end)
+  end
+end

test/support/file_helper.ex

@@ -0,0 +1,14 @@
+defmodule Diffuser.Test.FileHelper do
+  def base64_image() do
+    binary_image() |> Base.encode64()
+  end
+
+  def binary_image() do
+    "#{__DIR__}/../fixtures/image.png"
+    |> File.read!()
+  end
+
+  def base64_image(_), do: {:ok, image: base64_image()}
+
+  def image_path(), do: "#{__DIR__}/../fixtures/image.png"
+end

test/support/fixtures/generator_fixtures.ex

@@ -0,0 +1,20 @@
+defmodule Diffuser.GeneratorFixtures do
+  @moduledoc """
+  This module defines test helpers for creating
+  entities via the `Diffuser.Generator` context.
+  """
+
+  @doc """
+  Generate a prompt_request.
+  """
+  def prompt_request_fixture(attrs \\ %{}) do
+    {:ok, prompt_request} =
+      attrs
+      |> Enum.into(%{
+        prompt: "some prompt"
+      })
+      |> Diffuser.Generator.create_prompt_request()
+
+    prompt_request
+  end
+end