93 lines
3.2 KiB
Elixir
93 lines
3.2 KiB
Elixir
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
|