def test_hostname(self): with replace_env('SSH_CLIENT', '192.168.0.12 40921 22'): with replace_module_module(common, 'socket', gethostname=lambda: 'abc'): self.assertEqual(common.hostname(), 'abc') self.assertEqual(common.hostname(only_if_ssh=True), 'abc') os.environ.pop('SSH_CLIENT') self.assertEqual(common.hostname(), 'abc') self.assertEqual(common.hostname(only_if_ssh=True), None)
def test_hostname(self): pl = Pl() with replace_env('SSH_CLIENT', '192.168.0.12 40921 22') as segment_info: with replace_module_module(common, 'socket', gethostname=lambda: 'abc'): self.assertEqual(common.hostname(pl=pl, segment_info=segment_info), 'abc') self.assertEqual(common.hostname(pl=pl, segment_info=segment_info, only_if_ssh=True), 'abc') with replace_module_module(common, 'socket', gethostname=lambda: 'abc.mydomain'): self.assertEqual(common.hostname(pl=pl, segment_info=segment_info), 'abc.mydomain') self.assertEqual(common.hostname(pl=pl, segment_info=segment_info, exclude_domain=True), 'abc') self.assertEqual(common.hostname(pl=pl, segment_info=segment_info, only_if_ssh=True), 'abc.mydomain') self.assertEqual(common.hostname(pl=pl, segment_info=segment_info, only_if_ssh=True, exclude_domain=True), 'abc') segment_info['environ'].pop('SSH_CLIENT') with replace_module_module(common, 'socket', gethostname=lambda: 'abc'): self.assertEqual(common.hostname(pl=pl, segment_info=segment_info), 'abc') self.assertEqual(common.hostname(pl=pl, segment_info=segment_info, only_if_ssh=True), None) with replace_module_module(common, 'socket', gethostname=lambda: 'abc.mydomain'): self.assertEqual(common.hostname(pl=pl, segment_info=segment_info), 'abc.mydomain') self.assertEqual(common.hostname(pl=pl, segment_info=segment_info, exclude_domain=True), 'abc') self.assertEqual(common.hostname(pl=pl, segment_info=segment_info, only_if_ssh=True, exclude_domain=True), None)
def user_hostname(pl, segment_info, only_if_ssh=False, exclude_domain=False): user_seg = user(pl, segment_info) host = hostname(pl, segment_info, only_if_ssh, exclude_domain) user_seg[0]['contents'] = "{0}@{1}".format(user_seg[0]['contents'], host) return user_seg