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_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_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_add_duplicate(self):
"""
Tests that adding a repo that already exists throws a duplication error.
"""
# Setup
repo_file = RepoFile(TEST_REPO_FILENAME)
repo_file.add_repo(Repo('foo'))
# Test
self.assertRaises(DuplicateSectionError, repo_file.add_repo, Repo('foo'))
def test_delete_repo_no_repo(self):
"""
Ensures that an error is not thrown when a repo that does not exist is
deleted from the repo file.
"""
# Setup
repo_file = RepoFile(TEST_REPO_FILENAME)
repo_file.add_repo(Repo('test-repo-1'))
# Test
repo_file.remove_repo_by_name('foo')
# Verify
self.assertTrue(repo_file.get_repo('test-repo-1') is not None)
def test_mirrorlist_not_baseurl(self):
"""
Tests that if a mirrorlist is specified, a baseurl entry isn't written to the
saved repo file.
"""
# Setup
repo = Repo('test-repo-1')
repo['mirrorlist'] = 'file://etc/pulp/mirrorlist'
repo_file = RepoFile(TEST_REPO_FILENAME)
repo_file.add_repo(repo)
repo_file.save()
# Test
loaded = RepoFile(TEST_REPO_FILENAME)
loaded.load()
loaded_repo = loaded.get_repo('test-repo-1')
self.assertEqual(loaded_repo['mirrorlist'], 'file://etc/pulp/mirrorlist')
self.assertTrue('baseurl' not in loaded_repo)
def test_update_repo(self):
"""
Tests that updating an existing repo is correctly saved.
"""
# Setup
repo1 = Repo('test-repo-1')
repo1['baseurl'] = 'http://localhost/repo1'
repo_file = RepoFile(TEST_REPO_FILENAME)
repo_file.add_repo(repo1)
repo_file.save()
# Test
repo1['baseurl'] = 'http://localhost/repo-updated'
repo_file.update_repo(repo1)
repo_file.save()
# Verify
loaded = RepoFile(TEST_REPO_FILENAME)
loaded.load()
found_repo = loaded.get_repo('test-repo-1')
self.assertEqual(found_repo['baseurl'], 'http://localhost/repo-updated')
def bind(repo_filename,
mirror_list_filename,
keys_root_dir,
cert_root_dir,
repo_id,
repo_data,
url_list,
gpg_keys,
cacert,
clientcert,
enabled,
lock=None):
"""
Uses the given data to safely bind a repo to a repo file. This call will
determine the best method for representing the repo given the data in the
repo object as well as the list of URLs where the repo can be found.
The default lock is defined at the module level and is
used to ensure that concurrent access to the give files is prevented. Specific
locks can be passed in for testing purposes to circumvent the default
location of the lock which requires root access.
@param repo_filename: full path to the location of the repo file in which
the repo will be bound; this file does not need to
exist prior to this call
@type repo_filename: string
@param mirror_list_filename: full path to the location of the mirror list file
that should be written for the given repo if
necessary; this should be unique for the given repo
@type mirror_list_filename: string
@param keys_root_dir: absolute path to the root directory in which the keys for
all repos will be stored
@type keys_root_dir: string
@param cert_root_dir: absolute path to the root directory in which the certs for
all repos will be stored
@type cert_root_dir: string
@param repo_id: uniquely identifies the repo being updated
@type repo_id: string
@param repo_data: contains data on the repo being bound
@type repo_data: dict {string: string}
@param url_list: list of URLs that will be used to access the repo; this call
will determine the best way to represent the URL list in
the repo definition
@type url_list: list of strings
@param gpg_keys: mapping of key name to contents for GPG keys to be used when
verifying packages from this repo
@type gpg_keys: dict {string: string}
@param cacert: The CA certificate (PEM).
@type cacert: str
@param clientcert: The client certificate (PEM).
@type clientcert: str
@param lock: if the default lock is unacceptble, it may be overridden in this variable
@type lock: L{Lock}
"""
if not lock:
lock = Lock('/var/run/subsys/pulp/repolib.pid')
lock.acquire()
try:
log.info('Binding repo [%s]' % repo_id)
repo_file = RepoFile(repo_filename)
repo_file.load()
# In the case of an update, only the changed values will have been sent.
# Therefore, any of the major data components (repo data, url list, keys)
# may be None.
if repo_data is not None:
repo = _convert_repo(repo_id, enabled, repo_data['display_name'])
else:
repo = repo_file.get_repo(repo_id)
if gpg_keys is not None:
_handle_gpg_keys(repo, gpg_keys, keys_root_dir)
_handle_certs(repo, cert_root_dir, cacert, clientcert)
if url_list is not None:
_handle_host_urls(repo, url_list, mirror_list_filename)
if repo_file.get_repo(repo.id):
log.info('Updating existing repo [%s]' % repo.id)
repo_file.update_repo(repo)
else:
log.info('Adding new repo [%s]' % repo.id)
repo_file.add_repo(repo)
repo_file.save()
finally:
lock.release()
