def test_transfer_entitlement_keys(self):
"""Test the entitlement keys transfer method of the subscription token transfer task."""
# simulate entitlement keys not existing
with tempfile.TemporaryDirectory() as sysroot:
task = TransferSubscriptionTokensTask(
sysroot=sysroot, transfer_subscription_tokens=True)
task._copy_pem_files = Mock()
task._copy_pem_files.return_value = True
task._transfer_entitlement_keys()
sysroot_path = util.join_paths(
sysroot,
TransferSubscriptionTokensTask.RHSM_ENTITLEMENT_KEYS_PATH)
task._copy_pem_files.assert_called_once_with(
TransferSubscriptionTokensTask.RHSM_ENTITLEMENT_KEYS_PATH,
sysroot_path)
# simulate entitlement keys not existing
# - this is a critical error and should raise an exception
# (without proper certificates and keys the target system
# would be unable to communicate with the Red Hat subscription
# infrastructure)
with tempfile.TemporaryDirectory() as sysroot:
task = TransferSubscriptionTokensTask(
sysroot=sysroot, transfer_subscription_tokens=True)
task._copy_pem_files = Mock()
task._copy_pem_files.return_value = False
with self.assertRaises(SubscriptionTokenTransferError):
task._transfer_entitlement_keys()
def test_no_transfer(self):
"""Test transfer_tokens being False is handled correctly for token transfer task."""
# if transfer_subscription_tokens is False, only system purpose tokens should be
# transferred and others ignored
with tempfile.TemporaryDirectory() as sysroot:
task = TransferSubscriptionTokensTask(
sysroot=sysroot, transfer_subscription_tokens=False)
task._transfer_system_purpose = Mock()
task._transfer_entitlement_keys = Mock()
task._transfer_file = Mock()
# run the task
task.run()
# only the system purpose transfer method should have been called
task._transfer_system_purpose.assert_called_once()
task._transfer_entitlement_keys.assert_not_called()
task._transfer_file.assert_not_called()
def test_transfer(self):
"""Test transfer_tokens being True is handled correctly for token transfer task."""
# If transfer_subscription_tokens is True, all token should be transferred.
# As we test each transfer method individually we just check here that all
# expected method are called.
with tempfile.TemporaryDirectory() as sysroot:
task = TransferSubscriptionTokensTask(
sysroot=sysroot, transfer_subscription_tokens=True)
task._transfer_system_purpose = Mock()
task._transfer_entitlement_keys = Mock()
task._transfer_file = Mock()
# run the task
task.run()
# all the transfer operations should have been done
task._transfer_system_purpose.assert_called_once()
task._transfer_entitlement_keys.assert_called_once()
task._transfer_file.assert_has_calls([
call('/etc/pki/consumer/key.pem', 'RHSM consumer key'),
call('/etc/pki/consumer/cert.pem', 'RHSM consumer cert'),
call('/etc/yum.repos.d/redhat.repo', 'RHSM repo file'),
call('/etc/rhsm/rhsm.conf', 'RHSM config file')
])