humulus/tests/test_styles.py

# 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 = '''<subcategory id="1A">
    <name>Test Style</name>
    <aroma>Smelly</aroma>
    <appearance>Good looking</appearance>
    <flavor>Good tasting</flavor>
    <mouthfeel>Good feeling</mouthfeel>
    <impression>Refreshing</impression>
    <comments>Comments</comments>
    <history>Old</history>
    <ingredients>Grains, Hops, and Water</ingredients>
    <comparison>Comparison</comparison>
    <examples>Examples</examples>
    <tags>one, two</tags>
    <stats>
        <ibu flexible="false">
            <low>1</low>
            <high>2</high>
        </ibu>
        <og flexible="false">
            <low>1.010</low>
            <high>1.020</high>
        </og>
        <fg flexible="false">
            <low>1.000</low>
            <high>1.010</high>
        </fg>
        <srm flexible="false">
            <low>1</low>
            <high>2</high>
        </srm>
        <abv flexible="false">
            <low>1</low>
            <high>2</high>
        </abv>
    </stats>
</subcategory>
'''

INCOMPLETE_STYLE = '''<subcategory id="2B">
    <name>Test Style</name>
    <aroma>Smelly</aroma>
    <appearance>Good looking</appearance>
    <flavor>Good tasting</flavor>
    <mouthfeel>Good feeling</mouthfeel>
    <impression>Refreshing</impression>
</subcategory>
'''

TEST_XML = '''<?xml version="1.0" encoding="UTF-8"?>
<styleguide>
    <class type="beer">
        <category id="1">
            <subcategory id="1A">
                <name>Test Style</name>
                <aroma>Smelly</aroma>
                <appearance>Good looking</appearance>
                <flavor>Good tasting</flavor>
                <mouthfeel>Good feeling</mouthfeel>
                <impression>Refreshing</impression>
            </subcategory>
        </category>
    </class>
</styleguide>
'''


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