def test_multiple_entries_save_load(self):
"""
Tests creating a new mirror list file with multiple entries, saving it, and then
loading it back from disk.
"""
# Test Save
mirror_list = MirrorListFile(TEST_MIRROR_LIST_FILENAME)
mirror_list.add_entry('http://cds-01')
mirror_list.add_entry('http://cds-02')
mirror_list.save()
# Verify Save
self.assertTrue(os.path.exists(TEST_MIRROR_LIST_FILENAME))
# Test Load
loaded = MirrorListFile(TEST_MIRROR_LIST_FILENAME)
loaded.load()
# Verify Load
self.assertEqual(2, len(loaded.entries))
# Make sure the entries are returned in the correct order
self.assertEqual('http://cds-01', loaded.entries[0])
self.assertEqual('http://cds-02', loaded.entries[1])
def test_add_entries(self):
"""
Tests the ability to add a list of entries in a single operation.
"""
# Setup
mirror_list = MirrorListFile(TEST_MIRROR_LIST_FILENAME)
mirror_list.add_entry('http://cds-01')
add_us = ['http://cds-02', 'http://cds-03']
# Test
mirror_list.add_entries(add_us)
# Verify
self.assertEqual(3, len(mirror_list.entries))
