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