#!/usr/bin/python3 -cimport os, sys; os.execv(os.path.dirname(sys.argv[1]) + "/common/pywrap", sys.argv)
# SPDX-License-Identifier: LGPL-2.1-or-later
#
# Copyright (C) 2021 Red Hat, Inc.
import testlib
from machineslib import VirtualMachinesCase


@testlib.skipBeiboot("multi-host config not supported in beiboot scenario")
class TestMultiMachineVNC(VirtualMachinesCase):
    """
    Test for showing the wrong VNC console when adding another host via the add
    host feature the raw stream channel does not set the host option so would
    connect to the console of machine you connect to instead of the newly added
    host.

    https://issues.redhat.com/browse/RHEL-3959
    """

    provision = {  # noqa: RUF012
        "machine1": {"address": "10.111.113.1/20", "memory_mb": 1024},
        "machine2": {"address": "10.111.113.2/20", "memory_mb": 660},
    }

    def setUp(self):
        super().setUp()
        self.machine1 = self.machines['machine1']
        self.machine2 = self.machines['machine2']

        self.setup_ssh_auth()
        self.enable_multihost(self.machine1)
        self.machine1.write("/etc/cockpit/cockpit.conf", "[Session]\nWarnBeforeConnecting=false\n", append=True)

    def testBasic(self):
        b = self.browser
        m = self.machine
        m2 = self.machine2
        m2_host = "10.111.113.2"

        m.execute(f"ssh-keyscan {m2_host} > /etc/ssh/ssh_known_hosts")
        self.startLibvirt(m2)

        # uses too much RAM with lots of violations
        m.execute("systemctl mask setroubleshootd.service")

        # Start a VM with a VNC on machine1
        name = "subVmTest1"
        self.createVm(name, graphics="vnc", ptyconsole=True)

        self.login_and_go("/machines")
        self.waitPageInit()

        self.waitVmRow(name)
        self.goToVmPage(name)
        b.wait_in_text(f"#vm-{name}-system-state", "Running")

        b.add_machine(m2_host, known_host=True, password=None, expect_warning=False)

        b.switch_to_top()
        b.click("#hosts-sel button")

        b.click(f"a[href='/@{m2_host}']")

        b.wait_js_cond(f'window.location.pathname == "/@{m2_host}/system"')
        b.enter_page("/system", host=m2_host)
        b.become_superuser()

        b.go(f"/@{m2_host}/machines")
        b.enter_page("/machines", host=m2_host)

        self.createVm(name, graphics="vnc", ptyconsole=True, machine=m2)
        self.waitVmRow("subVmTest1")
        b.wait_in_text("#vm-subVmTest1-system-state", "Running")
        self.goToVmPage("subVmTest1")

        # Wait till we have a VNC connection
        b.wait_visible(".vm-console-vnc canvas")

        # Both machines should have the equal amount of VNC connections
        m1_open_connections = m.execute("ss -tupn | grep 127.0.0.1:5900 | grep qemu").splitlines()
        m2_open_connections = m2.execute("ss -tupn | grep 127.0.0.1:5900 | grep qemu").splitlines()
        self.assertTrue(len(m1_open_connections) == 1)
        self.assertTrue(len(m2_open_connections) == 1)


if __name__ == '__main__':
    testlib.test_main()
