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)
