1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75
|
# THIS FILE IS PART OF THE CYLC WORKFLOW ENGINE.
# Copyright (C) NIWA & British Crown (Met Office) & Contributors.
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
import os
import re
import pytest
from cylc.flow.hostuserutil import (
get_fqdn_by_host,
get_host,
get_user,
get_user_home,
is_remote_host,
is_remote_user
)
def test_is_remote_user_on_current_user():
"""is_remote_user with current user."""
assert not is_remote_user(None)
assert not is_remote_user(os.getenv('USER'))
def test_is_remote_host_on_localhost(monkeypatch):
"""is_remote_host with localhost."""
assert not is_remote_host(None)
assert not is_remote_host('localhost')
assert not is_remote_host('localhost4.localhost42')
assert not is_remote_host(os.getenv('HOSTNAME'))
assert not is_remote_host(get_host())
def test_get_fqdn_by_host_on_bad_host():
"""get_fqdn_by_host bad host.
Warning:
This test can fail due to ISP/network configuration
(for example ISP may reroute failed DNS to custom search page)
e.g: https://www.virginmedia.com/help/advanced-network-error-search
"""
bad_host = 'nosuchhost.nosuchdomain.org'
with pytest.raises(IOError) as exc:
get_fqdn_by_host(bad_host)
assert re.match(
r"(\[Errno -2\] Name or service|"
r"\[Errno 8\] nodename nor servname provided, or)"
r" not known: '{}'".format(bad_host),
str(exc.value)
)
assert exc.value.filename == bad_host
def test_get_user():
"""get_user."""
assert os.getenv('USER') == get_user()
def test_get_user_home():
"""get_user_home."""
assert os.getenv('HOME') == get_user_home()
|