def test_expand_target_dns(self):
"""
test expand_target when target is root@<dns>
"""
host = "localhost"
user = "******"
opts = self.opts
opts["tgt"] = user + host
with patch("salt.utils.network.is_reachable_host",
MagicMock(return_value=False)):
client = ssh.SSH(opts)
assert opts["tgt"] == user + host
with patch(
"salt.roster.get_roster_file",
MagicMock(return_value="/etc/salt/roster")), patch(
"salt.client.ssh.compile_template",
MagicMock(
return_value=salt.utils.yaml.safe_load(self.roster)),
):
client._expand_target()
assert opts["tgt"] == host
def test_expand_target_no_user(self):
"""
test expand_target when no user defined
"""
host = "127.0.0.1"
opts = self.opts
opts["tgt"] = host
with patch("salt.utils.network.is_reachable_host",
MagicMock(return_value=False)):
client = ssh.SSH(opts)
assert opts["tgt"] == host
with patch(
"salt.roster.get_roster_file",
MagicMock(return_value="/etc/salt/roster")), patch(
"salt.client.ssh.compile_template",
MagicMock(
return_value=salt.utils.yaml.safe_load(self.roster)),
):
client._expand_target()
assert opts["tgt"] == host
def test_expand_target_no_host(self):
"""
test expand_target when host is not included in the rosterdata
"""
host = "127.0.0.1"
user = "******"
opts = self.opts
opts["tgt"] = user + host
roster = """
localhost: 127.0.0.1
"""
roster_file = os.path.join(RUNTIME_VARS.TMP, "test_roster_no_host")
with salt.utils.files.fopen(roster_file, "w") as fp:
salt.utils.yaml.safe_dump(salt.utils.yaml.safe_load(roster), fp)
with patch(
"salt.utils.network.is_reachable_host", MagicMock(return_value=False)
):
client = ssh.SSH(opts)
assert opts["tgt"] == user + host
with patch("salt.roster.get_roster_file", MagicMock(return_value=roster_file)):
client._expand_target()
assert opts["tgt"] == host
