add el8toel9 upgrade facts for Satellite

repos/system_upgrade/el8toel9/actors/satellite_upgrade_facts/actor.py

@@ -0,0 +1,75 @@
+from leapp.actors import Actor
+from leapp.libraries.common.config import architecture
+from leapp.libraries.common.rpms import has_package
+from leapp.models import (
+    InstalledRPM,
+    RepositoriesSetupTasks,
+    RpmTransactionTasks,
+    SatelliteFacts,
+    SatellitePostgresqlFacts
+)
+from leapp.tags import FactsPhaseTag, IPUWorkflowTag
+
+SATELLITE_VERSION = '6.99'
+
+RELATED_PACKAGES = ('foreman', 'foreman-proxy', 'katello', 'candlepin')
+RELATED_PACKAGE_PREFIXES = ('rubygem-hammer', 'rubygem-foreman', 'rubygem-katello',
+                            'rubygem-smart_proxy', 'python3.11-pulp', 'foreman-installer',
+                            'satellite-installer')
+
+
+class SatelliteUpgradeFacts(Actor):
+    """
+    Report which Satellite packages require updates and how to handle PostgreSQL data
+    """
+
+    name = 'satellite_upgrade_facts'
+    consumes = (InstalledRPM, )
+    produces = (RepositoriesSetupTasks, RpmTransactionTasks, SatelliteFacts)
+    tags = (IPUWorkflowTag, FactsPhaseTag)
+
+    def process(self):
+        if not architecture.matches_architecture(architecture.ARCH_X86_64):
+            return
+
+        has_foreman = has_package(InstalledRPM, 'foreman') or has_package(InstalledRPM, 'foreman-proxy')
+        if not has_foreman:
+            return
+
+        local_postgresql = has_package(InstalledRPM, 'postgresql-server')
+
+        to_install = ['rubygem-foreman_maintain']
+
+        for rpm_pkgs in self.consume(InstalledRPM):
+            for pkg in rpm_pkgs.items:
+                if pkg.name in RELATED_PACKAGES or pkg.name.startswith(RELATED_PACKAGE_PREFIXES):
+                    to_install.append(pkg.name)
+
+        if local_postgresql:
+            to_install.extend(['postgresql', 'postgresql-server'])
+            if has_package(InstalledRPM, 'postgresql-contrib'):
+                to_install.append('postgresql-contrib')
+            if has_package(InstalledRPM, 'postgresql-evr'):
+                to_install.append('postgresql-evr')
+
+        self.produce(SatelliteFacts(
+            has_foreman=has_foreman,
+            installer_has_systemchecks=False,
+            postgresql=SatellitePostgresqlFacts(
+                local_postgresql=local_postgresql,
+            ),
+        ))
+
+        satellite = has_package(InstalledRPM, 'satellite')
+        capsule = has_package(InstalledRPM, 'satellite-capsule')
+        if satellite or capsule:
+            repositories_to_enable = [f'satellite-maintenance-{SATELLITE_VERSION}-for-rhel-9-x86_64-rpms']
+            if satellite:
+                repositories_to_enable.append(f'satellite-{SATELLITE_VERSION}-for-rhel-9-x86_64-rpms')
+                to_install.append('satellite')
+            elif capsule:
+                repositories_to_enable.append(f'satellite-capsule-{SATELLITE_VERSION}-for-rhel-9-x86_64-rpms')
+                to_install.append('satellite-capsule')
+            self.produce(RepositoriesSetupTasks(to_enable=repositories_to_enable))
+
+        self.produce(RpmTransactionTasks(to_install=to_install))

repos/system_upgrade/el8toel9/actors/satellite_upgrade_facts/tests/unit_test_satellite_upgrade_facts.py

