def test_options_virt(self):
for virt in ['esx', 'hyperv', 'rhevm']:
self.clearEnv()
sys.argv = [
"virtwho.py",
"--%s" % virt,
"--%s-owner=owner" % virt,
"--%s-env=env" % virt,
"--%s-server=localhost" % virt,
"--%s-username=username" % virt,
"--%s-password=password" % virt
]
_, options = parseOptions()
self.assertEqual(options.virtType, virt)
self.assertEqual(options.owner, 'owner')
self.assertEqual(options.env, 'env')
self.assertEqual(options.server, 'localhost')
self.assertEqual(options.username, 'username')
self.assertEqual(options.password, 'password')
sys.argv = ["virtwho.py"]
virt_up = virt.upper()
os.environ["VIRTWHO_%s" % virt_up] = "1"
os.environ["VIRTWHO_%s_OWNER" % virt_up] = "xowner"
os.environ["VIRTWHO_%s_ENV" % virt_up] = "xenv"
os.environ["VIRTWHO_%s_SERVER" % virt_up] = "xlocalhost"
os.environ["VIRTWHO_%s_USERNAME" % virt_up] = "xusername"
os.environ["VIRTWHO_%s_PASSWORD" % virt_up] = "xpassword"
_, options = parseOptions()
self.assertEqual(options.virtType, virt)
self.assertEqual(options.owner, 'xowner')
self.assertEqual(options.env, 'xenv')
self.assertEqual(options.server, 'xlocalhost')
self.assertEqual(options.username, 'xusername')
self.assertEqual(options.password, 'xpassword')
def test_options_hierarchy_for_reporter_id(self, parseFile, getLogger):
# Set the value in all three possible locations
# Mock /etc/virt-who.conf file
global_conf_dict = {
'global': {
'reporter_id': "/etc/virt-who.conf"
}
}
parseFile.return_value = global_conf_dict
# cli option
sys.argv = ["virtwho.py", "--reporter-id=cli"]
# environment var
os.environ["VIRTWHO_REPORTER_ID"] = "env"
_, options = parseOptions()
# cli option should beat environment vars and virt-who.conf
self.assertEqual(options.reporter_id, "cli")
sys.argv = ["virtwho.py"]
_, options = parseOptions()
self.assertEqual(options.reporter_id, "env")
self.clearEnv()
_, options = parseOptions()
self.assertEqual(options.reporter_id, "/etc/virt-who.conf")
parseFile.return_value = {'global': {}}
_, options = parseOptions()
self.assertEqual(options.reporter_id, util.generateReporterId())
def test_options_virt(self, parseFile, getLogger):
self.setUpParseFile(parseFile)
for virt in ['esx', 'hyperv', 'rhevm']:
self.clearEnv()
sys.argv = ["virtwho.py", "--%s" % virt, "--%s-owner=owner" % virt,
"--%s-env=env" % virt, "--%s-server=localhost" % virt,
"--%s-username=username" % virt,
"--%s-password=password" % virt]
_, options = parseOptions()
self.assertEqual(options.virtType, virt)
self.assertEqual(options.owner, 'owner')
self.assertEqual(options.env, 'env')
self.assertEqual(options.server, 'localhost')
self.assertEqual(options.username, 'username')
self.assertEqual(options.password, 'password')
sys.argv = ["virtwho.py"]
virt_up = virt.upper()
os.environ["VIRTWHO_%s" % virt_up] = "1"
os.environ["VIRTWHO_%s_OWNER" % virt_up] = "xowner"
os.environ["VIRTWHO_%s_ENV" % virt_up] = "xenv"
os.environ["VIRTWHO_%s_SERVER" % virt_up] = "xlocalhost"
os.environ["VIRTWHO_%s_USERNAME" % virt_up] = "xusername"
os.environ["VIRTWHO_%s_PASSWORD" % virt_up] = "xpassword"
_, options = parseOptions()
self.assertEqual(options.virtType, virt)
self.assertEqual(options.owner, 'xowner')
self.assertEqual(options.env, 'xenv')
self.assertEqual(options.server, 'xlocalhost')
self.assertEqual(options.username, 'xusername')
self.assertEqual(options.password, 'xpassword')
def test_options_debug(self):
sys.argv = ["virtwho.py", "-d"]
_, options = parseOptions()
self.assertTrue(options.debug)
sys.argv = ["virtwho.py"]
os.environ["VIRTWHO_DEBUG"] = "1"
_, options = parseOptions()
self.assertTrue(options.debug)
def test_options_debug(self):
sys.argv = ["virtwho.py", "-d"]
_, options = parseOptions()
self.assertTrue(options.debug)
sys.argv = ["virtwho.py"]
os.environ["VIRTWHO_DEBUG"] = "1"
_, options = parseOptions()
self.assertTrue(options.debug)
def test_options_debug(self, parseFile, getLogger):
self.setUpParseFile(parseFile)
sys.argv = ["virtwho.py", "-d"]
_, options = parseOptions()
self.assertTrue(options.debug)
sys.argv = ["virtwho.py"]
os.environ["VIRTWHO_DEBUG"] = "1"
_, options = parseOptions()
self.assertTrue(options.debug)
def test_sm_config_cmd(self):
os.environ = {}
sys.argv = ["virt-who", "--sam", "--libvirt"]
logger, options = parseOptions()
config = Config("env/cmdline", options.virtType, defaults={}, **options)
config.checkOptions(logger)
manager = Manager.fromOptions(logger, options, config)
self.assertTrue(isinstance(manager, SubscriptionManager))
def test_sm_config_env(self):
os.environ = {"VIRTWHO_SAM": "1", "VIRTWHO_LIBVIRT": "1"}
sys.argv = ["virt-who"]
logger, options = parseOptions()
config = Config("env/cmdline", options.virtType, defaults={}, **options)
config.checkOptions(logger)
manager = Manager.fromOptions(logger, options, config)
self.assertTrue(isinstance(manager, SubscriptionManager))
def test_minimum_interval_options(self, parseFile, getLogger):
self.setUpParseFile(parseFile)
sys.argv = ["virtwho.py", "--interval=5"]
_, options = parseOptions()
self.assertEqual(options.interval, 60)
sys.argv = ["virtwho.py"]
os.environ["VIRTWHO_INTERVAL"] = '1'
_, options = parseOptions()
self.assertEqual(options.interval, 60)
self.clearEnv()
bad_conf = {'global': {'interval': 1}}
parseFile.return_value = bad_conf
_, options = parseOptions()
self.assertEqual(options.interval, 60)
def test_default_cmdline_options(self):
sys.argv = ["virtwho.py"]
_, options = parseOptions()
self.assertFalse(options.debug)
self.assertFalse(options.background)
self.assertFalse(options.oneshot)
self.assertEqual(options.interval, 3600)
self.assertEqual(options.smType, 'sam')
self.assertEqual(options.virtType, None)
def test_default_cmdline_options(self):
sys.argv = ["virtwho.py"]
_, options = parseOptions()
self.assertFalse(options.debug)
self.assertFalse(options.background)
self.assertFalse(options.oneshot)
self.assertEqual(options.interval, 3600)
self.assertEqual(options.smType, 'sam')
self.assertEqual(options.virtType, None)
def test_default_cmdline_options(self, parseFile, getLogger):
self.setUpParseFile(parseFile)
sys.argv = ["virtwho.py"]
_, options = parseOptions()
self.assertFalse(options.debug)
self.assertFalse(options.background)
self.assertFalse(options.oneshot)
self.assertEqual(options.interval, 60)
self.assertEqual(options.smType, 'sam')
self.assertEqual(options.virtType, None)
self.assertEqual(options.reporter_id, util.generateReporterId())
def test_options_virt_satellite(self, parseFile, getLogger):
self.setUpParseFile(parseFile)
for virt in ['esx', 'hyperv', 'rhevm']:
self.clearEnv()
sys.argv = ["virtwho.py",
"--satellite",
"--satellite-server=localhost",
"--satellite-username=username",
"--satellite-password=password",
"--%s" % virt,
"--%s-server=localhost" % virt,
"--%s-username=username" % virt,
"--%s-password=password" % virt]
_, options = parseOptions()
self.assertEqual(options.virtType, virt)
self.assertEqual(options.owner, '')
self.assertEqual(options.env, '')
self.assertEqual(options.server, 'localhost')
self.assertEqual(options.username, 'username')
self.assertEqual(options.password, 'password')
sys.argv = ["virtwho.py"]
virt_up = virt.upper()
os.environ["VIRTWHO_SATELLITE"] = "1"
os.environ["VIRTWHO_SATELLITE_SERVER"] = "xlocalhost"
os.environ["VIRTWHO_SATELLITE_USERNAME"] = "******"
os.environ["VIRTWHO_SATELLITE_PASSWORD"] = "******"
os.environ["VIRTWHO_%s" % virt_up] = "1"
os.environ["VIRTWHO_%s_SERVER" % virt_up] = "xlocalhost"
os.environ["VIRTWHO_%s_USERNAME" % virt_up] = "xusername"
os.environ["VIRTWHO_%s_PASSWORD" % virt_up] = "xpassword"
_, options = parseOptions()
self.assertEqual(options.virtType, virt)
self.assertEqual(options.owner, '')
self.assertEqual(options.env, '')
self.assertEqual(options.server, 'xlocalhost')
self.assertEqual(options.username, 'xusername')
self.assertEqual(options.password, 'xpassword')
def test_satellite_config_env(self):
os.environ = {
"VIRTWHO_SATELLITE": "1",
"VIRTWHO_SATELLITE_SERVER": "sat.example.com",
"VIRTWHO_SATELLITE_USERNAME": "******",
"VIRTWHO_SATELLITE_PASSWORD": "******",
"VIRTWHO_LIBVIRT": "1",
}
sys.argv = ["virt-who"]
logger, options = parseOptions()
config = Config("env/cmdline", options.virtType, defaults={}, **options)
config.checkOptions(logger)
manager = Manager.fromOptions(logger, options, config)
self.assertTrue(isinstance(manager, Satellite))
def test_options_virt_satellite(self, getLogger):
for virt in ['esx', 'hyperv', 'rhevm']:
self.clearEnv()
sys.argv = [
"virtwho.py", "--satellite", "--satellite-server=localhost",
"--satellite-username=username",
"--satellite-password=password",
"--%s" % virt,
"--%s-server=localhost" % virt,
"--%s-username=username" % virt,
"--%s-password=password" % virt
]
_, options = parseOptions()
self.assertEqual(options.virtType, virt)
self.assertEqual(options.owner, '')
self.assertEqual(options.env, '')
self.assertEqual(options.server, 'localhost')
self.assertEqual(options.username, 'username')
self.assertEqual(options.password, 'password')
sys.argv = ["virtwho.py"]
virt_up = virt.upper()
os.environ["VIRTWHO_SATELLITE"] = "1"
os.environ["VIRTWHO_SATELLITE_SERVER"] = "xlocalhost"
os.environ["VIRTWHO_SATELLITE_USERNAME"] = "******"
os.environ["VIRTWHO_SATELLITE_PASSWORD"] = "******"
os.environ["VIRTWHO_%s" % virt_up] = "1"
os.environ["VIRTWHO_%s_SERVER" % virt_up] = "xlocalhost"
os.environ["VIRTWHO_%s_USERNAME" % virt_up] = "xusername"
os.environ["VIRTWHO_%s_PASSWORD" % virt_up] = "xpassword"
_, options = parseOptions()
self.assertEqual(options.virtType, virt)
self.assertEqual(options.owner, '')
self.assertEqual(options.env, '')
self.assertEqual(options.server, 'xlocalhost')
self.assertEqual(options.username, 'xusername')
self.assertEqual(options.password, 'xpassword')
def test_satellite_config_cmd(self):
os.environ = {}
sys.argv = [
"virt-who",
"--satellite",
"--satellite-server=sat.example.com",
"--satellite-username=username",
"--satellite-password=password",
"--libvirt",
]
logger, options = parseOptions()
config = Config("env/cmdline", options.virtType, defaults={}, **options)
config.checkOptions(logger)
manager = Manager.fromOptions(logger, options, config)
self.assertTrue(isinstance(manager, Satellite))
def test_minimum_interval_options(self, getLogger):
sys.argv = ["virtwho.py", "--interval=5"]
_, options = parseOptions()
self.assertEqual(options.interval, 60)
def test_minimum_interval_options(self, getLogger):
sys.argv = ["virtwho.py", "--interval=5"]
_, options = parseOptions()
self.assertEqual(options.interval, 60)
