Source code for rhui4_tests.test_sync_management

'''Repo syncing and scheduling tests'''

from os.path import basename

import logging
import nose
import yaml

from rhui4_tests_lib.conmgr import ConMgr
from rhui4_tests_lib.rhuimanager import RHUIManager
from rhui4_tests_lib.rhuimanager_repo import RHUIManagerRepo
from rhui4_tests_lib.rhuimanager_sync import RHUIManagerSync
from rhui4_tests_lib.rhuimanager_entitlement import RHUIManagerEntitlements
from rhui4_tests_lib.util import Util

logging.basicConfig(level=logging.DEBUG)

RHUA = ConMgr.connect()


[docs]
class TestSync():
    '''
       class for repository synchronization tests
    '''

    def __init__(self):
        # Test the RHEL-7 x86_64 repo
        version = 7
        arch = "x86_64"
        with open("/etc/rhui4_tests/tested_repos.yaml", encoding="utf-8") as configfile:
            doc = yaml.safe_load(configfile)
            try:
                self.yum_repo_name = doc["yum_repos"][version][arch]["name"]
                self.yum_repo_version = doc["yum_repos"][version][arch]["version"]
                self.yum_repo_kind = doc["yum_repos"][version][arch]["kind"]
            except KeyError as version:
                raise nose.SkipTest(f"No test repo defined for RHEL {version} on {arch}.")


[docs]
    @staticmethod
    def setup_class():
        '''
           announce the beginning of the test run
        '''
        print(f"*** Running {basename(__file__)}: ***")



[docs]
    def test_01_setup(self):
        '''add a repo to sync '''
        RHUIManager.initial_run(RHUA)
        RHUIManagerEntitlements.upload_rh_certificate(RHUA)
        entlist = RHUIManagerEntitlements.list_rh_entitlements(RHUA)
        nose.tools.assert_not_equal(len(entlist), 0)
        nose.tools.ok_(self.yum_repo_name in entlist)
        RHUIManagerRepo.add_rh_repo_by_repo(RHUA, [Util.format_repo(self.yum_repo_name,
                                                                    self.yum_repo_version,
                                                                    self.yum_repo_kind)])



[docs]
    def test_02_sync_repo(self):
        '''sync a RH repo '''
        RHUIManagerSync.sync_repo(RHUA, [Util.format_repo(self.yum_repo_name,
                                                          self.yum_repo_version)])



[docs]
    def test_03_check_sync_started(self):
        '''ensure that the sync started'''
        RHUIManagerSync.check_sync_started(RHUA, [Util.format_repo(self.yum_repo_name,
                                                                   self.yum_repo_version)])



[docs]
    def test_04_wait_till_repo_synced(self):
        '''wait until the repo is synced'''
        RHUIManagerSync.wait_till_repo_synced(RHUA, [Util.format_repo(self.yum_repo_name,
                                                                      self.yum_repo_version)])



[docs]
    def test_05_export_repo(self):
        '''export the repo'''
        RHUIManagerSync.export_repos(RHUA, [Util.format_repo(self.yum_repo_name,
                                                             self.yum_repo_version)])



[docs]
    def test_99_cleanup(self):
        '''remove the RH repo and cert'''
        RHUIManagerRepo.delete_repo(RHUA,
                                    [Util.format_repo(self.yum_repo_name, self.yum_repo_version)])
        RHUIManager.remove_rh_certs(RHUA)



[docs]
    @staticmethod
    def teardown_class():
        '''
           announce the end of the test run
        '''
        print(f"*** Finished running {basename(__file__)}. ***")