def test_save_inventory(self): """ Test save process """ # Copy real data into temp_file temp_inventory_file = mktemp(prefix='test_inventory') temp_bkp_inventory_file = mktemp(prefix='test_bkp_inventory') try: copyfile(TEST_FILES_PATH + "hosts3", temp_inventory_file) ansible_file = AnsibleInventoryManager( inventory_file=temp_inventory_file) ansible_file.save_inventory(backup_file=temp_bkp_inventory_file) bk_ansible_file = AnsibleInventoryManager( inventory_file=temp_bkp_inventory_file) self.assertEqual(ansible_file.get_hosts(), bk_ansible_file.get_hosts()) self.assertEqual( [group.name for group in ansible_file.get_groups()], [group.name for group in bk_ansible_file.get_groups()]) ansible_file.add_host("host7") ansible_file.save_inventory(backup_file=temp_bkp_inventory_file) bk_ansible_file = AnsibleInventoryManager( inventory_file=temp_bkp_inventory_file) self.assertNotEqual(ansible_file.get_hosts(), bk_ansible_file.get_hosts()) self.assertEqual( [group.name for group in ansible_file.get_groups()], [group.name for group in bk_ansible_file.get_groups()]) finally: os.unlink(temp_inventory_file) os.unlink(temp_bkp_inventory_file)
def test_save_inventory(self): """ Test save process """ # Copy real data into temp_file temp_inventory_file = mktemp(prefix='test_inventory') temp_bkp_inventory_file = mktemp(prefix='test_bkp_inventory') try: copyfile(TEST_FILES_PATH + "hosts3", temp_inventory_file) ansible_file = AnsibleInventoryManager(inventory_file=temp_inventory_file) ansible_file.save_inventory(backup_file=temp_bkp_inventory_file) bk_ansible_file = AnsibleInventoryManager(inventory_file=temp_bkp_inventory_file) self.assertEqual(ansible_file.get_hosts(), bk_ansible_file.get_hosts()) self.assertEqual([group.name for group in ansible_file.get_groups()], [group.name for group in bk_ansible_file.get_groups()]) ansible_file.add_host("host7") ansible_file.save_inventory(backup_file=temp_bkp_inventory_file) bk_ansible_file = AnsibleInventoryManager(inventory_file=temp_bkp_inventory_file) self.assertNotEqual(ansible_file.get_hosts(), bk_ansible_file.get_hosts()) self.assertEqual([group.name for group in ansible_file.get_groups()], [group.name for group in bk_ansible_file.get_groups()]) finally: os.unlink(temp_inventory_file) os.unlink(temp_bkp_inventory_file)