mirror of
https://github.com/datafolklabs/cement.git
synced 2026-02-06 20:56:49 +00:00
67 lines
1.6 KiB
Python
67 lines
1.6 KiB
Python
import os
|
|
import json
|
|
from unittest.mock import patch
|
|
from cement.utils.test import TestApp
|
|
from cement.utils import fs
|
|
|
|
CONFIG_PARSED = dict(
|
|
section=dict(
|
|
subsection=dict(
|
|
list=['item1', 'item2', 'item3', 'item4'],
|
|
key='value'),
|
|
key1='ok1',
|
|
key2='ok2',
|
|
),
|
|
)
|
|
|
|
|
|
CONFIG = fs.join(os.path.dirname(__file__), '..',
|
|
'data', 'config', 'config.json')
|
|
|
|
|
|
class JsonApp(TestApp):
|
|
class Meta:
|
|
extensions = ['json']
|
|
output_handler = 'json'
|
|
config_handler = 'json'
|
|
config_files = [CONFIG]
|
|
argv = ['-o', 'json']
|
|
meta_defaults = {'output.json': {'overridable': True}}
|
|
|
|
|
|
def test_json():
|
|
with JsonApp() as app:
|
|
app.run()
|
|
res = app.render(dict(foo='bar'))
|
|
json_res = json.dumps(dict(foo='bar'))
|
|
assert res == json_res
|
|
|
|
|
|
def test_has_section():
|
|
with JsonApp() as app:
|
|
assert app.config.has_section('section')
|
|
|
|
|
|
def test_keys():
|
|
with JsonApp() as app:
|
|
assert 'subsection' in app.config.keys('section')
|
|
|
|
|
|
@patch('cement.ext.ext_json.JsonConfigHandler._parse_file')
|
|
def test_parse_file_bad_path(parser):
|
|
with JsonApp(config_files=['./some_bogus_path']):
|
|
assert not parser.called
|
|
|
|
|
|
def test_parse_file():
|
|
with JsonApp() as app:
|
|
assert app.config.get('section', 'key1') == 'ok1'
|
|
assert app.config.get_section_dict('section') == \
|
|
CONFIG_PARSED['section']
|
|
|
|
|
|
def test_get_dict():
|
|
with JsonApp() as app:
|
|
_config = app.config.get_dict()
|
|
assert _config['log.logging']['level'] == 'INFO'
|