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)