diff --git a/src/humulus/app.py b/src/humulus/app.py index 5bf78c5..df149fb 100644 --- a/src/humulus/app.py +++ b/src/humulus/app.py @@ -46,6 +46,11 @@ def create_app(test_config=None): from . import auth app.register_blueprint(auth.bp) + # Register styles blueprint and cli commands + from . import styles + styles.init_app(app) + app.register_blueprint(styles.bp) + # Register custom filters from . import filters filters.create_filters(app) diff --git a/src/humulus/designs/recipes.json b/src/humulus/designs/recipes.json index 28aca01..6ea08ac 100644 --- a/src/humulus/designs/recipes.json +++ b/src/humulus/designs/recipes.json @@ -13,6 +13,9 @@ }, "by-type": { "map": "function (doc) {\n if (doc.$type == \"recipe\" && doc.type && doc.name) {\n emit(doc.type, doc.name)\n }\n}" + }, + "by-style": { + "map": "function (doc) {\n if (doc.$type == \"recipe\" && doc.style && doc.name) {\n emit(doc.style, doc.name)\n }\n}" } }, "lists": {}, diff --git a/src/humulus/designs/styles.json b/src/humulus/designs/styles.json new file mode 100644 index 0000000..d216671 --- /dev/null +++ b/src/humulus/designs/styles.json @@ -0,0 +1,15 @@ +{ + "_id": "_design/styles", + "language": "javascript", + "views": { + "by-category": { + "map": "function (doc) {\n if (doc.$type == \"style\") {\n category = doc._id.match(/[0-9]+|[a-zA-Z]+/g)\n category[0] = parseInt(category[0])\n emit(category, doc.name)\n }\n}" + }, + "by-name": { + "map": "function (doc) {\n if (doc.$type == \"style\") {\n emit(doc.name, doc.name)\n }\n}" + } + }, + "lists": {}, + "indexes": {}, + "shows": {} +} diff --git a/src/humulus/recipes.py b/src/humulus/recipes.py index d147532..fde93f2 100644 --- a/src/humulus/recipes.py +++ b/src/humulus/recipes.py @@ -27,7 +27,9 @@ from wtforms import (Form, StringField, DecimalField, TextAreaField, FieldList, from wtforms.validators import DataRequired, Optional from humulus.auth import login_required -from humulus.couch import get_doc_or_404, put_doc, update_doc, get_view +from humulus.couch import (get_doc, get_doc_or_404, put_doc, update_doc, + get_view) +from humulus.styles import get_styles_list bp = Blueprint('recipes', __name__, url_prefix='/recipes') @@ -169,6 +171,7 @@ class RecipeForm(FlaskForm): max_entries=20 ) yeast = FormField(YeastForm) + style = SelectField('Style', choices=[], validators=[Optional()]) @property def doc(self): @@ -182,7 +185,8 @@ class RecipeForm(FlaskForm): 'volume': str(self.volume.data), 'notes': self.notes.data, '$type': 'recipe', - 'type': self.type.data + 'type': self.type.data, + 'style': self.style.data } recipe['fermentables'] = [f.doc for f in self.fermentables] @@ -202,6 +206,7 @@ class RecipeForm(FlaskForm): self.efficiency.data = Decimal(data['efficiency']) self.volume.data = Decimal(data['volume']) self.notes.data = data['notes'] + self.style.data = data['style'] for fermentable in data['fermentables']: self.fermentables.append_entry({ @@ -282,6 +287,7 @@ def index(): @login_required def create(): form = RecipeForm() + form.style.choices = get_styles_list() if form.validate_on_submit(): response = put_doc(form.doc) @@ -308,7 +314,17 @@ def create_json(): @bp.route('/info/') def info(id): - return render_template('recipes/info.html', recipe=get_doc_or_404(id)) + recipe = get_doc_or_404(id) + + style = None + if recipe['style'] != '': + try: + style = get_doc(recipe['style']) + except KeyError: + flash('Could not find style `{}`.'.format(recipe['style']), + 'warning') + + return render_template('recipes/info.html', recipe=recipe, style=style) @bp.route('/info//json') @@ -334,6 +350,7 @@ def delete(id): def update(id): # Get the recipe from the database and validate it is the same revision form = RecipeForm() + form.style.choices = get_styles_list() recipe = get_doc_or_404(id) if form.validate_on_submit(): if recipe['_rev'] != request.args.get('rev', None): diff --git a/src/humulus/styles.py b/src/humulus/styles.py new file mode 100644 index 0000000..c7c4334 --- /dev/null +++ b/src/humulus/styles.py @@ -0,0 +1,176 @@ +"""This module has functions for working with BJCP styles.""" + +# Copyright 2019 Mike Shoup +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. + +import math +import xml.etree.ElementTree as ET + +import click +import requests +from flask import Blueprint, abort, current_app, render_template, request +from flask.cli import with_appcontext + +from humulus.auth import login_required +from humulus.couch import get_db, put_doc, get_view, get_doc_or_404 + +bp = Blueprint('styles', __name__, url_prefix='/styles') + + +def sub_to_doc(sub): + """Coverts sub (XML) to a dictionary document. + + The returned dictionary can be placed right into CouchDB if you want. + """ + doc = { + '_id': '{}'.format(sub.attrib['id']), + '$type': 'style', + 'name': sub.find('name').text, + 'aroma': sub.find('aroma').text, + 'appearance': sub.find('appearance').text, + 'flavor': sub.find('flavor').text, + 'mouthfeel': sub.find('mouthfeel').text, + 'impression': sub.find('impression').text, + 'ibu': {}, + 'og': {}, + 'fg': {}, + 'srm': {}, + 'abv': {} + } + if sub.find('comments') is not None: + doc['comments'] = sub.find('comments').text + if sub.find('history') is not None: + doc['history'] = sub.find('history').text + if sub.find('ingredients') is not None: + doc['ingredients'] = sub.find('ingredients').text + if sub.find('comparison') is not None: + doc['comparison'] = sub.find('comparison').text + if sub.find('examples') is not None: + doc['examples'] = sub.find('examples').text + if sub.find('tags') is not None: + doc['tags'] = sub.find('tags').text.split(', ') + + doc['ibu']['low'] = (sub.find('./stats/ibu/low').text + if sub.find('./stats/ibu/low') is not None else '0') + doc['ibu']['high'] = (sub.find('./stats/ibu/high').text + if sub.find('./stats/ibu/high') is not None else '100') + doc['og']['low'] = (sub.find('./stats/og/low').text + if sub.find('./stats/og/low') is not None else '1.0') + doc['og']['high'] = (sub.find('./stats/og/high').text + if sub.find('./stats/og/high') is not None else '1.2') + doc['fg']['low'] = (sub.find('./stats/fg/low').text + if sub.find('./stats/fg/low') is not None else '1.0') + doc['fg']['high'] = (sub.find('./stats/fg/high').text + if sub.find('./stats/fg/high') is not None else '1.2') + doc['srm']['low'] = (sub.find('./stats/srm/low').text + if sub.find('./stats/srm/low') is not None else '0') + doc['srm']['high'] = (sub.find('./stats/srm/high').text + if sub.find('./stats/srm/high') is not None else '100') + doc['abv']['low'] = (sub.find('./stats/abv/low').text + if sub.find('./stats/abv/low') is not None else '0') + doc['abv']['high'] = (sub.find('./stats/abv/high').text + if sub.find('./stats/abv/high') is not None else '100') + return doc + + +def import_styles(url): + """Parses BJCP styles in XML format from `url`. + + `url` defaults to the official BJCP XML styleguide. + + Each subcategory is converted to JSON and then put to a couchdb. The _id of + the subsequent document will be ``, i.e., 1A. + If the style already exists in the database, it will be skipped. + """ + db = get_db() + root = ET.fromstring(requests.get(url).text) + subs = root.findall('./class[@type="beer"]/category/subcategory') + for sub in subs: + doc = sub_to_doc(sub) + if doc['_id'] not in db: + put_doc(doc) + + +def get_styles_list(): + """Returns a list containing id and names of all styles.""" + view = get_view('_design/styles', 'by-category') + styles = [['', '']] + for row in view(include_docs=False)['rows']: + print(row) + styles.append([row['id'], '{}{} {}'.format( + row['key'][0], + row['key'][1], + row['value'] + )]) + return styles + + +@click.command('import-styles') +@with_appcontext +def import_command(): + """CLI command to import BJCP styles.""" + url = current_app.config.get( + 'BJCP_STYLES_URL', + ('https://raw.githubusercontent.com/meanphil' + '/bjcp-guidelines-2015/master/styleguide.xml') + ) + import_styles(url) + click.echo("Imported BJCP styles.") + + +def init_app(app): + """Register the CLI command with the app.""" + app.cli.add_command(import_command) + + +@bp.route('/') +@login_required +def index(): + descending = ( + request.args.get('descending', default='false', type=str).lower() in + ['true', 'yes'] + ) + sort_by = request.args.get('sort_by', default='category', type=str) + page = request.args.get('page', default=1, type=int) + limit = request.args.get('limit', default=20, type=int) + + view = get_view('_design/styles', 'by-{}'.format(sort_by)) + try: + rows = view(include_docs=True, descending=descending)['rows'] + except requests.exceptions.HTTPError: + abort(400) + + return render_template( + 'styles/index.html', + rows=rows[(page-1)*limit:page*limit], + descending=descending, + sort_by=sort_by, + page=page, + num_pages=math.ceil(len(rows)/limit), + limit=limit + ) + + +@bp.route('/info/') +@login_required +def info(id): + return render_template('styles/info.html', style=get_doc_or_404(id)) + + +@bp.route('/info//recipes') +def recipes(id): + style = get_doc_or_404(id) + view = get_view('_design/recipes', 'by-style') + rows = view(include_docs=True, descending=True, key=id)['rows'] + return render_template('styles/recipes.html', style=style, rows=rows) diff --git a/src/humulus/templates/_base.html b/src/humulus/templates/_base.html index 4b0f264..678401e 100644 --- a/src/humulus/templates/_base.html +++ b/src/humulus/templates/_base.html @@ -40,6 +40,11 @@
  • Recipes
    • + {% if session.logged_in %} +
  • + Styles +
    • + {% endif %}
    • diff --git a/src/humulus/templates/recipes/_macros.html b/src/humulus/templates/recipes/_macros.html index 6ac50ac..b4f5baf 100644 --- a/src/humulus/templates/recipes/_macros.html +++ b/src/humulus/templates/recipes/_macros.html @@ -26,9 +26,12 @@ -#}
      {{ render_field_with_errors(form.name) }}
      -
      {{ render_field_with_errors(form.type) }}
      -
      {{ render_field_with_errors(form.efficiency, 'ingredient-field') }}
      -
      {{ render_field_with_errors(form.volume, 'ingredient-field') }}
      +
      {{ render_field_with_errors(form.efficiency, 'ingredient-field') }}
      +
      {{ render_field_with_errors(form.volume, 'ingredient-field') }}
      +
      +
      +
      {{ render_field_with_errors(form.style) }}
      +
      {{ render_field_with_errors(form.type) }}
      {#- Fermentable Ingredients diff --git a/src/humulus/templates/recipes/info.html b/src/humulus/templates/recipes/info.html index 0a8695c..d55e07c 100644 --- a/src/humulus/templates/recipes/info.html +++ b/src/humulus/templates/recipes/info.html @@ -20,7 +20,11 @@ {% block body %}

      {{ recipe.name }}

      - +{% if style %} +
      +

      {{ style._id }} {{ style.name }}

      +
      +{% endif %} {#- Recipe Details -#} diff --git a/src/humulus/templates/styles/index.html b/src/humulus/templates/styles/index.html new file mode 100644 index 0000000..e4a78e6 --- /dev/null +++ b/src/humulus/templates/styles/index.html @@ -0,0 +1,84 @@ +{#- + Copyright 2019 Mike Shoup + + Licensed under the Apache License, Version 2.0 (the "License"); + you may not use this file except in compliance with the License. + You may obtain a copy of the License at + + http://www.apache.org/licenses/LICENSE-2.0 + + Unless required by applicable law or agreed to in writing, software + distributed under the License is distributed on an "AS IS" BASIS, + WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + See the License for the specific language governing permissions and + limitations under the License. +-#} + +{% extends '_base.html' %} +{% block title %}Styles{% endblock %} + +{% block body %} +

      Styles

      +
      + + + + + + + + {% for row in rows %} + + + + + {% endfor %} +
      + {% if sort_by == 'category' and descending %} + Category ↓ + {% elif sort_by == 'category' %} + Category ↑ + {% else %} + Category + {% endif %} + + {% if sort_by == 'name' and descending %} + Name ↓ + {% elif sort_by == 'name' %} + Name ↑ + {% else %} + Name + {% endif %} +
      {{ row.doc._id }}{{ row.doc.name }}
      +
      +
      + +
      + +
      + 5 + 10 + 20 + 50 + 100 + 200 +
      +
      +
      +{% endblock %} diff --git a/src/humulus/templates/styles/info.html b/src/humulus/templates/styles/info.html new file mode 100644 index 0000000..562c689 --- /dev/null +++ b/src/humulus/templates/styles/info.html @@ -0,0 +1,85 @@ +{#- + Copyright 2019 Mike Shoup + + Licensed under the Apache License, Version 2.0 (the "License"); + you may not use this file except in compliance with the License. + You may obtain a copy of the License at + + http://www.apache.org/licenses/LICENSE-2.0 + + Unless required by applicable law or agreed to in writing, software + distributed under the License is distributed on an "AS IS" BASIS, + WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + See the License for the specific language governing permissions and + limitations under the License. +-#} +{% from "_macros.html" import moment %} + +{% macro render_detail(field, header) %} +
      +

      {{ header }}

      +
      +
      +

      {{ field }}

      +
      +{% endmacro %} + + +{% extends '_base.html' %} +{% block title %}{{ style._id }} {{ style.name }}{% endblock %} + +{% block body %} +

      {{ style._id }} {{ style.name }}

      +

      View recipes using this style

      +{{ render_detail(style.impression, 'Overall Impression') }} +{{ render_detail(style.appearance, 'Appearance') }} +{{ render_detail(style.aroma, 'Aroma') }} +{{ render_detail(style.flavor, 'Flavor') }} +{{ render_detail(style.mouthfeel, 'Mouthfeel') }} + +{% if style.comments %} + {{ render_detail(style.comments, 'Comments') }} +{% endif %} +{% if style.history %} +{{ render_detail(style.history, 'History') }} +{% endif %} +{% if style.ingredients %} +{{ render_detail(style.ingredients, 'Characteristic Ingredients') }} +{% endif %} +{% if style.comparison %} +{{ render_detail(style.comparison, 'Style Comparisons') }} +{% endif %} + +
      +

      Vital Statistics

      + + + + + + + + + + + + + + + + + + + + + +
      IBU{{ style.ibu.low }} - {{ style.ibu.high }}
      OG{{ '%.3f' | format(style.og.low|float) }} - {{ '%.3f' | format(style.og.high|float) }}
      FG{{ '%.3f' | format(style.fg.low|float) }} - {{ '%.3f' | format(style.fg.high|float) }}
      SRM{{ style.srm.low }} - {{ style.srm.high }}
      ABV{{ style.abv.low }} - {{ style.abv.high }}
      +
      + +{% if style.examples %} +{{ render_detail(style.examples, 'Commercial Examples') }} +{% endif %} + +
      Back to styles list
      + +{% endblock %} diff --git a/src/humulus/templates/styles/recipes.html b/src/humulus/templates/styles/recipes.html new file mode 100644 index 0000000..55e86b2 --- /dev/null +++ b/src/humulus/templates/styles/recipes.html @@ -0,0 +1,39 @@ +{#- + Copyright 2019 Mike Shoup + + Licensed under the Apache License, Version 2.0 (the "License"); + you may not use this file except in compliance with the License. + You may obtain a copy of the License at + + http://www.apache.org/licenses/LICENSE-2.0 + + Unless required by applicable law or agreed to in writing, software + distributed under the License is distributed on an "AS IS" BASIS, + WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + See the License for the specific language governing permissions and + limitations under the License. +-#} + +{% extends '_base.html' %} +{% block title %}Recipes for {{ style._id }} {{ style.name }}{% endblock %} + +{% block body %} +{% if session.logged_in %} +

      Recipes for {{ style._id }} {{ style.name }}

      +{% else %} +

      Recipes for {{ style._id }} {{ style.name }}

      +{% endif %} + +
      + + + + + {% for row in rows %} + + + + {% endfor %} +
      Name
      {{ row.doc.name }}
      +
      +{% endblock %} diff --git a/tests/conftest.py b/tests/conftest.py index d997ccb..52bd0c3 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -50,7 +50,8 @@ def app(): 'notes': 'Test', 'volume': '5.5', 'fermentables': [], - 'hops': [] + 'hops': [], + 'style': '' }) put_doc({ '_id': 'partial-yeast-recipe', @@ -66,7 +67,8 @@ def app(): 'name': 'US-05', 'low_attenuation': '60', 'high_attenuation': '72', - } + }, + 'style': '' }) put_doc({ '_id': 'full-recipe', @@ -76,6 +78,7 @@ def app(): 'name': 'Awesome Beer', 'notes': 'This is a test beer that contains most possible fields.', 'volume': '2.5', + 'style': '1A', 'fermentables': [ { 'name': '2row', @@ -122,6 +125,23 @@ def app(): } }) + # Add a test style + put_doc({'$type': 'style', + '_id': '1A', + 'abv': {'high': '100', 'low': '0'}, + 'appearance': 'Good looking', + 'aroma': 'Smelly', + 'fg': {'high': '1.2', 'low': '1.0'}, + 'flavor': 'Good tasting', + 'ibu': {'high': '100', 'low': '0'}, + 'id': '1A', + 'impression': 'Refreshing', + 'mouthfeel': 'Good feeling', + 'name': 'Test Style', + 'og': {'high': '1.2', 'low': '1.0'}, + 'srm': {'high': '100', 'low': '0'} + }) + yield app with app.app_context(): @@ -164,6 +184,7 @@ def sample_recipes(): 'lager': { 'efficiency': '72', 'type': 'All-Grain', + 'style': '', 'fermentables': [ { 'amount': '9.5', @@ -222,6 +243,7 @@ def sample_recipes(): 'sweetstout': { 'efficiency': '72', 'type': 'All-Grain', + 'style': '', 'fermentables': [ { 'amount': '2.75', diff --git a/tests/test_recipes.py b/tests/test_recipes.py index 43f2c2f..6b1a1fd 100644 --- a/tests/test_recipes.py +++ b/tests/test_recipes.py @@ -136,6 +136,7 @@ def test_create(client, app, auth): 'name': 'Test', 'notes': 'Test', 'volume': '5.5', + 'style': '1A' } response = client.post('/recipes/create', data=data) assert response.status_code == 302 @@ -147,6 +148,7 @@ def test_create(client, app, auth): assert doc['notes'] == 'Test' assert doc['volume'] == '5.5' assert doc['efficiency'] == '65' + assert doc['style'] == '1A' def test_update(client, app, auth): @@ -205,8 +207,12 @@ def test_update(client, app, auth): assert 'Update conflict' in flash_message -def test_info(client): +def test_info(client, monkeypatch): """Test success in retrieving a recipe document.""" + def mock_get_doc(id): + # This function always raises KeyError + raise KeyError(id) + # Validate 404 response = client.get('/recipes/info/thisdoesnotexist') assert response.status_code == 404 @@ -216,6 +222,17 @@ def test_info(client): assert response.status_code == 200 assert b'Awesome Lager' in response.data + # Validate response for recipe with style + response = client.get('/recipes/info/full-recipe') + assert response.status_code == 200 + assert b'Awesome Beer' in response.data + assert b'Test Style' in response.data + + # Validate warning is flashed when style cannot be found + monkeypatch.setattr('humulus.recipes.get_doc', mock_get_doc) + response = client.get('/recipes/info/full-recipe') + assert b'Could not find style' in response.data + def test_info_json(client): """Test success in retrieving a JSON recipe.""" @@ -279,6 +296,7 @@ def test_recipe_form_doc(app): recipe.volume.data = Decimal('5.5') recipe.notes.data = 'This is a test' recipe.type.data = 'All-Grain' + recipe.style.data = '1A' assert recipe.doc == { 'name': 'Test', @@ -289,6 +307,7 @@ def test_recipe_form_doc(app): 'fermentables': [], 'hops': [], '$type': 'recipe', + 'style': '1A' } ferm = FermentableForm() @@ -321,6 +340,7 @@ def test_recipe_form_doc(app): 'volume': '5.5', 'notes': 'This is a test', '$type': 'recipe', + 'style': '1A', 'fermentables': [{ 'name': 'Test', 'type': 'Grain', diff --git a/tests/test_styles.py b/tests/test_styles.py new file mode 100644 index 0000000..80a07bc --- /dev/null +++ b/tests/test_styles.py @@ -0,0 +1,228 @@ +# Copyright 2019 Mike Shoup +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. + +import xml.etree.ElementTree as ET + +from humulus.styles import import_styles, sub_to_doc, get_styles_list + +COMPLETE_STYLE = ''' + Test Style + Smelly + Good looking + Good tasting + Good feeling + Refreshing + Comments + Old + Grains, Hops, and Water + Comparison + Examples + one, two + + + 1 + 2 + + + 1.010 + 1.020 + + + 1.000 + 1.010 + + + 1 + 2 + + + 1 + 2 + + + +''' + +INCOMPLETE_STYLE = ''' + Test Style + Smelly + Good looking + Good tasting + Good feeling + Refreshing + +''' + +TEST_XML = ''' + + + + + Test Style + Smelly + Good looking + Good tasting + Good feeling + Refreshing + + + + +''' + + +def test_sub_to_doc(): + assert sub_to_doc(ET.fromstring(COMPLETE_STYLE)) == { + '_id': '1A', + '$type': 'style', + 'name': 'Test Style', + 'aroma': 'Smelly', + 'appearance': 'Good looking', + 'flavor': 'Good tasting', + 'mouthfeel': 'Good feeling', + 'impression': 'Refreshing', + 'comments': 'Comments', + 'history': 'Old', + 'ingredients': 'Grains, Hops, and Water', + 'comparison': 'Comparison', + 'examples': 'Examples', + 'tags': ['one', 'two'], + 'ibu': {'low': '1', 'high': '2'}, + 'og': {'low': '1.010', 'high': '1.020'}, + 'fg': {'low': '1.000', 'high': '1.010'}, + 'srm': {'low': '1', 'high': '2'}, + 'abv': {'low': '1', 'high': '2'} + } + + assert sub_to_doc(ET.fromstring(INCOMPLETE_STYLE)) == { + '_id': '2B', + '$type': 'style', + 'name': 'Test Style', + 'aroma': 'Smelly', + 'appearance': 'Good looking', + 'flavor': 'Good tasting', + 'mouthfeel': 'Good feeling', + 'impression': 'Refreshing', + 'ibu': {'low': '0', 'high': '100'}, + 'og': {'low': '1.0', 'high': '1.2'}, + 'fg': {'low': '1.0', 'high': '1.2'}, + 'srm': {'low': '0', 'high': '100'}, + 'abv': {'low': '0', 'high': '100'} + } + + +def test_import_styles(monkeypatch): + class PutRecorder: + doc = None + + class MockDB: + db = {} + + def fake_get_db(): + return MockDB.db + + def fake_put_doc(doc): + PutRecorder.doc = doc + + def fake_requests_get(url): + class TestXML: + text = TEST_XML + return TestXML + + monkeypatch.setattr('requests.get', fake_requests_get) + monkeypatch.setattr('humulus.styles.get_db', fake_get_db) + monkeypatch.setattr('humulus.styles.put_doc', fake_put_doc) + + import_styles(None) + assert PutRecorder.doc == {'$type': 'style', + '_id': '1A', + 'abv': {'high': '100', 'low': '0'}, + 'appearance': 'Good looking', + 'aroma': 'Smelly', + 'fg': {'high': '1.2', 'low': '1.0'}, + 'flavor': 'Good tasting', + 'ibu': {'high': '100', 'low': '0'}, + 'impression': 'Refreshing', + 'mouthfeel': 'Good feeling', + 'name': 'Test Style', + 'og': {'high': '1.2', 'low': '1.0'}, + 'srm': {'high': '100', 'low': '0'} + } + + MockDB.db = {'1A': ''} + PutRecorder.doc = None + import_styles(None) + assert PutRecorder.doc is None + + +def test_import_command(runner, monkeypatch): + class Recorder: + called = False + + def fake_import_styles(url): + Recorder.called = True + + monkeypatch.setattr('humulus.styles.import_styles', fake_import_styles) + result = runner.invoke(args=['import-styles']) + assert Recorder.called + assert 'Imported BJCP styles.' in result.output + + +def test_get_styles_choices(app): + """Test success in getting list of styles.""" + with app.app_context(): + styles = get_styles_list() + assert styles == [ + ['', ''], + ['1A', '1A Test Style'] + ] + + +def test_index(auth, client): + """Test success in retrieving index.""" + # Test not logged in + response = client.get('/styles/') + assert response.status_code == 302 + + # Login and test get + auth.login() + response = client.get('/styles/') + assert response.status_code == 200 + assert b'1A' in response.data + assert b'Test Style' in response.data + + # Test for bad request + response = client.get('/styles/?sort_by=foobar') + assert response.status_code == 400 + + +def test_info(auth, client): + """Test success in retrieving a style's info page""" + # Test not logged in + response = client.get('/styles/info/1A') + assert response.status_code == 302 + + # Login and test + auth.login() + response = client.get('/styles/info/1A') + assert response.status_code == 200 + assert b'1A' in response.data + assert b'Test Style' in response.data + + +def test_recipes(client): + """Test success in retrieving list of recipes matching style.""" + response = client.get('/styles/info/1A/recipes') + assert response.status_code == 200 + assert b'Awesome Beer' in response.data