TestMu 2025 - The Largest Virtual Software Testing Conference is Back | 50,000+ Attendees

How to use setup_ssh_key method in autotest

Best Python code snippet using autotest_python

ssh_copy_id.py

...77    try:78        yield cnn79    finally:80        cnn.close()81def setup_ssh_key(cnn):82    '''83    ssh-keygen -t ed2551984    '''85    key0 = ("<>")86    rc = ssh_run(cnn, "mkdir -p ~/.ssh")87    print(view_resp(rc))88    rc = ssh_run(cnn, '''printf "\n{}" >> ~/.ssh/authorized_keys'''.format(key0))89    print(view_resp(rc))90    # remove duplicate91    with cnn.prefix("cd ~/.ssh"):92        ssh_run(cnn, "rm -f bak")93        ssh_run(cnn, "cat authorized_keys | sort | uniq > bak")94        ssh_run(cnn, "mv -f bak authorized_keys")95def main():96    onlines = []97    offlines = []98    try:99        hosts = [100        ]101        for i, host in enumerate(hosts):102            print(f"-[{i}]--------------- {host}")103            sys.stdout.flush()104            for _ in range(2):105                try:106                    with connect(host) as cnn:107                        setup_ssh_key(cnn)108                        sys.stdout.flush()109                        onlines.append(host)110                        break111                except SSHException as ssherr:112                    print(f"{type(ssherr)} {ssherr}")113                    continue114                except sotimeout as tmoerr:115                    print(f"{type(tmoerr)} {tmoerr}")116                    continue117                except soerror as soerr:118                    print(f"{type(soerr)} {soerr}")119                    continue120            else:121                offlines.append(host)...

auth.py

Source: auth.py

...17    for host in c["host_list"]:18        priv_key_path = get_key_path(host)19        if not os.path.exists(priv_key_path):20            print(f"No smurf ssh key found for host '{host}' : setting up new key...")21            setup_ssh_key(host)22        else:23            print(f"Smurf ssh key found for host '{host}' : '{priv_key_path}'")24    25def get_key_dir():26    """ Return the key directory and make sure it exists.27    Returns28    -------29    str30        Path to the key directory.31    """32    c = config.Config()33    key_dir = c.data["key_dir"]34    if not os.path.isdir(key_dir):35        os.makedirs(key_dir)36    return key_dir37def get_key_path(host):38    """ Return the path the private key.39    Parameters40    ----------41    host : str42        Hostname of the remote host.43    Returns44    -------45    str46        Path to private key file.47    """48    key_dir = get_key_dir()49    priv_key_path = os.path.join(key_dir, "id_rsa_smurf_" + host)50    return priv_key_path51def generate_ssh_key(host):52    """ Generate a ssh key to allow smurf to connect to host.53    Keys are stored in the ~/.smurf directory under the keys directory.54    Parameters55    ----------56    host : str57        Hostname of the remote host to setup keys for.58    """59    priv_key_path = get_key_path(host)60    comment = "'limited smurf access from {}@{}'".format(61        os.environ["USER"], os.uname()[1])62    subprocess.run(["ssh-keygen", "-C", comment, "-N",63                    "", "-f", priv_key_path], check=True)64def get_pub_key(host):65    """ Get the public key data for a host. 66    Parameters67    ----------68    host : str69        Hostname of the remote host.70    Returns71    -------72    str73        Public key and comment.74    """75    pub_key_path = get_key_path(host) + ".pub"76    with open(pub_key_path, "r") as infile:77        content = infile.read().strip()78    return content79def copy_ssh_key(host):80    """ Copy the smurf key to the remote host and setup command restriction.81    This is done by preparing a temporary file and copying82    its content using ssh-copy-id.83    Parameters84    ----------85    host : str86        Remote host to register key on.87    """88    preprend = 'command="~/.local/bin/smurfnet-shell",permitopen="localhost:*",no-agent-forwarding,no-pty,no-user-rc,no-X11-forwarding'89    content = preprend + " " + get_pub_key(host)90    pub_key_path = get_key_path(host) + ".pub"91    os.rename(pub_key_path, pub_key_path + ".bak")92    with open(pub_key_path, "w") as outfile:93        outfile.write(content)94    subprocess.run(["ssh-copy-id", "-f", "-i", pub_key_path, host], check=True)95    os.rename(pub_key_path + ".bak", pub_key_path)96def setup_ssh_key(host):97    """ Generates a key for smurf and copy it to the remote host.98    Parameters99    ----------100    host : str101        Hostname of the remote host.102    """103    logging.info(f"Setting up new ssh key for host '{host}'")104    generate_ssh_key(host)105    copy_ssh_key(host)106def ensure_key(host):107    """ Create a key if it does not exists and then return the path to it.108    109    Attempt to create the key once.110    Parameters111    ----------112    str: host113        Name of the host.114    Returns115    -------116    str117        Path to the private key.118    """119    key_path = get_key_path(host)120    if not os.path.exists(key_path):121        setup_ssh_key(host)122    return key_path123if __name__ == "__main__":...

fabfile.py

Source: fabfile.py

...33         '&& chmod 600 /root/.ssh/apitrary-staging-deploy')34def add_ssh_config():35    put('./assets/ssh_config', '/tmp/ssh_config')36    sudo('cat /tmp/ssh_config >> /root/.ssh/config')37def setup_ssh_key():38    sudo('mkdir -p /root/.ssh')39    execute(copy_public_ssh_key)40    execute(copy_secret_ssh_key)41    execute(add_ssh_config)42# ROLES-BASED CALLS43#44#45@roles('TEST')46def test_setup():47    execute(setup_ssh_key)48@roles('STAGING')49def staging_setup():50    execute(setup_ssh_key)51@roles('PRODUCTION')...

Blogs

Check out the latest blogs from LambdaTest on this topic:

LIVE With Automation Testing For OTT Streaming Devices ????

People love to watch, read and interact with quality content — especially video content. Whether it is sports, news, TV shows, or videos captured on smartphones, people crave digital content. The emergence of OTT platforms has already shaped the way people consume content. Viewers can now enjoy their favorite shows whenever they want rather than at pre-set times. Thus, the OTT platform’s concept of viewing anything, anytime, anywhere has hit the right chord.

Two-phase Model-based Testing

Most test automation tools just do test execution automation. Without test design involved in the whole test automation process, the test cases remain ad hoc and detect only simple bugs. This solution is just automation without real testing. In addition, test execution automation is very inefficient.

A Complete Guide To Flutter Testing

Mobile devices and mobile applications – both are booming in the world today. The idea of having the power of a computer in your pocket is revolutionary. As per Statista, mobile accounts for more than half of the web traffic worldwide. Mobile devices (excluding tablets) contributed to 54.4 percent of global website traffic in the fourth quarter of 2021, increasing consistently over the past couple of years.

Now Log Bugs Using LambdaTest and DevRev

In today’s world, an organization’s most valuable resource is its customers. However, acquiring new customers in an increasingly competitive marketplace can be challenging while maintaining a strong bond with existing clients. Implementing a customer relationship management (CRM) system will allow your organization to keep track of important customer information. This will enable you to market your services and products to these customers better.

Using ChatGPT for Test Automation

ChatGPT broke all Internet records by going viral in the first week of its launch. A million users in 5 days are unprecedented. A conversational AI that can answer natural language-based questions and create poems, write movie scripts, write social media posts, write descriptive essays, and do tons of amazing things. Our first thought when we got access to the platform was how to use this amazing platform to make the lives of web and mobile app testers easier. And most importantly, how we can use ChatGPT for automated testing.

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.