Add a few tests for RattailConfig

This commit is contained in:
Lance Edgar 2016-05-14 18:30:41 -05:00
parent a056997c04
commit 9917ed60ef
2 changed files with 129 additions and 8 deletions

View file

@ -65,8 +65,8 @@ def parse_list(value):
return [] return []
# Per the shlex docs (https://docs.python.org/2/library/shlex.html): # Per the shlex docs (https://docs.python.org/2/library/shlex.html):
# "Prior to Python 2.7.3, this module did not support Unicode input." # "Prior to Python 2.7.3, this module did not support Unicode input."
if sys.version_info < (2, 7, 3) and isinstance(value, unicode): # pragma: no cover if isinstance(value, unicode):
value = value.encode(u'utf-8') value = value.encode('utf-8')
parser = shlex.shlex(value) parser = shlex.shlex(value)
parser.whitespace += u',' parser.whitespace += u','
parser.whitespace_split = True parser.whitespace_split = True
@ -112,7 +112,9 @@ class RattailConfig(object):
self.usedb = usedb self.usedb = usedb
if self.usedb is None: if self.usedb is None:
self.usedb = self.getbool('rattail.config', 'usedb', usedb=False, default=False) self.usedb = self.getbool('rattail.config', 'usedb', usedb=False, default=False)
self.preferdb = self.getbool('rattail.config', 'preferdb', usedb=False, default=False) self.preferdb = preferdb
if self.usedb and self.preferdb is None:
self.preferdb = self.getbool('rattail.config', 'preferdb', usedb=False, default=False)
# Attempt to detect lack of SQLAlchemy libraries etc. This allows us # Attempt to detect lack of SQLAlchemy libraries etc. This allows us
# to avoid installing those on a machine which will not need to access # to avoid installing those on a machine which will not need to access

View file

@ -1,13 +1,39 @@
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
from __future__ import unicode_literals from __future__ import unicode_literals, absolute_import
from unittest import TestCase import os
import unittest
from fixture import TempIO
from rattail import config from rattail import config
class TestParseList(TestCase): class TestParseBoolFunc(unittest.TestCase):
def test_none(self):
self.assertIsNone(config.parse_bool(None))
def test_true(self):
self.assertIs(config.parse_bool(True), True)
def test_false(self):
self.assertIs(config.parse_bool(False), False)
def test_string(self):
self.assertTrue(config.parse_bool('true'))
self.assertTrue(config.parse_bool('yes'))
self.assertTrue(config.parse_bool('on'))
self.assertTrue(config.parse_bool('1'))
self.assertFalse(config.parse_bool('false'))
self.assertFalse(config.parse_bool('no'))
self.assertFalse(config.parse_bool('off'))
self.assertFalse(config.parse_bool('0'))
class TestParseListFunc(unittest.TestCase):
def test_none(self): def test_none(self):
value = config.parse_list(None) value = config.parse_list(None)
@ -50,12 +76,105 @@ class TestParseList(TestCase):
self.assertEqual(value[2], u'baz') self.assertEqual(value[2], u'baz')
def test_multiple_values_separated_by_whitespace_and_commas_with_some_quoting(self): def test_multiple_values_separated_by_whitespace_and_commas_with_some_quoting(self):
value = config.parse_list(u""" value = config.parse_list("""
foo foo
"C:\\some path\\with spaces\\and, a comma", "C:\\some path\\with spaces\\and, a comma",
baz baz
""") """)
self.assertEqual(len(value), 3) self.assertEqual(len(value), 3)
self.assertEqual(value[0], u'foo') self.assertEqual(value[0], u'foo')
self.assertEqual(value[1], u'C:\\some path\\with spaces\\and, a comma') self.assertEqual(value[1], u'C:\\some path\\with spaces\\and, a comma')
self.assertEqual(value[2], u'baz') self.assertEqual(value[2], u'baz')
class TestRattailConfig(unittest.TestCase):
def setUp(self):
self.tempio = TempIO()
def tearDown(self):
del self.tempio
def setup_files(self):
self.site_path = self.tempio.putfile('site.conf', """
[rattail]
""")
self.host_path = self.tempio.putfile('host.conf', """
[rattail.config]
include = "{}"
""".format(self.site_path))
self.app_path = self.tempio.putfile('app.conf', """
[rattail.config]
include = "{}"
""".format(self.host_path))
self.custom_path = self.tempio.putfile('custom.conf', """
[rattail.config]
include = "%(here)s/app.conf"
""")
def test_init_defaults(self):
cfg = config.RattailConfig()
self.assertEqual(cfg.files_requested, [])
self.assertEqual(cfg.files_read, [])
self.assertIsNone(cfg._session_factory)
def test_init_params(self):
self.setup_files()
# files
cfg = config.RattailConfig()
self.assertEqual(cfg.files_requested, [])
self.assertEqual(cfg.files_read, [])
cfg = config.RattailConfig(files=[self.site_path])
self.assertEqual(cfg.files_requested, [self.site_path])
self.assertEqual(cfg.files_read, [self.site_path])
# usedb
cfg = config.RattailConfig()
self.assertFalse(cfg.usedb)
cfg = config.RattailConfig(usedb=True)
self.assertTrue(cfg.usedb)
# preferdb
cfg = config.RattailConfig()
self.assertFalse(cfg.preferdb)
cfg = config.RattailConfig(preferdb=True)
self.assertTrue(cfg.preferdb)
def test_read_file_with_recurse(self):
self.setup_files()
cfg = config.RattailConfig()
cfg.read_file(self.custom_path, recurse=True)
self.assertEqual(cfg.files_requested, [self.custom_path, self.app_path, self.host_path, self.site_path])
self.assertEqual(cfg.files_read, [self.site_path, self.host_path, self.app_path, self.custom_path])
def test_read_file_once_only(self):
self.setup_files()
another_path = self.tempio.putfile('another.conf', """
[rattail.config]
include = "{custom}" "{site}" "{app}" "{site}" "{custom}"
""".format(custom=self.custom_path, app=self.app_path, site=self.site_path))
cfg = config.RattailConfig()
cfg.read_file(another_path, recurse=True)
self.assertEqual(cfg.files_requested, [another_path, self.custom_path, self.app_path, self.host_path, self.site_path])
self.assertEqual(cfg.files_read, [self.site_path, self.host_path, self.app_path, self.custom_path, another_path])
def test_read_file_skip_missing(self):
self.setup_files()
bogus_path = '/tmp/does-not/exist'
self.assertFalse(os.path.exists(bogus_path))
another_path = self.tempio.putfile('another.conf', """
[rattail.config]
include = "{bogus}" "{app}" "{bogus}" "{site}"
""".format(bogus=bogus_path, app=self.app_path, site=self.site_path))
cfg = config.RattailConfig()
cfg.read_file(another_path, recurse=True)
self.assertEqual(cfg.files_requested, [another_path, bogus_path, self.app_path, self.host_path, self.site_path])
self.assertEqual(cfg.files_read, [self.site_path, self.host_path, self.app_path, another_path])