def test_key_gen(self):
HEADING()
inv_file = '~/.cloudmesh/config_test.yaml'
inv = Inventory(inv_file)
inv.add_directory(host='test_host1',
keyfile='~/.ssh/id_rsa.pub',
service='manager')
inv.save()
c = Configure(inventory=inv_file)
priv_key, pub_key = c.generate_ssh_key(hostname='test_host1')
u = c.build_user_data(name='test_host1')
keys = readfile(filename='~/.ssh/id_rsa.pub').strip().split('\n')
t1 = Userdata()\
.with_authorized_keys(keys=keys)\
.with_ssh_password_login(ssh_pwauth=False)\
.with_locale()\
.with_net_tools()\
.with_hostname(hostname='test_host1')\
.with_hosts(hosts=['127.0.0.1:test_host1'])\
.with_packages(packages='avahi-daemon')\
.with_runcmd(cmd=f'cat /boot/firmware/id_rsa.pub > /home/ubuntu/.ssh/id_rsa.pub')\
.with_runcmd(cmd=f'cat /boot/firmware/id_rsa > /home/ubuntu/.ssh/id_rsa')\
.with_fix_user_dir_owner(user='******')\
.with_runcmd(cmd=f'chmod 600 /home/ubuntu/.ssh/id_rsa')\
.with_runcmd(cmd=f'sudo rm /boot/firmware/id_rsa.pub')\
.with_runcmd(cmd=f'sudo rm /boot/firmware/id_rsa')
assert (t1.content == u.content)
os.system("rm -f " + path_expand(inv_file))
def test_build_networkdata(self):
HEADING()
inv_file = '~/.cloudmesh/config_test.yaml'
inv = Inventory(inv_file)
inv.add_directory(host='test_host1',
keyfile='~/.ssh/id_rsa.pub',
service='manager',
ip='10.1.1.11',
router='10.1.1.1',
dns=['8.8.8.8', '8.8.4.4'])
inv.add_directory(host='test_host2',
service='worker',
ip='10.1.1.10',
router='10.1.1.1',
dns=['8.8.8.8', '8.8.4.4'])
inv.save()
c = Configure(inventory=inv_file)
b1 = c.build_network_data(name='test_host1')
b2 = c.build_network_data(name='test_host2')
keys = readfile(filename='~/.ssh/id_rsa.pub').strip().split('\n')
t1 = Networkdata()\
.with_defaults()\
.with_ip(ip='10.1.1.11')\
.with_nameservers(nameservers=['8.8.8.8', '8.8.4.4'])\
.with_gateway(gateway='10.1.1.1')
t2 = Networkdata()\
.with_defaults()\
.with_ip(ip='10.1.1.10')\
.with_nameservers(nameservers=['8.8.8.8', '8.8.4.4'])\
.with_gateway(gateway='10.1.1.1')
assert (t1.content == b1.content)
assert (t2.content == b2.content)
os.system("rm -f " + path_expand(inv_file))
def test_build_user_data(self):
HEADING()
inv_file = '~/.cloudmesh/config_test.yaml'
inv = Inventory(inv_file)
inv.add_directory(host='test_host1',
keyfile='~/.ssh/id_rsa.pub',
service='manager')
inv.add_directory(host='test_host2', service='worker')
inv.save()
c = Configure(inventory=inv_file)
b1 = c.build_user_data(name='test_host1')
b2 = c.build_user_data(name='test_host2')
keys = readfile(filename='~/.ssh/id_rsa.pub').strip().split('\n')
t1 = Userdata()\
.with_authorized_keys(keys=keys)\
.with_ssh_password_login(ssh_pwauth=False)\
.with_locale()\
.with_net_tools()\
.with_hostname(hostname='test_host1')\
.with_hosts(hosts=['127.0.0.1:test_host1'])\
.with_packages(packages='avahi-daemon')
t2 = Userdata()\
.with_default_user()\
.with_ssh_password_login()\
.with_locale()\
.with_net_tools()\
.with_hostname(hostname='test_host2')\
.with_hosts(hosts=['127.0.0.1:test_host2'])\
.with_packages(packages='avahi-daemon')
assert (t1.content == b1.content)
assert (t2.content == b2.content)
os.system("rm -f " + path_expand(inv_file))
