FIX:Handle boolean fields

mrmicahcooper · Dec 2, 2020 · b72ecfd · b72ecfd
1 parent a257e3a
commit b72ecfd
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git a/lib/ecto_factory.ex b/lib/ecto_factory.ex
@@ -130,7 +130,7 @@ defmodule EctoFactory do
     {schema, attrs}
   end
 
-  defp gen_attribute({key, value}) when is_atom(value), do: {key, gen(value)}
+  defp gen_attribute({key, value}) when is_atom(value) and value not in [true, false], do: {key, gen(value)}
   defp gen_attribute({key, {_, _} = value}), do: {key, gen(value)}
   defp gen_attribute(key_value), do: key_value
 

diff --git a/test/ecto_factory_test.exs b/test/ecto_factory_test.exs
@@ -21,6 +21,7 @@ defmodule EctoFactoryTest do
   test "build by directly passing in a schema" do
     user = EctoFactory.schema(User)
     assert user
+    assert Enum.member?([true, false], user.admin)
   end
 
   test "build with attributes" do

diff --git a/test/support/user.ex b/test/support/user.ex
@@ -7,5 +7,6 @@ defmodule User do
     field(:date_of_birth, :date)
     field(:addresses, {:array, :string})
     field(:profile, :map)
+    field(:admin, :boolean)
   end
 end