def get_server_service(self, cluster_num_partitions=None, num_nodes=None): """ Return a Waltz Server service that uses config.ini for configuration. Optional arguments can be pass to override default settings. """ cluster_num_partitions = cluster_num_partitions or int(self.zk_cfg['ClusterNumPartitions']) num_nodes = num_nodes or int(self.server_cfg['NumNodes']) cpu = int(self.server_cfg['NumCpuCores']) mem = self.server_cfg['MemSize'] disk = self.server_cfg['DiskSize'] cluster_spec = ClusterSpec.from_dict({'cpu':cpu, 'mem':mem, 'disk':disk, 'num_nodes':num_nodes}) zk = self.zk_cfg['ZkUrl'] cluster_root = self.zk_cfg['ClusterRoot'] cluster_name = self.zk_cfg['ClusterName'] port = int(self.server_cfg['Port']) jetty_port = int(self.server_cfg['JettyPort']) lib_dir = self.server_cfg['LibDir'] config_file_dir = self.server_cfg['ConfigFileDir'] ssl_configs = {"ssl_keystore_loc": self.server_cfg['SslKeystoreLoc'], "ssl_keystore_pwd": self.server_cfg['SslKeystorePwd'], "ssl_truststore_loc": self.server_cfg['SslTruststoreLoc'], "ssl_truststore_pwd": self.server_cfg['SslTruststorePwd']} return WaltzServerService(self.test_context, cluster_spec, zk, cluster_root, cluster_name, cluster_num_partitions, \ port, jetty_port, lib_dir, config_file_dir, ssl_configs)
def get_verifiable_client(self, num_nodes=None): """ Return a verifiable Waltz client that uses config.ini for configuration. Optional arguments can be pass to override default settings. """ num_nodes = num_nodes or int(self.client_cfg['NumNodes']) cpu = int(self.client_cfg['NumCpuCores']) mem = self.client_cfg['MemSize'] disk = self.client_cfg['DiskSize'] cluster_spec = ClusterSpec.from_dict({ 'cpu': cpu, 'mem': mem, 'disk': disk, 'num_nodes': num_nodes }) zk = self.zk_cfg['ZkUrl'] cluster_root = self.zk_cfg['ClusterRoot'] lib_dir = self.client_cfg['LibDir'] config_file_dir = self.client_cfg['ConfigFileDir'] ssl_configs = { "ssl_keystore_loc": self.client_cfg['SslKeystoreLoc'], "ssl_keystore_pwd": self.client_cfg['SslKeystorePwd'], "ssl_truststore_loc": self.client_cfg['SslTruststoreLoc'], "ssl_truststore_pwd": self.client_cfg['SslTruststorePwd'] } return VerifiableClient(self.test_context, cluster_spec, zk, cluster_root, lib_dir, config_file_dir, ssl_configs)
def check_from_dict(self): empty = ClusterSpec.empty() assert "[]" == str(empty) node_specs_dict = { 'cpu': 2, 'mem': '2GB', 'disk': '30GB', 'num_nodes': 2 } custom_linux_1 = ClusterSpec.from_dict(node_specs_dict) assert '[{"additional_disks(GB)": {}, "cpu": 2, "disk(GB)": 30.0, "mem(GB)": 2.0, "num_nodes": 2, "os": "linux"}]' \ == str(custom_linux_1)
def get_storage_service(self, num_nodes=None): """ Return a Waltz Storage service that uses config.ini for configuration. Optional arguments can be pass to override default settings. """ num_nodes = num_nodes or int(self.storage_cfg['NumNodes']) cpu = int(self.storage_cfg['NumCpuCores']) mem = self.storage_cfg['MemSize'] disk = self.storage_cfg['DiskSize'] additional_disks = json.loads(self.storage_cfg['AdditionalDisks']) cluster_spec = ClusterSpec.from_dict({ 'cpu': cpu, 'mem': mem, 'disk': disk, 'num_nodes': num_nodes, 'additional_disks': additional_disks }) zk = self.zk_cfg['ZkUrl'] cluster_root = self.zk_cfg['ClusterRoot'] port = int(self.storage_cfg['Port']) admin_port = int(self.storage_cfg['AdminPort']) jetty_port = int(self.storage_cfg['JettyPort']) lib_dir = self.storage_cfg['LibDir'] data_dir = self.storage_cfg['DataDir'] config_file_dir = self.storage_cfg['ConfigFileDir'] ssl_configs = { "ssl_keystore_loc": self.storage_cfg['SslKeystoreLoc'], "ssl_keystore_pwd": self.storage_cfg['SslKeystorePwd'], "ssl_truststore_loc": self.storage_cfg['SslTruststoreLoc'], "ssl_truststore_pwd": self.storage_cfg['SslTruststorePwd'] } return WaltzStorageService(self.test_context, cluster_spec, zk, cluster_root, port, admin_port, jetty_port, lib_dir, data_dir, config_file_dir, ssl_configs)
def get_storage_service(self, cluster_key, cluster_num_partitions=None, num_nodes=None): """ Return a Waltz Storage service that uses config.ini for configuration. Optional arguments can be pass to override default settings. """ cluster_num_partitions = cluster_num_partitions or int( self.zk_cfg['ClusterNumPartitions']) num_nodes = num_nodes or int(self.storage_cfg['NumNodes']) node_type = self.storage_cfg['NodeType'] cluster_spec = ClusterSpec.from_dict({node_type: num_nodes}) zk = self.zk_cfg['ZkUrl'] cluster_root = self.zk_cfg['ClusterRoot'] port = int(self.storage_cfg['Port']) admin_port = int(self.storage_cfg['AdminPort']) jetty_port = int(self.storage_cfg['JettyPort']) lib_dir = self.storage_cfg['LibDir'] data_dir = self.storage_cfg['DataDir'] config_file_dir = self.storage_cfg['ConfigFileDir'] return WaltzStorageService(self.test_context, cluster_spec, zk, cluster_root, cluster_num_partitions, \ cluster_key, port, admin_port, jetty_port, lib_dir, data_dir, config_file_dir)