def test_with_cli_params_2(self):
os.environ.clear()
sys.argv = [
'prog', '--user', 'testuser', '--password', 'testpw31!', '--port',
'1234', '-a', '/path/to/atlas.yaml', '-m',
'tests/inventory_config.yaml', '-l', '--sleep', '180', '-t', '300',
'--v'
]
parse_params(logger)
self.assertEqual(os.getenv('USER'), 'testuser',
'The user was not set correctly!')
self.assertEqual(os.getenv('PASSWORD'), 'testpw31!',
'The password was not set correctly!')
self.assertEqual(os.getenv('PORT'), '1234',
'The port was not set correctly!')
self.assertEqual(os.getenv('ATLAS'), '/path/to/atlas.yaml',
'Atlas was not set correctly')
self.assertEqual(os.getenv('INVENTORY_CONFIG'),
'tests/inventory_config.yaml', 'Config was not set')
self.assertEqual(os.getenv('LOOPBACK'), '1',
'Loopback was not set correctly')
self.assertEqual(os.getenv('SLEEP'), '180',
'Sleep time was not set correctly')
self.assertEqual(os.getenv('TIMEOUT'), '300',
'Timeout time was not set correctly')
self.assertEqual(logger.level, 20)
def test_with_cli_and_env_params(self):
sys.argv = [
'prog', '--user', 'cli_testuser', '--password', 'testpw31!',
'--port', '1234', '-a', '/path/to/atlas.yaml', '-m',
'tests/inventory_config.yaml', '-l', '-s', '180', '-t', '300'
]
os.environ['USER'] = '******'
os.environ['PASSWORD'] = '******'
os.environ['PORT'] = '1123'
os.environ['ATLAS'] = '/wrong/path/to/atlas.yaml'
os.environ['INVENTORY_CONFIG'] = 'wrong/path/to/config.yaml'
os.environ['LOOPBACK'] = '0'
os.environ['SLEEP'] = '199'
os.environ['TIMEOUT'] = '301'
parse_params(logger)
# cli params preferred
self.assertEqual(os.getenv('USER'), 'cli_testuser',
'The user was not set correctly!')
self.assertEqual(os.getenv('PASSWORD'), 'testpw31!',
'The password was not set correctly!')
self.assertEqual(os.getenv('PORT'), '1234',
'The port was not set correctly!')
self.assertEqual(os.getenv('ATLAS'), '/path/to/atlas.yaml',
'Atlas was not set correctly')
self.assertEqual(os.getenv('INVENTORY_CONFIG'),
'tests/inventory_config.yaml', 'Config was not set')
self.assertEqual(os.getenv('LOOPBACK'), '1',
'Loopback was not set correctly')
self.assertEqual(os.getenv('SLEEP'), '180',
'Sleep time was not set correctly')
self.assertEqual(os.getenv('TIMEOUT'), '300',
'Timeout time was not set correctly')
self.assertEqual(logger.level, 30)
def test_env_params(self):
os.environ.clear()
os.environ['USER'] = '******'
os.environ['PASSWORD'] = '******'
os.environ['PORT'] = '1234'
os.environ['ATLAS'] = '/path/to/atlas.yaml'
os.environ['INVENTORY_CONFIG'] = 'tests/inventory_config.yaml'
os.environ['LOOPBACK'] = '0'
os.environ['SLEEP'] = '180'
os.environ['TIMEOUT'] = '60'
parse_params(logger)
self.assertEqual(os.getenv('USER'), 'testuser',
'The user was not set correctly!')
self.assertEqual(os.getenv('PASSWORD'), 'testpw31!',
'The password was not set correctly!')
self.assertEqual(os.getenv('PORT'), '1234',
'The port was not set correctly!')
self.assertEqual(os.getenv('ATLAS'), '/path/to/atlas.yaml',
'Atlas was not set correctly')
self.assertEqual(os.getenv('INVENTORY_CONFIG'),
'tests/inventory_config.yaml', 'Config was not set')
self.assertEqual(os.getenv('LOOPBACK'), '0',
'Loopback was not set correctly')
self.assertEqual(os.getenv('SLEEP'), '180',
'Sleep time was not set correctly')
self.assertEqual(os.getenv('TIMEOUT'), '60',
'Timeout time was not set correctly')
def test_with_cli_and_env_params(self):
sys.argv = [
'prog', '--user', 'cli_testuser', '--password', 'testpw31!',
'--port', '1234', "--debug", '-a', '/path/to/atlas.yaml', '-l',
'-s', '180'
]
os.environ['USER'] = '******'
os.environ['PASSWORD'] = '******'
os.environ['PORT'] = '1123'
os.environ['ATLAS'] = '/wrong/path/to/atlas.yaml'
os.environ['LOOPBACK'] = '0'
os.environ['SLEEP'] = '199'
parse_params()
# cli params preferred
self.assertEqual(os.getenv('USER'), 'cli_testuser',
'The user was not set correctly!')
self.assertEqual(os.getenv('PASSWORD'), 'testpw31!',
'The password was not set correctly!')
self.assertEqual(os.getenv('PORT'), '1234',
'The port was not set correctly!')
self.assertEqual(os.getenv('ATLAS'), '/path/to/atlas.yaml',
'Atlas was not set correctly')
self.assertEqual(os.getenv('LOOPBACK'), '1',
'Loopback was not set correctly')
self.assertEqual(os.getenv('SLEEP'), '180',
'Sleep time was not set correctly')
def test_with_bogus_options(self):
os.environ.clear()
sys.argv = ['prog', '-z', 'foo', '-x', 'bar', '-w', 'bar']
with self.assertRaises(SystemExit) as se:
parse_params(logger)
self.assertEqual(se.exception.code, 2, 'PORT, USER, ATLAS or PASSWORD are not set properly in ENV or command '
'line!')
def test_without_params(self):
os.environ.clear()
sys.argv = ['prog']
with self.assertRaises(SystemExit) as se:
parse_params()
self.assertEqual(
se.exception.code, 0,
'PORT, USER, ATLAS or PASSWORD are not set in ENV or command line!'
)
def test_with_cli_params_1(self):
sys.argv = ['prog', '-u', 'testuser', '-p', 'testpw31!', '-o', '1234',
'-a', '/path/to/atlas.yaml', '-l', '-s', '180', '-t', '300', '-d']
parse_params(logger)
self.assertEqual(os.getenv('USER'), 'testuser', 'The user was not set correctly!')
self.assertEqual(os.getenv('PASSWORD'), 'testpw31!', 'The password was not set correctly!')
self.assertEqual(os.getenv('PORT'), '1234', 'The port was not set correctly!')
self.assertEqual(os.getenv('ATLAS'), '/path/to/atlas.yaml', 'Atlas was not set correctly')
self.assertEqual(os.getenv('LOOPBACK'), '1', 'Loopback was not set correctly')
self.assertEqual(os.getenv('SLEEP'), '180', 'Sleep time was not set correctly')
self.assertEqual(os.getenv('TIMEOUT'), '300', 'Timeout time was not set correctly')
self.assertEqual(logger.level, 10)
def test_with_cli_params_1(self):
sys.argv = [
'prog', '-u', 'testuser', '-p', 'testpw31!', '-o', '1234', '-d',
'-a', '/path/to/atlas.yaml'
]
parse_params()
self.assertEqual(os.getenv('USER'), 'testuser',
'The user was not set correctly!')
self.assertEqual(os.getenv('PASSWORD'), 'testpw31!',
'The password was not set correctly!')
self.assertEqual(os.getenv('PORT'), '1234',
'The port was not set correctly!')
self.assertEqual(os.getenv('DEBUG'), '1',
'Debug was not set correctly!')
self.assertEqual(os.getenv('ATLAS'), '/path/to/atlas.yaml',
'Atlas was not set correctly')
def test_with_cli_params_2(self):
os.environ.clear()
sys.argv = [
'prog', '--user', 'testuser', '--password', 'testpw31!', '--port',
'1234', '-a', '/path/to/atlas.yaml', '-l', '--sleep', '180'
]
parse_params()
self.assertEqual(os.getenv('USER'), 'testuser',
'The user was not set correctly!')
self.assertEqual(os.getenv('PASSWORD'), 'testpw31!',
'The password was not set correctly!')
self.assertEqual(os.getenv('PORT'), '1234',
'The port was not set correctly!')
self.assertEqual(os.getenv('DEBUG'), '0',
'Debug was not set correctly')
self.assertEqual(os.getenv('ATLAS'), '/path/to/atlas.yaml',
'Atlas was not set correctly')
self.assertEqual(os.getenv('LOOPBACK'), '1',
'Loopback was not set correctly')
self.assertEqual(os.getenv('SLEEP'), '180',
'Sleep time was not set correctly')