def test_flatten_orgs(self):
cfg = utils.cfg_init()
self.assertEquals(False, utils.get_flatten_orgs(cfg))
self.unmock(utils, 'cfg_init')
cfg = utils.cfg_init(config_file="../etc/splice/checkin.test.conf")
cfg.set('katello', 'flatten_orgs', 'True')
self.assertEquals(True, utils.get_autoentitle(cfg))
def test_multisw_config(self):
# test first with mocked config (single spacewalk)
cfg = utils.cfg_init()
self.assertEquals(['spacewalk'], utils.get_multi_sw_cfg(cfg))
# unmock cfg_init, which is mocked in base.py
self.unmock(utils, 'cfg_init')
cfg = utils.cfg_init(config_file="../etc/splice/checkin.test.conf")
self.assertTrue('spacewalk_foo' in utils.get_multi_sw_cfg(cfg))
self.assertTrue('spacewalk_bar' in utils.get_multi_sw_cfg(cfg))
self.assertTrue('spacewalk_baz' in utils.get_multi_sw_cfg(cfg))
def main(options):
global CONFIG
CONFIG = utils.cfg_init(config_file=constants.SPLICE_CHECKIN_CONFIG)
_LOG.info("run starting")
socket.setdefaulttimeout(CONFIG.getfloat('main', 'socket_timeout'))
if options.spacewalk_sync:
spacewalk_sync(options)
if options.splice_sync:
splice_sync(options)
_LOG.info("run complete")
def main(options):
global CONFIG
CONFIG = utils.cfg_init(config_file=constants.SPLICE_CHECKIN_CONFIG)
_LOG.info("run starting")
socket.setdefaulttimeout(CONFIG.getfloat('main', 'socket_timeout'))
if options.spacewalk_sync:
# ensure we don't try to sync users from multiple spacewalks
if len(utils.get_multi_sw_cfg(CONFIG)) > 1 and \
CONFIG.has_option('main', 'sync_users') and \
CONFIG.getboolean('main', 'sync_users'):
raise RuntimeError("cannot sync users from multiple satellites")
spacewalk_sync(options)
if options.splice_sync:
splice_sync(options)
_LOG.info("run complete")
from katello.client.api.organization import OrganizationAPI
from katello.client.api.environment import EnvironmentAPI
from katello.client.api.system import SystemAPI
from katello.client.api.permission import PermissionAPI
from katello.client.api.provider import ProviderAPI
from katello.client.api.user import UserAPI
from katello.client.api.distributor import DistributorAPI
from katello.client.api.user_role import UserRoleAPI
from katello.client.api.custom_info import CustomInfoAPI
from katello.client import server
from katello.client.server import BasicAuthentication
import logging
from spacewalk_splice_tool import utils, constants
_LOG = logging.getLogger(__name__)
CONFIG = utils.cfg_init(config_file=constants.SPLICE_CHECKIN_CONFIG)
class NotFoundException():
pass
class KatelloConnection():
def __init__(self):
self.orgapi = OrganizationAPI()
self.systemapi = SystemAPI()
self.userapi = UserAPI()
self.envapi = EnvironmentAPI()
self.rolesapi = UserRoleAPI()
self.permissionapi = PermissionAPI()
def test_config_reinit(self):
# unmock cfg_init, which is mocked in base.py
self.unmock(utils, 'cfg_init')
cfg = utils.cfg_init(config_file="../etc/splice/checkin.test.conf")
self.assertEquals(443, cfg.getint("splice", "port"))
def __init__(self):
# TODO: this should get pulled in from the caller instead of being read twice
global CONFIG
CONFIG = utils.cfg_init(config_file=constants.SPLICE_CHECKIN_CONFIG)
