tyler.durden/powerline
1
0
Fork 0
mirror of https://github.com/powerline/powerline.git synced 2025-07-28 08:14:41 +02:00
Code Issues Packages Projects Releases Wiki Activity

Add test for hostname exclude_domain option

Browse Source
This commit is contained in:
Matt Spaulding 2013-03-30 15:32:29 -07:00
parent e03e864f69
commit 3a6c0ab8fd
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tests/test_segments.py
View File

@ -37,6 +37,16 @@ class TestCommon(TestCase):
pl.environ.pop('SSH_CLIENT')
self.assertEqual(common.hostname(pl=pl), 'abc')
self.assertEqual(common.hostname(pl=pl, only_if_ssh=True), None)
with replace_env('SSH_CLIENT', '192.168.0.12 40921 22') as pl:
with replace_module_module(common, 'socket', gethostname=lambda: 'abc.mydomain'):
self.assertEqual(common.hostname(pl=pl), 'abc.mydomain')
self.assertEqual(common.hostname(pl=pl, exclude_domain=True), 'abc')
self.assertEqual(common.hostname(pl=pl, only_if_ssh=True), 'abc.mydomain')
self.assertEqual(common.hostname(pl=pl, only_if_ssh=True, exclude_domain=True), 'abc')
pl.environ.pop('SSH_CLIENT')
self.assertEqual(common.hostname(pl=pl), 'abc.mydomain')
self.assertEqual(common.hostname(pl=pl, exclude_domain=True), 'abc')
self.assertEqual(common.hostname(pl=pl, only_if_ssh=True, exclude_domain=True), None)
def test_user(self):
new_os = new_module('os', getpid=lambda: 1)