def test_04_check_env_parse(self):
os.environ['SCT_CLUSTER_BACKEND'] = 'aws'
os.environ['SCT_REGION_NAME'] = '["eu-west-1", "us-east-1"]'
os.environ['SCT_N_DB_NODES'] = '2 2'
os.environ['SCT_INSTANCE_TYPE_DB'] = 'i3.large'
os.environ['SCT_AMI_ID_DB_SCYLLA'] = 'ami-b4f8b4cb ami-b4f8b4cb'
conf = SCTConfiguration()
conf.verify_configuration()
conf.dump_config()
self.assertEqual(conf.get('security_group_ids'), 'sg-059a7f66a947d4b5c sg-c5e1f7a0')
def test_04_check_env_parse(self):
os.environ['SCT_CLUSTER_BACKEND'] = 'aws'
os.environ['SCT_REGION_NAME'] = '["eu-west-1", "us-east-1"]'
os.environ['SCT_N_DB_NODES'] = '2 2'
os.environ['SCT_INSTANCE_TYPE_DB'] = 'i3.large'
os.environ['SCT_AMI_ID_DB_SCYLLA'] = 'ami-06f919eb ami-eae4f795'
os.environ['SCT_CONFIG_FILES'] = 'internal_test_data/minimal_test_case.yaml'
conf = SCTConfiguration()
conf.verify_configuration()
conf.dump_config()
self.assertEqual(conf.get('ami_id_db_scylla'), 'ami-06f919eb ami-eae4f795')
def test_06_libvirt(self):
os.environ['SCT_CLUSTER_BACKEND'] = 'libvirt'
os.environ['SCT_SCYLLA_VERSION'] = '3.0.3'
conf = SCTConfiguration()
conf.verify_configuration()
self.assertIn('libvirt_uri', conf.dump_config())
self.assertEqual(conf.get('libvirt_uri'), 'qemu:///system')
def test_05_docker(self):
os.environ['SCT_CLUSTER_BACKEND'] = 'docker'
os.environ['SCT_SCYLLA_VERSION'] = '3.0.3'
conf = SCTConfiguration()
conf.verify_configuration()
self.assertIn('docker_image', conf.dump_config())
self.assertEqual(conf.get('docker_image'), 'scylladb/scylla')
def test_08_baremetal(self):
os.environ['SCT_CLUSTER_BACKEND'] = 'baremetal'
os.environ['SCT_DB_NODES_PRIVATE_IP'] = '["1.2.3.4", "1.2.3.5"]'
os.environ['SCT_DB_NODES_PUBLIC_IP'] = '["1.2.3.4", "1.2.3.5"]'
conf = SCTConfiguration()
conf.verify_configuration()
self.assertIn('db_nodes_private_ip', conf.dump_config())
self.assertEqual(conf.get('db_nodes_private_ip'), ["1.2.3.4", "1.2.3.5"])
def test_12_scylla_version_repo_ubuntu(self):
os.environ['SCT_CLUSTER_BACKEND'] = 'docker'
os.environ['SCT_SCYLLA_LINUX_DISTRO'] = 'ubuntu-xenial'
os.environ['SCT_SCYLLA_VERSION'] = '3.0.3'
conf = SCTConfiguration()
conf.verify_configuration()
self.assertIn('scylla_repo', conf.dump_config())
self.assertEqual(conf.get('scylla_repo'), "https://s3.amazonaws.com/downloads.scylladb.com/deb/ubuntu/scylla-3.0-xenial.list")
def output_conf(config_files, backend):
add_file_logger()
if backend:
os.environ['SCT_CLUSTER_BACKEND'] = backend
if config_files:
os.environ['SCT_CONFIG_FILES'] = config_files
config = SCTConfiguration()
click.secho(config.dump_config(), fg='green')
sys.exit(0)
def test_12_k8s_scylla_version_ubuntu_loader_centos(self): # pylint: disable=invalid-name
os.environ['SCT_CLUSTER_BACKEND'] = 'k8s-gce-minikube'
os.environ['SCT_SCYLLA_LINUX_DISTRO'] = 'ubuntu-xenial'
os.environ['SCT_SCYLLA_LINUX_DISTRO_LOADER'] = 'centos'
os.environ['SCT_SCYLLA_VERSION'] = 'latest'
conf = SCTConfiguration()
conf.verify_configuration()
self.assertIn('scylla_repo', conf.dump_config())
self.assertFalse(conf.get('scylla_repo'))
self.assertEqual(conf.get('scylla_repo_loader'),
'https://s3.amazonaws.com/downloads.scylladb.com/rpm/centos/scylla-nightly.repo')
def conf(config_file, backend):
if backend:
os.environ['SCT_CLUSTER_BACKEND'] = backend
os.environ['SCT_CONFIG_FILES'] = config_file
config = SCTConfiguration()
try:
config.verify_configuration()
except Exception as ex: # pylint: disable=broad-except
click.secho(str(ex), fg='red')
sys.exit(1)
else:
click.secho(config.dump_config(), fg='green')
sys.exit(0)
def test_12_scylla_version_repo_ubuntu_loader_centos(self): # pylint: disable=invalid-name
os.environ['SCT_CLUSTER_BACKEND'] = 'gce'
os.environ['SCT_SCYLLA_LINUX_DISTRO'] = 'ubuntu-xenial'
os.environ['SCT_SCYLLA_LINUX_DISTRO_LOADER'] = 'centos'
os.environ['SCT_SCYLLA_VERSION'] = '3.0.3'
conf = SCTConfiguration()
conf.verify_configuration()
self.assertIn('scylla_repo', conf.dump_config())
self.assertEqual(conf.get('scylla_repo'),
"https://s3.amazonaws.com/downloads.scylladb.com/deb/ubuntu/scylla-3.0-xenial.list")
self.assertEqual(conf.get('scylla_repo_loader'),
"https://s3.amazonaws.com/downloads.scylladb.com/rpm/centos/scylla-3.0.repo")
def conf(config_file, backend):
add_file_logger()
if backend:
os.environ['SCT_CLUSTER_BACKEND'] = backend
if config_file:
os.environ['SCT_CONFIG_FILES'] = config_file
config = SCTConfiguration()
try:
config.verify_configuration()
config.check_required_files()
except Exception as ex: # pylint: disable=broad-except
logging.exception(str(ex))
click.secho(str(ex), fg='red')
sys.exit(1)
else:
click.secho(config.dump_config(), fg='green')
sys.exit(0)
def test_11_openstack(self):
os.environ['SCT_CLUSTER_BACKEND'] = 'openstack'
conf = SCTConfiguration()
conf.verify_configuration()
self.assertIn('openstack_auth_url', conf.dump_config())
self.assertEqual(conf.get('openstack_auth_url'), 'http://1.2.3.4:5000')
