union test cases (#204)

ash-project · Jun 6, 2024 · 8f9209e · 8f9209e
1 parent 3ae44d7
commit 8f9209e
Show file tree

Hide file tree

Showing 2 changed files with 61 additions and 17 deletions.
diff --git a/test/auto_form_test.exs b/test/auto_form_test.exs
@@ -117,23 +117,67 @@ defmodule AshPhoenix.AutoFormTest do
                |> Map.get(:union)
     end
 
+    test "simple unions with same value" do
+      assert %Ash.Union{type: :predefined, value: :update2} =
+               SimplePost
+               |> AshPhoenix.Form.for_create(:create,
+                 domain: Domain,
+                 forms: [
+                   auto?: true
+                 ],
+                 params: %{
+                   "text" => "foobar"
+                 }
+               )
+               |> AshPhoenix.Form.add_form(:union,
+                 params: %{"_union_type" => "predefined", "value" => "update"}
+               )
+               |> AshPhoenix.Form.submit!(
+                 params: %{"union" => %{"_union_type" => "predefined", "value" => "update2"}}
+               )
+               |> Map.get(:union)
+    end
+
+    test "simple unions with cutom value" do
+      assert %Ash.Union{type: :custom, value: "update"} =
+               SimplePost
+               |> AshPhoenix.Form.for_create(:create,
+                 domain: Domain,
+                 forms: [
+                   auto?: true
+                 ],
+                 params: %{
+                   "text" => "foobar"
+                 }
+               )
+               |> AshPhoenix.Form.add_form(:union,
+                 params: %{"_union_type" => "predefined"}
+               )
+               |> AshPhoenix.Form.submit!(
+                 params: %{"union" => %{"_union_type" => "custom", "value" => "update"}}
+               )
+               |> Map.get(:union)
+    end
+
     test "simple unions with invalid values" do
-      assert_raise Ash.Error.Invalid, ~r/atom must be one of "update", got: :create/, fn ->
-        SimplePost
-        |> AshPhoenix.Form.for_create(:create,
-          domain: Domain,
-          forms: [
-            auto?: true
-          ],
-          params: %{
-            "text" => "foobar"
-          }
-        )
-        |> AshPhoenix.Form.add_form(:union,
-          params: %{"_union_type" => "predefined", "value" => "create"}
-        )
-        |> AshPhoenix.Form.submit!()
-      end
+      assert_raise Ash.Error.Invalid,
+                   ~r/atom must be one of "update, update2", got: :create/,
+                   fn ->
+                     SimplePost
+                     |> AshPhoenix.Form.for_create(:create,
+                       domain: Domain,
+                       forms: [
+                         auto?: true
+                       ],
+                       params: %{
+                         "text" => "foobar"
+                       }
+                     )
+                     |> AshPhoenix.Form.add_form(:union,
+                       params: %{"_union_type" => "predefined", "value" => "create"}
+                     )
+                     |> AshPhoenix.Form.submit!()
+                   end
     end
 
     test "deeply nested unions" do

diff --git a/test/support/resources/simple_post.ex b/test/support/resources/simple_post.ex
@@ -8,7 +8,7 @@ defmodule AshPhoenix.Test.SimplePost.SimpleUnion do
         ],
         predefined: [
           type: :atom,
-          constraints: [one_of: [:update]]
+          constraints: [one_of: [:update, :update2]]
         ]
       ]
     ]