forked from mozilla/configman
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
8 changed files
with
382 additions
and
41 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,184 @@ | ||
| # ***** BEGIN LICENSE BLOCK ***** | ||
| # Version: MPL 1.1/GPL 2.0/LGPL 2.1 | ||
| # | ||
| # The contents of this file are subject to the Mozilla Public License Version | ||
| # 1.1 (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.mozilla.org/MPL/ | ||
| # | ||
| # Software distributed under the License is distributed on an "AS IS" basis, | ||
| # WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License | ||
| # for the specific language governing rights and limitations under the | ||
| # License. | ||
| # | ||
| # The Original Code is configman | ||
| # | ||
| # The Initial Developer of the Original Code is | ||
| # Mozilla Foundation | ||
| # Portions created by the Initial Developer are Copyright (C) 2011 | ||
| # the Initial Developer. All Rights Reserved. | ||
| # | ||
| # Contributor(s): | ||
| # K Lars Lohn, lars@mozilla.com | ||
| # Peter Bengtsson, peterbe@mozilla.com | ||
| # | ||
| # Alternatively, the contents of this file may be used under the terms of | ||
| # either the GNU General Public License Version 2 or later (the "GPL"), or | ||
| # the GNU Lesser General Public License Version 2.1 or later (the "LGPL"), | ||
| # in which case the provisions of the GPL or the LGPL are applicable instead | ||
| # of those above. If you wish to allow use of your version of this file only | ||
| # under the terms of either the GPL or the LGPL, and not to allow others to | ||
| # use your version of this file under the terms of the MPL, indicate your | ||
| # decision by deleting the provisions above and replace them with the notice | ||
| # and other provisions required by the GPL or the LGPL. If you do not delete | ||
| # the provisions above, a recipient may use your version of this file under | ||
| # the terms of any one of the MPL, the GPL or the LGPL. | ||
| # | ||
| # ***** END LICENSE BLOCK ***** | ||
|
|
||
| import unittest | ||
| import os | ||
| import tempfile | ||
| from cStringIO import StringIO | ||
| import contextlib | ||
|
|
||
| import configman.datetime_util as dtu | ||
| import configman.config_manager as config_manager | ||
| try: | ||
| from ..value_sources.for_configobj import ValueSource | ||
| except ImportError: | ||
| # this module is optional. If it doesn't exsit, that's ok, we'll just | ||
| # igrore the tests | ||
| pass | ||
| else: | ||
|
|
||
| def stringIO_context_wrapper(a_stringIO_instance): | ||
| @contextlib.contextmanager | ||
| def stringIS_context_manager(): | ||
| yield a_stringIO_instance | ||
| return stringIS_context_manager | ||
|
|
||
|
|
||
| class TestCase(unittest.TestCase): | ||
| def _some_namespaces(self): | ||
| """set up some namespaces""" | ||
| n = config_manager.Namespace(doc='top') | ||
| n.add_option('aaa', '2011-05-04T15:10:00', 'the a', | ||
| short_form='a', | ||
| from_string_converter=dtu.datetime_from_ISO_string | ||
| ) | ||
| n.c = config_manager.Namespace(doc='c space') | ||
| n.c.add_option('fred', 'stupid', 'husband from Flintstones') | ||
| n.c.add_option('wilma', 'waspish', 'wife from Flintstones') | ||
| n.d = config_manager.Namespace(doc='d space') | ||
| n.d.add_option('fred', 'crabby', 'male neighbor from I Love Lucy') | ||
| n.d.add_option('ethel', 'silly', 'female neighbor from I Love Lucy') | ||
| n.x = config_manager.Namespace(doc='x space') | ||
| n.x.add_option('size', 100, 'how big in tons', short_form='s') | ||
| n.x.add_option('password', 'secret', 'the password') | ||
| return n | ||
|
|
||
| def test_for_configobj_basics(self): | ||
| """test basic use of for_configobj""" | ||
| tmp_filename = os.path.join(tempfile.gettempdir(), 'test.ini') | ||
| open(tmp_filename, 'w').write(""" | ||
| # comment | ||
| name=Peter | ||
| awesome= | ||
| # comment | ||
| [othersection] | ||
| foo=bar # other comment | ||
| """) | ||
|
|
||
| try: | ||
| o = ValueSource(tmp_filename) | ||
| r = {'othersection': {'foo': 'bar'}, | ||
| 'name': 'Peter', | ||
| 'awesome': ''} | ||
| print o.get_values(None, None) | ||
| assert o.get_values(None, None) == r | ||
| # in the case of this implementation of a ValueSource, | ||
| # the two parameters to get_values are dummies. That may | ||
| # not be true for all ValueSource implementations | ||
| self.assertEqual(o.get_values(0, False), r) | ||
| self.assertEqual(o.get_values(1, True), r) | ||
| self.assertEqual(o.get_values(2, False), r) | ||
| self.assertEqual(o.get_values(3, True), r) | ||
|
|
||
| # XXX (peterbe): commented out because I'm not sure if | ||
| # OptionsByIniFile get_values() should depend on the configuration | ||
| # manager it is given as first argument or not. | ||
| #c = config_manager.ConfigurationManager([], | ||
| #use_admin_controls=True, | ||
| ##use_config_files=False, | ||
| #auto_help=False, | ||
| #argv_source=[]) | ||
| #self.assertEqual(o.get_values(c, True), {}) | ||
| #self.assertRaises(config_manager.NotAnOptionError, | ||
| # o.get_values, c, False) | ||
|
|
||
| #c.option_definitions.add_option('limit', default=0) | ||
| #self.assertEqual(o.get_values(c, False), {'limit': '20'}) | ||
| #self.assertEqual(o.get_values(c, True), {'limit': '20'}) | ||
| finally: | ||
| if os.path.isfile(tmp_filename): | ||
| os.remove(tmp_filename) | ||
|
|
||
| def test_for_configobj_basics_2(self): | ||
| tmp_filename = os.path.join(tempfile.gettempdir(), 'test.ini') | ||
| open(tmp_filename, 'w').write(""" | ||
| # comment | ||
| name=Peter | ||
| awesome= | ||
| # comment | ||
| [othersection] | ||
| foo=bar # other comment | ||
| """) | ||
|
|
||
| try: | ||
| o = ValueSource(tmp_filename) | ||
| c = config_manager.ConfigurationManager([], | ||
| use_admin_controls=True, | ||
| #use_config_files=False, | ||
| use_auto_help=False, | ||
| argv_source=[]) | ||
|
|
||
| self.assertEqual(o.get_values(c, False), | ||
| {'othersection': {'foo': 'bar'}, | ||
| 'name': 'Peter', | ||
| 'awesome': ''}) | ||
| self.assertEqual(o.get_values(c, True), | ||
| {'othersection': {'foo': 'bar'}, | ||
| 'name': 'Peter', | ||
| 'awesome': ''}) | ||
| finally: | ||
| if os.path.isfile(tmp_filename): | ||
| os.remove(tmp_filename) | ||
|
|
||
| def test_write_ini(self): | ||
| n = self._some_namespaces() | ||
| c = config_manager.ConfigurationManager( | ||
| [n], | ||
| use_admin_controls=True, | ||
| #use_config_files=False, | ||
| use_auto_help=False, | ||
| argv_source=[] | ||
| ) | ||
| expected = """aaa = 2011-05-04T15:10:00 | ||
| [x] | ||
| password = secret | ||
| size = 100 | ||
| [c] | ||
| wilma = waspish | ||
| fred = stupid | ||
| [d] | ||
| ethel = silly | ||
| fred = crabby | ||
| """ | ||
| out = StringIO() | ||
| c.write_conf('ini', opener=stringIO_context_wrapper(out)) | ||
| received = out.getvalue() | ||
| out.close() | ||
| print received.strip() | ||
| print expected.strip() | ||
| self.assertEqual(expected.strip(), received.strip()) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.