#!/usr/bin/python
# -*- coding: utf-8 -*-

# This file is part of Cockpit.
#
# Copyright (C) 2013 Red Hat, Inc.
#
# Cockpit is free software; you can redistribute it and/or modify it
# under the terms of the GNU Lesser General Public License as published by
# the Free Software Foundation; either version 2.1 of the License, or
# (at your option) any later version.
#
# Cockpit is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
# Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with Cockpit; If not, see <http://www.gnu.org/licenses/>.

import parent
from testlib import *

@skipImage("Atomic always uses loopback", "fedora-atomic", "rhel-atomic", "continuous-atomic")
class TestLoopback(MachineCase):
    def testBasic(self):
        b = self.browser
        m = self.machine

        if "debian" in m.image or "ubuntu" in m.image:
            cockpit_ws = "/usr/lib/cockpit/cockpit-ws"
        else:
            cockpit_ws = "/usr/libexec/cockpit-ws"

        # Start Cockpit with option to access SSH
        m.spawn(cockpit_ws + " --local-ssh --no-tls", "cockpit-ws");
        m.wait_for_cockpit_running()

        b.login_and_go("/system", user="admin")
        b.logout()
        b.wait_visible("#login")

        m.execute("rm /usr/bin/cockpit-bridge")
        b.set_val('#login-user-input', "admin")
        b.set_val('#login-password-input', "foobar")
        b.click('#login-button')
        b.wait_visible('#login-fatal')
        self.assertIn("no-cockpit", b.text('#login-fatal'));

        m.disconnect()
        m.execute("sed -i 's/.*PasswordAuthentication.*/PasswordAuthentication no/' /etc/ssh/sshd_config", direct=True)
        m.execute("( ! systemctl is-active sshd.socket || systemctl stop sshd.socket) && systemctl restart sshd.service", direct=True)
        m.wait_execute()

        b.reload()
        b.wait_visible("#login")
        b.set_val('#login-user-input', "admin")
        b.set_val('#login-password-input', "foobar")
        b.click('#login-button')
        b.wait_visible('#login-fatal')
        self.assertIn("The server refused to authenticate 'admin' using password authentication, and no other supported authentication methods are available.", b.text('#login-fatal'));

        self.allow_journal_messages("Cannot run program cockpit-bridge: No such file or directory",
                                    ".*server offered unsupported authentication methods: public-key.*")

if __name__ == '__main__':
    test_main()
