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'] == [