fix: tests

Author: yanksyoon

juju/machine.py

@@ -154,9 +154,9 @@ async def ssh(
         """
         if proxy:
             raise NotImplementedError('proxy option is not implemented')
-        address = self.dns_name
         if wait_for_active:
             await block_until(lambda: self.addresses, timeout=timeout)
+        address = self.dns_name
         destination = "{}@{}".format(user, address)
         _, id_path = juju_ssh_key_paths()
         cmd = [

tests/integration/test_machine.py

@@ -71,7 +71,6 @@ async def test_scp(event_loop):
 async def test_machine_ssh():
     async with base.CleanModel() as model:
         machine: Machine = await model.add_machine()
-        await machine.wait()
         out = await machine.ssh("echo hello world!")
 
         assert out == "hello world!\n"