Test your AI Agents with the all-new Agent to Agent Testing Platform.Learn More

How to use build_cluster_run_command method in localstack

Best Python code snippet using localstack_python

cluster.py

Source:cluster.py

...82    tmp_dir = os.path.join(data_path, "tmp")83    data_dir = os.path.join(data_path, "data")84    backup_dir = os.path.join(data_path, "backup")85    return Directories(install_dir, tmp_dir, modules_dir, data_dir, backup_dir)86def build_cluster_run_command(cluster_bin: str, settings: CommandSettings) -> List[str]:87    """88    Takes the command settings dict and builds the actual command (which can then be executed as a shell command).89    :param cluster_bin: path to the OpenSearch/Elasticsearch binary (including the binary)90    :param settings: dictionary where each item will be set as a command arguments91    :return: list of strings for the command with the settings to be executed as a shell command92    """93    cmd_settings = [f"-E {k}={v}" for k, v, in settings.items()]94    return [cluster_bin] + cmd_settings95class OpensearchCluster(Server):96    """Manages an OpenSearch cluster which is installed an operated by LocalStack."""97    # TODO: legacy default port should be removed here98    def __init__(99        self, port=4571, host="localhost", version: str = None, directories: Directories = None100    ) -> None:101        super().__init__(port, host)102        self._version = version or self.default_version103        self.command_settings = {}104        self.directories = directories or self._resolve_directories()105    @property106    def default_version(self) -> str:107        return constants.OPENSEARCH_DEFAULT_VERSION108    @property109    def version(self) -> str:110        return self._version111    @property112    def install_version(self) -> str:113        _, install_version = versions.get_install_type_and_version(self._version)114        return install_version115    @property116    def bin_name(self) -> str:117        return "opensearch"118    @property119    def os_user(self):120        return constants.OS_USER_OPENSEARCH121    def health(self) -> Optional[str]:122        return get_cluster_health_status(self.url)123    def do_start_thread(self) -> FuncThread:124        self._ensure_installed()125        self._init_directories()126        cmd = self._create_run_command(additional_settings=self.command_settings)127        cmd = " ".join(cmd)128        if is_root() and self.os_user:129            # run the opensearch process as a non-root user (when running in docker)130            cmd = f"su {self.os_user} -c '{cmd}'"131        env_vars = self._create_env_vars()132        LOG.info("starting %s: %s with env %s", self.bin_name, cmd, env_vars)133        t = ShellCommandThread(134            cmd,135            env_vars=env_vars,136            strip_color=True,137            log_listener=self._log_listener,138        )139        t.start()140        return t141    def _resolve_directories(self) -> Directories:142        return resolve_directories(version=self.version, cluster_path=self.version)143    def _ensure_installed(self):144        install.install_opensearch(self.version)145    def _init_directories(self):146        init_directories(self.directories)147    def _base_settings(self, dirs) -> CommandSettings:148        settings = {149            "http.port": self.port,150            "http.publish_port": self.port,151            "transport.port": "0",152            "network.host": self.host,153            "http.compression": "false",154            "path.data": f'"{dirs.data}"',155            "path.repo": f'"{dirs.backup}"',156            "plugins.security.disabled": "true",157        }158        if os.path.exists(os.path.join(dirs.mods, "x-pack-ml")):159            settings["xpack.ml.enabled"] = "false"160        return settings161    def _create_run_command(162        self, additional_settings: Optional[CommandSettings] = None163    ) -> List[str]:164        # delete opensearch data that may be cached locally from a previous test run165        dirs = self.directories166        bin_path = os.path.join(self.directories.install, "bin", self.bin_name)167        settings = self._base_settings(dirs)168        if additional_settings:169            settings.update(additional_settings)170        cmd = build_cluster_run_command(bin_path, settings)171        return cmd172    def _create_env_vars(self) -> Dict:173        return {174            "OPENSEARCH_JAVA_OPTS": os.environ.get("OPENSEARCH_JAVA_OPTS", "-Xms200m -Xmx600m"),175            "OPENSEARCH_TMPDIR": self.directories.tmp,176        }177    def _log_listener(self, line, **_kwargs):178        # logging the port before each line to be able to connect logs to specific instances179        LOG.info("[%s] %s", self.port, line.rstrip())180class CustomEndpoint:181    """182    Encapsulates a custom endpoint (combines CustomEndpoint and CustomEndpointEnabled within the DomainEndpointOptions183    of the cluster, i.e. combines two fields from the AWS OpenSearch service model).184    """...

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.