from datasciencebox.core.project import Project
from datasciencebox.core.cloud.cluster import Cluster
from datasciencebox.core.cloud.instance import Instance
cluster = Cluster()
cluster.instances.append(Instance(ip='0.0.0.0', username='******', keypair='/home/ubuntu/.ssh/id_rsa'))
cluster.instances.append(Instance(ip='1.1.1.1:2222',
username='******',
keypair='/home/ubuntu/.ssh/id_rsa2'))
cluster.instances.append(Instance(ip='2.2.2.2',
port='3333',
username='******',
keypair='/home/ubuntu/.ssh/id_rsa3'))
project = Project()
project.cluster = cluster
head_roles = ['head', 'head2', 'conda']
compute_roles = ['minion2', 'conda']
salt.HEAD_ROLES = head_roles
salt.COMPUTE_ROLES = compute_roles
def test_generate_salt_ssh_master_conf(tmpdir):
path = tmpdir.dirname
p = Project(path=path)
master = salt.generate_salt_ssh_master_conf(p)
assert master['root_dir'] == os.path.join(p.settings_dir)
assert master['cachedir'] == os.path.join(p.settings_dir, 'var', 'cache', 'salt')
assert master['file_roots']['base'] == [os.path.join(p.settings_dir, 'salt')]
assert master['pillar_roots']['base'] == [os.path.join(p.settings_dir, 'pillar')]
cluster = Cluster()
cluster.instances.append(
Instance(ip='0.0.0.0', username='******', keypair='/home/ubuntu/.ssh/id_rsa'))
cluster.instances.append(
Instance(ip='1.1.1.1:2222',
username='******',
keypair='/home/ubuntu/.ssh/id_rsa2'))
cluster.instances.append(
Instance(ip='2.2.2.2',
port='3333',
username='******',
keypair='/home/ubuntu/.ssh/id_rsa3'))
project = Project()
project.cluster = cluster
head_roles = ['head', 'head2', 'conda']
compute_roles = ['minion2', 'conda']
salt.HEAD_ROLES = head_roles
salt.COMPUTE_ROLES = compute_roles
def test_generate_salt_ssh_master_conf(tmpdir):
path = tmpdir.dirname
p = Project(path=path)
master = salt.generate_salt_ssh_master_conf(p)
assert master['root_dir'] == os.path.join(p.settings_dir)
assert master['cachedir'] == os.path.join(p.settings_dir, 'var', 'cache',
'salt')
assert master['file_roots']['base'] == [
