define the "stockable()" type as a union of meal|food|offer

openpantry · Oct 26, 2017 · 896a44d · 896a44d
1 parent e6898e3
commit 896a44d
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/lib/open_pantry/web/models/stock.ex b/lib/open_pantry/web/models/stock.ex
@@ -35,6 +35,8 @@ defmodule OpenPantry.Stock do
     timestamps()
   end
 
+  @type stockable() :: Meal.t | Food.t | Offer.t
+
   @doc """
   Builds a changeset based on the `struct` and `params`.
   """
@@ -47,7 +49,7 @@ defmodule OpenPantry.Stock do
     |> check_constraint(:quantity, name: :non_negative_quantity)
   end
 
-  @spec stockable!(Stock.t) :: Meal.t | Food.t | Offer.t
+  @spec stockable!(Stock.t) :: stockable()
   def stockable!(stock) do
     stock.food || stock.meal || stock.offer
   end
@@ -60,7 +62,7 @@ defmodule OpenPantry.Stock do
     (loaded_stock.offer && loaded_stock.offer.description)
   end
 
-  @spec stockable(Stock.t) :: Stock.t
+  @spec stockable(Stock.t) :: stockable()
   def stockable(stock) do
     stock
     |> stockable_load