else:
return hostname.split('.', 1)[0]
-def canonicalize_hostname(hostname, user: Optional[str] ='ubuntu'):
+def canonicalize_hostname(hostname, user: Optional[str] = None):
+ user = user or get_test_user()
hostname_expr = hostname_expr_templ.format(
lab_domain=config.lab_domain.replace('.', r'\.'))
match = re.match(hostname_expr, hostname)
"""
Turn off StrictHostKeyChecking
"""
+ user = misc.get_test_user()
run.wait(
ctx.cluster.run(
args=[
'echo',
'StrictHostKeyChecking no\n',
run.Raw('>'),
- run.Raw('/home/ubuntu/.ssh/config'),
+ run.Raw(f'/home/{user}/.ssh/config'),
run.Raw('&&'),
'echo',
'UserKnownHostsFile ',
run.Raw('/dev/null'),
run.Raw('>>'),
- run.Raw('/home/ubuntu/.ssh/config'),
+ run.Raw(f'/home/{user}/.ssh/config'),
run.Raw('&&'),
- run.Raw('chmod 600 /home/ubuntu/.ssh/config'),
+ run.Raw(f'chmod 600 /home/{user}/.ssh/config'),
],
wait=False,
)
finally:
run.wait(
ctx.cluster.run(
- args=['rm',run.Raw('/home/ubuntu/.ssh/config')],
+ args=['rm',run.Raw(f'/home/{user}/.ssh/config')],
wait=False
),
)