Closing of SSH connection to board over Labhost from the board in interactive session lands you on the Labhost

[cmhe]

I have this setup:

Lab:

import contextlib
import tbot
from tbot.machine import connector, linux


class StartBashInitializer(tbot.machine.Initializer):
    @contextlib.contextmanager
    def _init_machine(self):
        linux.util.wait_for_shell(self.ch)
        self.ch.sendline("bash --norc --noprofile")
        yield None


class Fuji(
    connector.ParamikoConnector,
    StartBashInitializer,
    linux.Bash,
    linux.Lab,
):
    hostname = "fuji.lan"
    username = "ch"

    @property
    def workdir(self):
        return linux.Workdir.static(self,
                                    f"/home/{self.username}/work/tbot-workdir")


def register_machines(ctx: tbot.Context) -> None:
    ctx.register(Fuji, tbot.role.LabHost)

Board:

import tbot
import contextlib
import time

from tbot.machine import connector, linux


class BoardLinux(connector.SSHConnector, linux.Bash):
    name = "board"
    hostname = "user"
    username = "root"

    @classmethod
    @contextlib.contextmanager
    def from_context(cls, ctx: tbot.Context):
        with tbot.ctx() as cx:
            lh = cx.request(tbot.role.LabHost)
            lo = cx.request(tbot.role.LocalHost)

            try:
                tbot.log.message("Power board on")
                lo.exec0("curl", "http://tasmota-0061.lan/cm?cmnd=Power%20On")

                tbot.log.message("Waiting for connection...")
                while not lh.test("timeout", "5", "ssh", BoardLinux.hostname, "true"):
                    tbot.log.message("still waiting ...")

                with cls(lh) as m:
                    yield m
            finally:
                tbot.log.message("Power board off")
                lo.exec0("curl", "http://tasmota-0061.lan/cm?cmnd=Power%20Off")


def register_machines(ctx: tbot.Context) -> None:
    ctx.register(BoardLinux, tbot.role.BoardLinux)

Starting a interactive session with the interactive_linux and closing the connection on the board via for instance a reboot command. Will drop me to the LabHost (fuji).

I use the current master commit: 0be8d7e of tbot

[Rahix]

I think this should be resolvable by this change:

 class StartBashInitializer(tbot.machine.Initializer):
     @contextlib.contextmanager
     def _init_machine(self):
         linux.util.wait_for_shell(self.ch)
-        self.ch.sendline("bash --norc --noprofile")
+        self.ch.sendline("bash --norc --noprofile; exit")
         yield None