"""Tests for the report_presets model."""

from reporting.models import report_presets


def test_get_report_presets(
    art_relations_with_data, valid_grass_account_vendor_id
):
    """Test get_report_presets."""
    response = report_presets.get_report_presets(
        'vendor', valid_grass_account_vendor_id
    )
    assert response
    assert response.message == [
        {
            'id': 1,
            'name': 'Nice Report',
            'report_id': 'awesome_report',
            'params': {'applesauce': 'bananas'},
            'datetime_created': '2018-01-29T11:43:23Z',
            'is_orchard_preset': False,
        },
        {
            'id': 2,
            'name': 'Best Report',
            'report_id': 'awesome_report',
            'params': {'applesauce': 'bananas'},
            'datetime_created': '2018-01-27T09:55:01Z',
            'is_orchard_preset': False,
        },
    ]


def test_get_orchard_report_presets():
    """Test get_orchard_report_presets."""
    response = report_presets.get_orchard_report_presets()
    assert response
    assert response.message == []


def test_delete_report_preset(
    art_relations_with_data, valid_grass_account_vendor_id
):
    """Test delete_orchard_report_presets."""
    product_id = 1
    account_type = 'vendor'

    response = report_presets.delete_report_preset(
        product_id, account_type, valid_grass_account_vendor_id
    )

    assert response.message == 'success'


def test_create_report_preset(
    art_relations_with_data, valid_grass_account_vendor_id
):
    """Test get_orchard_report_presets."""
    product_dict = {
        'name': 'name',
        'report_id': 'id',
        'params': {'cow': 'abunga'},
    }
    account_type = 'vendor'
    account_id = '1000000'
    user_id = '100000000'

    response = report_presets.create_report_preset(
        product_dict, account_type, account_id, user_id
    )

    response.message.pop('datetime_created')
    preset_id = response.message.pop('id')

    assert response.message == {
        'is_orchard_preset': False,
        'name': 'name',
        'params': {'cow': 'abunga'},
        'report_id': 'id',
    }

    presets = report_presets.get_report_presets(account_type, account_id)
    assert len(presets.message) == 1
    assert presets.message[0]['id'] == preset_id