Lubuntu/ubuntu-dev-tools
3
0
Fork 0
mirror of https://git.launchpad.net/ubuntu-dev-tools synced 2025-05-10 08:21:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

test_config.py: Use mox.

Browse Source
This commit is contained in:
Benjamin Drung 2010-12-27 13:37:40 +01:00
parent 04f52e9de6
commit 45a341dfd1
1 changed files with 12 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
ubuntutools/test/test_config.py
View File

@ -14,12 +14,14 @@
# OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR # OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
# PERFORMANCE OF THIS SOFTWARE. # PERFORMANCE OF THIS SOFTWARE.
import __builtin__
import os import os
import os.path import os.path
import sys import sys
from StringIO import StringIO from StringIO import StringIO
import ubuntutools.config import mox
from ubuntutools.config import UDTConfig, ubu_email from ubuntutools.config import UDTConfig, ubu_email
from ubuntutools.logger import Logger from ubuntutools.logger import Logger
from ubuntutools.test import unittest from ubuntutools.test import unittest
@ -41,25 +43,24 @@ def fake_open(filename, mode='r'):
return StringIO(files[filename]) return StringIO(files[filename])
class ConfigTestCase(unittest.TestCase): class ConfigTestCase(mox.MoxTestBase, unittest.TestCase):
def setUp(self): def setUp(self):
ubuntutools.config.open = fake_open super(ConfigTestCase, self).setUp()
self.mox.stubs.Set(__builtin__, 'open', fake_open)
Logger.stdout = StringIO() Logger.stdout = StringIO()
Logger.stderr = StringIO() Logger.stderr = StringIO()
self.cleanEnvironment() self.clean_environment()
def tearDown(self): def tearDown(self):
del ubuntutools.config.open
self.assertEqual(Logger.stdout.getvalue(), '') self.assertEqual(Logger.stdout.getvalue(), '')
self.assertEqual(Logger.stderr.getvalue(), '') self.assertEqual(Logger.stderr.getvalue(), '')
Logger.stdout = sys.stdout Logger.stdout = sys.stdout
Logger.stderr = sys.stderr Logger.stderr = sys.stderr
self.cleanEnvironment() self.clean_environment()
def cleanEnvironment(self): def clean_environment(self):
config_files['system'] = '' config_files['system'] = ''
config_files['user'] = '' config_files['user'] = ''
for k in os.environ.keys(): for k in os.environ.keys():
@ -154,12 +155,12 @@ REPEAT=yes
class UbuEmailTestCase(unittest.TestCase): class UbuEmailTestCase(unittest.TestCase):
def setUp(self): def setUp(self):
self.cleanEnvironment() self.clean_environment()
def tearDown(self): def tearDown(self):
self.cleanEnvironment() self.clean_environment()
def cleanEnvironment(self): def clean_environment(self):
for k in ('UBUMAIL', 'DEBEMAIL', 'DEBFULLNAME'): for k in ('UBUMAIL', 'DEBEMAIL', 'DEBFULLNAME'):
if k in os.environ: if k in os.environ:
del os.environ[k] del os.environ[k]