@@ -0,0 +1,137 @@
+from leapp.libraries.common.config import mock_configs
+from leapp.models import InstalledRPM, RepositoriesSetupTasks, RPM, RpmTransactionTasks, SatelliteFacts
+
+RH_PACKAGER = 'Red Hat, Inc. <http://bugzilla.redhat.com/bugzilla>'
+
+
+def fake_package(pkg_name):
+    return RPM(name=pkg_name, version='0.1', release='1.sm01', epoch='1', packager=RH_PACKAGER, arch='noarch',
+               pgpsig='RSA/SHA256, Mon 01 Jan 1970 00:00:00 AM -03, Key ID 199e2f91fd431d51')
+
+
+FOREMAN_RPM = fake_package('foreman')
+FOREMAN_PROXY_RPM = fake_package('foreman-proxy')
+KATELLO_INSTALLER_RPM = fake_package('foreman-installer-katello')
+KATELLO_RPM = fake_package('katello')
+RUBYGEM_KATELLO_RPM = fake_package('rubygem-katello')
+RUBYGEM_FOREMAN_PUPPET_RPM = fake_package('rubygem-foreman_puppet')
+POSTGRESQL_RPM = fake_package('postgresql-server')
+SATELLITE_RPM = fake_package('satellite')
+SATELLITE_CAPSULE_RPM = fake_package('satellite-capsule')
+
+SATELLITE_VERSION = '6.99'
+
+
+def test_no_satellite_present(current_actor_context):
+    current_actor_context.feed(InstalledRPM(items=[]))
+    current_actor_context.run(config_model=mock_configs.CONFIG)
+    message = current_actor_context.consume(SatelliteFacts)
+    assert not message
+
+
+def test_satellite_present(current_actor_context):
+    current_actor_context.feed(InstalledRPM(items=[FOREMAN_RPM]))
+    current_actor_context.run(config_model=mock_configs.CONFIG)
+    message = current_actor_context.consume(SatelliteFacts)[0]
+    assert message.has_foreman
+
+
+def test_wrong_arch(current_actor_context):
+    current_actor_context.feed(InstalledRPM(items=[FOREMAN_RPM]))
+    current_actor_context.run(config_model=mock_configs.CONFIG_S390X)
+    message = current_actor_context.consume(SatelliteFacts)
+    assert not message
+
+
+def test_satellite_capsule_present(current_actor_context):
+    current_actor_context.feed(InstalledRPM(items=[FOREMAN_PROXY_RPM]))
+    current_actor_context.run(config_model=mock_configs.CONFIG)
+    message = current_actor_context.consume(SatelliteFacts)[0]
+    assert message.has_foreman
+
+
+def test_no_katello_installer_present(current_actor_context):
+    current_actor_context.feed(InstalledRPM(items=[FOREMAN_RPM]))
+    current_actor_context.run(config_model=mock_configs.CONFIG)
+    message = current_actor_context.consume(SatelliteFacts)[0]
+    assert not message.installer_has_systemchecks
+
+
+def test_katello_installer_present(current_actor_context):
+    current_actor_context.feed(InstalledRPM(items=[FOREMAN_RPM, KATELLO_INSTALLER_RPM]))
+    current_actor_context.run(config_model=mock_configs.CONFIG)
+    message = current_actor_context.consume(SatelliteFacts)[0]
+    assert not message.installer_has_systemchecks
+
+
+def test_installs_related_package(current_actor_context):
+    current_actor_context.feed(InstalledRPM(items=[FOREMAN_RPM, KATELLO_RPM, RUBYGEM_KATELLO_RPM,
+                                                   RUBYGEM_FOREMAN_PUPPET_RPM]))
+    current_actor_context.run(config_model=mock_configs.CONFIG)
+    message = current_actor_context.consume(RpmTransactionTasks)[0]
+    assert 'katello' in message.to_install
+    assert 'rubygem-katello' in message.to_install
+    assert 'rubygem-foreman_puppet' in message.to_install
+
+
+def test_installs_satellite_package(current_actor_context):
+    current_actor_context.feed(InstalledRPM(items=[FOREMAN_RPM, SATELLITE_RPM]))
+    current_actor_context.run(config_model=mock_configs.CONFIG)
+    message = current_actor_context.consume(RpmTransactionTasks)[0]
+    assert 'satellite' in message.to_install
+    assert 'satellite-capsule' not in message.to_install
+
+
+def test_installs_satellite_capsule_package(current_actor_context):
+    current_actor_context.feed(InstalledRPM(items=[FOREMAN_PROXY_RPM, SATELLITE_CAPSULE_RPM]))
+    current_actor_context.run(config_model=mock_configs.CONFIG)
+    message = current_actor_context.consume(RpmTransactionTasks)[0]
+    assert 'satellite-capsule' in message.to_install
+    assert 'satellite' not in message.to_install
+
+
+def test_detects_local_postgresql(current_actor_context):
+    current_actor_context.feed(InstalledRPM(items=[FOREMAN_RPM, POSTGRESQL_RPM]))
+    current_actor_context.run(config_model=mock_configs.CONFIG)
+
+    satellitemsg = current_actor_context.consume(SatelliteFacts)[0]
+    assert satellitemsg.postgresql.local_postgresql
+
+
+def test_detects_remote_postgresql(current_actor_context):
+    current_actor_context.feed(InstalledRPM(items=[FOREMAN_RPM]))
+    current_actor_context.run(config_model=mock_configs.CONFIG)
+
+    satellitemsg = current_actor_context.consume(SatelliteFacts)[0]
+    assert not satellitemsg.postgresql.local_postgresql
+
+
+def test_enables_right_repositories_on_satellite(current_actor_context):
+    current_actor_context.feed(InstalledRPM(items=[FOREMAN_RPM, SATELLITE_RPM]))
+    current_actor_context.run(config_model=mock_configs.CONFIG)
+
+    rpmmessage = current_actor_context.consume(RepositoriesSetupTasks)[0]
+
+    assert f'satellite-maintenance-{SATELLITE_VERSION}-for-rhel-9-x86_64-rpms' in rpmmessage.to_enable
+    assert f'satellite-{SATELLITE_VERSION}-for-rhel-9-x86_64-rpms' in rpmmessage.to_enable
+    assert f'satellite-capsule-{SATELLITE_VERSION}-for-rhel-9-x86_64-rpms' not in rpmmessage.to_enable
+
+
+def test_enables_right_repositories_on_capsule(current_actor_context):
+    current_actor_context.feed(InstalledRPM(items=[FOREMAN_PROXY_RPM, SATELLITE_CAPSULE_RPM]))
+    current_actor_context.run(config_model=mock_configs.CONFIG)
+
+    rpmmessage = current_actor_context.consume(RepositoriesSetupTasks)[0]
+
+    assert f'satellite-maintenance-{SATELLITE_VERSION}-for-rhel-9-x86_64-rpms' in rpmmessage.to_enable
+    assert f'satellite-{SATELLITE_VERSION}-for-rhel-9-x86_64-rpms' not in rpmmessage.to_enable
+    assert f'satellite-capsule-{SATELLITE_VERSION}-for-rhel-9-x86_64-rpms' in rpmmessage.to_enable
+
+
+def test_enables_right_repositories_on_upstream(current_actor_context):
+    current_actor_context.feed(InstalledRPM(items=[FOREMAN_RPM]))
+    current_actor_context.run(config_model=mock_configs.CONFIG)
+
+    message = current_actor_context.consume(RepositoriesSetupTasks)
+
+    assert not message