"""Test presets logic layer.""" from oto import response from reporting.logic import presets from reporting.models import report_presets as report_presets_model def test_get_report_presets(mocker): """Test it returns proper report presets.""" mocker.patch.object( report_presets_model, 'get_report_presets', return_value=response.Response(message=[{'name': 'account presets'}]), autospec=True, ) mocker.patch.object( report_presets_model, 'get_orchard_report_presets', return_value=response.Response(message=[{'name': 'orchard presets'}]), autospec=True, ) report_presets = presets.get_report_presets('vendor', 123456) assert report_presets.message == { 'items': [{'name': 'account presets'}, {'name': 'orchard presets'}] } def test_get_report_presets_orchard_fail(mocker): """Test it returns error on orchard preset fetch failure.""" mocker.patch.object( report_presets_model, 'get_report_presets', return_value=response.Response(), autospec=True, ) mocker.patch.object( report_presets_model, 'get_orchard_report_presets', return_value=response.Response(status=400), autospec=True, ) report_presets = presets.get_report_presets('vendor', 123456) assert not report_presets def test_get_report_presets_account_fail(mocker): """Test it returns error on account preset fetch failure.""" mocker.patch.object( report_presets_model, 'get_report_presets', return_value=response.Response(status=400), autospec=True, ) report_presets = presets.get_report_presets('vendor', 123456) assert not report_presets def test_delete_report_preset(mocker): """Test it delegates to the model correctly.""" product_id = 1 mocker.patch.object( report_presets_model, 'delete_report_preset', return_value='response' ) assert ( report_presets_model.delete_report_preset(product_id, 'vendor', 1) == 'response' ) def test_create_report_preset(mocker): """Test it delegates to the model correctly.""" product = {'test': 'test'} mocker.patch.object( report_presets_model, 'create_report_preset', return_value='response' ) assert ( report_presets_model.create_report_preset(product, 'vendor', 1, 1) == 'response' )