def test_generate_roster():
roster = salt.generate_roster(project, mine=False)
ans = {
'head': {
'host': '0.0.0.0',
'port': 22,
'sudo': True,
'user': '******',
'priv': '/home/ubuntu/.ssh/id_rsa',
'grains': {'roles': head_roles}
},
'compute-1': {
'host': '1.1.1.1',
'port': 2222,
'sudo': True,
'user': '******',
'priv': '/home/ubuntu/.ssh/id_rsa2',
'grains': {'roles': compute_roles}
},
'compute-2': {
'host': '2.2.2.2',
'port': 3333,
'sudo': True,
'user': '******',
'priv': '/home/ubuntu/.ssh/id_rsa3',
'grains': {'roles': compute_roles}
},
}
assert roster == ans
def test_generate_roster():
roster = salt.generate_roster(cluster, mine=False)
ans = {
'master': {
'host': '0.0.0.0',
'sudo': True,
'user': '******',
'priv': '/home/ubuntu/.ssh/id_rsa',
'grains': {'roles': master_roles}
},
'minion-1': {
'host': '1.1.1.1',
'sudo': True,
'user': '******',
'priv': '/home/ubuntu/.ssh/id_rsa',
'grains': {'roles': minion_roles}
},
'minion-2': {
'host': '2.2.2.2',
'sudo': True,
'user': '******',
'priv': '/home/ubuntu/.ssh/id_rsa',
'grains': {'roles': minion_roles}
},
}
assert roster == ans
def test_generate_roster():
roster = salt.generate_roster(project, mine=False)
ans = {
'head': {
'host': '0.0.0.0',
'port': 22,
'sudo': True,
'user': '******',
'priv': '/home/ubuntu/.ssh/id_rsa',
'grains': {
'roles': head_roles
}
},
'compute-1': {
'host': '1.1.1.1',
'port': 2222,
'sudo': True,
'user': '******',
'priv': '/home/ubuntu/.ssh/id_rsa2',
'grains': {
'roles': compute_roles
}
},
'compute-2': {
'host': '2.2.2.2',
'port': 3333,
'sudo': True,
'user': '******',
'priv': '/home/ubuntu/.ssh/id_rsa3',
'grains': {
'roles': compute_roles
}
},
}
assert roster == ans
def create_roster(self):
logger.debug('Creating roster file to: %s' % self.roster_path)
with open(self.roster_path, 'w') as f:
yaml.safe_dump(salt.generate_roster(self.cluster), f, default_flow_style=False)
def create_roster_file(self):
logger.debug('Creating roster file to: %s', self.roster_path)
with open(self.roster_path, 'w') as f:
dict_ = salt.generate_roster(self)
yaml.safe_dump(dict_, f, default_flow_style=False)
def create_roster_file(self):
logger.debug('Creating roster file to: %s', self.roster_path)
with open(self.roster_path, 'w') as f:
dict_ = salt.generate_roster(self)
yaml.safe_dump(dict_, f, default_flow_style=False)
