def test_multiple_repos(self):
"""
Tests saving and loading multiple repos.
"""
# Setup
repo1 = Repo('test-repo-1')
repo1['baseurl'] = 'http://localhost/repo1'
repo2 = Repo('test-repo-2')
repo2['baseurl'] = 'http://localhost/repo2'
repo_file = RepoFile(TEST_REPO_FILENAME)
# Test
repo_file.add_repo(repo1)
repo_file.add_repo(repo2)
repo_file.save()
# Verify
loaded = RepoFile(TEST_REPO_FILENAME)
loaded.load()
self.assertEqual(2, len(loaded.all_repos()))
found_repo1 = loaded.get_repo('test-repo-1')
self.assertTrue(found_repo1 is not None)
self.assertTrue(_repo_eq(repo1, found_repo1))
found_repo2 = loaded.get_repo('test-repo-2')
self.assertTrue(found_repo2 is not None)
self.assertTrue(_repo_eq(repo2, found_repo2))
def test_one_repo_save_and_load(self):
"""
Tests the normal flow of saving and loading, using only one repo to
minimize complications.
"""
# Setup
add_me = Repo('test-repo-1')
add_me['baseurl'] = 'http://localhost/repo'
add_me['enabled'] = 1
add_me['gpgkey'] = '/tmp/key'
add_me['sslverify'] = 0
add_me['gpgcheck'] = 0
add_me['sslcacert'] = '/tmp/sslcacert'
add_me['sslclientcert'] = '/tmp/clientcert'
repo_file = RepoFile(TEST_REPO_FILENAME)
# Test Save
repo_file.add_repo(add_me)
repo_file.save()
# Verify Save
self.assertTrue(os.path.exists(TEST_REPO_FILENAME))
# Test Load
loaded = RepoFile(TEST_REPO_FILENAME)
loaded.load()
# Verify Load
self.assertEqual(1, len(loaded.all_repos()))
found_repo = loaded.get_repo('test-repo-1')
self.assertTrue(found_repo is not None)
self.assertTrue(_repo_eq(add_me, found_repo))
def test_delete_repo(self):
"""
Tests removing an existing repo is correctly saved and loaded
"""
# Setup
repo1 = Repo('test-repo-1')
repo2 = Repo('test-repo-2')
repo_file = RepoFile(TEST_REPO_FILENAME)
repo_file.add_repo(repo1)
repo_file.add_repo(repo2)
repo_file.save()
# Test
repo_file.remove_repo_by_name('test-repo-1')
repo_file.save()
# Verify
loaded = RepoFile(TEST_REPO_FILENAME)
loaded.load()
self.assertEqual(1, len(loaded.all_repos()))
self.assertTrue(loaded.get_repo('test-repo-1') is None)
self.assertTrue(loaded.get_repo('test-repo-2') is not None)
def test_no_repos_save_load(self):
"""
Tests that saving and loading a file with no repos is successful.
"""
# Test
repo_file = RepoFile(TEST_REPO_FILENAME)
repo_file.save()
# Verify
loaded = RepoFile(TEST_REPO_FILENAME)
loaded.load()
# Verify
self.assertEqual(0, len(loaded.all_repos()))
def test_unbind_repo_with_mirrorlist(self):
"""
Tests that unbinding a repo that had a mirror list deletes the mirror list
file.
"""
# Setup
url_list = ['http://pulp1', 'http://pulp2', 'http://pulp3']
repolib.bind(TEST_REPO_FILENAME, TEST_MIRROR_LIST_FILENAME, TEST_KEYS_DIR, TEST_CERT_DIR, REPO_ID, REPO_NAME, url_list, {}, None, None, ENABLED, LOCK)
self.assertTrue(os.path.exists(TEST_MIRROR_LIST_FILENAME))
# Test
repolib.unbind(TEST_REPO_FILENAME, TEST_MIRROR_LIST_FILENAME, TEST_KEYS_DIR, TEST_CERT_DIR, REPO_ID, LOCK)
# Verify
repo_file = RepoFile(TEST_REPO_FILENAME)
repo_file.load()
self.assertEqual(0, len(repo_file.all_repos()))
self.assertTrue(not os.path.exists(TEST_MIRROR_LIST_FILENAME))
def test_bind_new_file(self):
"""
Tests binding a repo when the underlying .repo file does not exist.
"""
# Test
url_list = ['http://pulpserver']
repolib.bind(TEST_REPO_FILENAME, TEST_MIRROR_LIST_FILENAME, TEST_KEYS_DIR, TEST_CERT_DIR, REPO_ID, REPO_NAME, url_list, {}, CACERT, CLIENTCERT, ENABLED, LOCK)
# Verify
self.assertTrue(os.path.exists(TEST_REPO_FILENAME))
self.assertTrue(not os.path.exists(TEST_MIRROR_LIST_FILENAME))
repo_file = RepoFile(TEST_REPO_FILENAME)
repo_file.load()
self.assertEqual(1, len(repo_file.all_repos()))
loaded = repo_file.get_repo(REPO_ID)
self.assertTrue(loaded is not None)
self.assertEqual(loaded['name'], REPO_NAME)
self.assertTrue(loaded['enabled'])
self.assertEqual(loaded['gpgcheck'], '0')
self.assertEqual(loaded['gpgkey'], None)
self.assertEqual(loaded['baseurl'], url_list[0])
self.assertTrue('mirrorlist' not in loaded)
path = loaded['sslcacert']
f = open(path)
content = f.read()
f.close()
self.assertEqual(CACERT, content)
path = loaded['sslclientcert']
f = open(path)
content = f.read()
f.close()
self.assertEqual(CLIENTCERT, content)
self.assertTrue(loaded['sslverify'], '1')
def test_unbind_repo_exists(self):
"""
Tests the normal case of unbinding a repo that exists in the repo file.
"""
# Setup
repoid = 'test-unbind-repo'
repo_file = RepoFile(TEST_REPO_FILENAME)
repo_file.add_repo(Repo(repoid))
repo_file.save()
# Test
repolib.unbind(TEST_REPO_FILENAME, TEST_MIRROR_LIST_FILENAME, TEST_KEYS_DIR, TEST_CERT_DIR, 'test-unbind-repo', LOCK)
# verify
repo_file = RepoFile(TEST_REPO_FILENAME)
repo_file.load(allow_missing=False) # the file should still be there, so error if it doesn't
self.assertEqual(0, len(repo_file.all_repos()))
certdir = os.path.join(TEST_CERT_DIR, repoid)
self.assertFalse(os.path.exists(certdir))
def test_bind_existing_file(self):
"""
Tests binding a new repo when the underlying file exists and has repos in it
(the existing repo shouldn't be deleted).
"""
# Setup
repo_file = RepoFile(TEST_REPO_FILENAME)
repo_file.add_repo(Repo('existing-repo-1'))
repo_file.save()
# Test
url_list = ['http://pulpserver']
repolib.bind(TEST_REPO_FILENAME, TEST_MIRROR_LIST_FILENAME, TEST_KEYS_DIR, TEST_CERT_DIR, REPO_ID, REPO_NAME, url_list, {}, None, None, ENABLED, LOCK)
# Verify
self.assertTrue(os.path.exists(TEST_REPO_FILENAME))
repo_file = RepoFile(TEST_REPO_FILENAME)
repo_file.load()
self.assertEqual(2, len(repo_file.all_repos()))
