# 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 decimal import Decimal

from humulus.filters import *
from humulus.recipes import HopForm


def test_recipe_og(sample_recipes):
    assert recipe_og(sample_recipes['lager']) == '1.054'
    assert recipe_og(sample_recipes['sweetstout']) == '1.038'
    # Remove fermentables, verify 0 is returned
    sample_recipes['lager'].pop('fermentables')
    assert recipe_og(sample_recipes['lager']) == '0.000'


def test_recipe_fg(sample_recipes):
    assert recipe_fg(sample_recipes['lager']) == '1.014'
    assert recipe_fg(sample_recipes['sweetstout']) == '1.015'
    # Remove fermentables, verify 0 is returned
    sample_recipes['lager'].pop('fermentables')
    assert recipe_fg(sample_recipes['lager']) == '0.000'
    # Remove yeast, verify 0 is returned
    sample_recipes['sweetstout'].pop('yeast')
    assert recipe_fg(sample_recipes['sweetstout']) == '0.000'


def test_recipe_ibu(sample_recipes):
    assert recipe_ibu(sample_recipes['lager']) == '24'
    assert recipe_ibu(sample_recipes['sweetstout']) == '34'
    # Remove hops, verify 0 is returned
    sample_recipes['lager'].pop('hops')
    assert recipe_ibu(sample_recipes['lager']) == '0'


def test_recipe_ibu_ratio(sample_recipes):
    assert recipe_ibu_ratio(sample_recipes['lager']) == '0.44'
    assert recipe_ibu_ratio(sample_recipes['sweetstout']) == '0.89'
    # Remove fermentables, verify 0 is returned
    sample_recipes['lager'].pop('fermentables')
    assert recipe_ibu_ratio(sample_recipes['lager']) == '0'
    # Remove hops, verify 0 is returned
    sample_recipes['sweetstout'].pop('hops')
    assert recipe_ibu_ratio(sample_recipes['sweetstout']) == '0'


def test_recipe_abv(sample_recipes):
    assert recipe_abv(sample_recipes['lager']) == '5.3'
    assert recipe_abv(sample_recipes['sweetstout']) == '3.0'
    # Remove fermentables, verify 0 is returned
    sample_recipes['lager'].pop('fermentables')
    assert recipe_abv(sample_recipes['lager']) == '0'
    # Remove yeast, verify 0 is returned
    sample_recipes['sweetstout'].pop('yeast')
    assert recipe_abv(sample_recipes['sweetstout']) == '0'


def test_recipe_srm(sample_recipes):
    assert recipe_srm(sample_recipes['lager']) == '3'
    assert recipe_srm(sample_recipes['sweetstout']) == '21'
    # Remove fermentables, verify 0 is returned
    sample_recipes['lager'].pop('fermentables')
    assert recipe_srm(sample_recipes['lager']) == '0'


def test_sort_hops():
    # Test with no form
    hops = [
        {'name': '4', 'use': 'Dry-Hop', 'duration': '5'},
        {'name': '3', 'use': 'Whirlpool', 'duration': '10'},
        {'name': '2', 'use': 'Boil', 'duration': '5'},
        {'name': '1', 'use': 'Boil', 'duration': '15'},
        {'name': '0', 'use': 'FWH', 'duration': '60'},
    ]
    assert sort_hops(hops) == [
        {'name': '0', 'use': 'FWH', 'duration': '60'},
        {'name': '1', 'use': 'Boil', 'duration': '15'},
        {'name': '2', 'use': 'Boil', 'duration': '5'},
        {'name': '3', 'use': 'Whirlpool', 'duration': '10'},
        {'name': '4', 'use': 'Dry-Hop', 'duration': '5'},
    ]

    # Test with form
    hop_forms = []
    for hop in hops:
        form = HopForm()
        form.name.data = hop['name']
        form.use.data = hop['use']
        form.duration.data = Decimal(hop['duration'])
        hop_forms.append(form)

    for num, hop in enumerate(sort_hops(hop_forms, form=True)):
        assert hop.name.data == str(num)


def test_ferm_pct():
    ferms = [{'amount': '4'}, {'amount': '2'}, {'amount': '2'}]
    assert ferm_pct(ferms) == [
        {'amount': '4', 'pct': 50.0},
        {'amount': '2', 'pct': 25.0},
        {'amount': '2', 'pct': 25.0}
    ]