From d5e159c9b2ded3fe9b8e2b7c306474d1f2e7dc83 Mon Sep 17 00:00:00 2001
From: Mike Shoup <mike@shoup.io>
Date: Tue, 25 Jun 2019 15:55:39 -0600
Subject: [PATCH] Add $type field

---
 src/humulus/recipes.py | 3 ++-
 tests/conftest.py      | 3 +++
 tests/test_recipes.py  | 2 ++
 3 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/humulus/recipes.py b/src/humulus/recipes.py
index 931149d..74bdd61 100644
--- a/src/humulus/recipes.py
+++ b/src/humulus/recipes.py
@@ -170,7 +170,8 @@ class RecipeForm(FlaskForm):
             'name': self.name.data,
             'efficiency': str(self.efficiency.data),
             'volume': str(self.volume.data),
-            'notes': self.notes.data
+            'notes': self.notes.data,
+            '$type': 'recipe',
         }
 
         recipe['fermentables'] = [f.doc for f in self.fermentables]
diff --git a/tests/conftest.py b/tests/conftest.py
index 103d8c7..6c39411 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -40,6 +40,7 @@ def app():
         # Add a couple test recipe
         put_doc({
             '_id': 'awesome-lager',
+            '$type': 'recipe',
             'efficiency': '65',
             'name': 'Awesome Lager',
             'notes': 'Test',
@@ -49,6 +50,7 @@ def app():
         })
         put_doc({
             '_id': 'partial-yeast-recipe',
+            '$type': 'recipe',
             'efficiency': '75',
             'name': 'Partial Beer',
             'notes': 'Contains only required fields for yeast.',
@@ -63,6 +65,7 @@ def app():
         })
         put_doc({
             '_id': 'full-recipe',
+            '$type': 'recipe',
             'efficiency': '78',
             'name': 'Awesome Beer',
             'notes': 'This is a test beer that contains most possible fields.',
diff --git a/tests/test_recipes.py b/tests/test_recipes.py
index dbfa623..54e26b6 100644
--- a/tests/test_recipes.py
+++ b/tests/test_recipes.py
@@ -162,6 +162,7 @@ def test_recipe_form_doc(app):
         'notes': 'This is a test',
         'fermentables': [],
         'hops': [],
+        '$type': 'recipe',
     }
 
     ferm = FermentableForm()
@@ -192,6 +193,7 @@ def test_recipe_form_doc(app):
         'efficiency': '65',
         'volume': '5.5',
         'notes': 'This is a test',
+        '$type': 'recipe',
         'fermentables': [{
             'name': 'Test',
             'type': 'Grain',