#!/usr/bin/python3

# checkpoint:

# 1. osbuild-composer.socket can be started and enabled

# 2. no blueprint by default and can add and delete blueprint

import string

import random

import composerlib

import testlib

@testlib.nondestructive

class TestService(composerlib.ComposerCase):

    def setUp(self):

        # remove all blueprints

        # stop and disable osbuild-composer.socket service

        super().setUp()

        self.machine.execute("""

            for bp in $(composer-cli blueprints list); do

                composer-cli blueprints delete $bp

            done

            systemctl disable --now osbuild-composer.socket

        """)

    def testBasic(self):

        b = self.browser

        self.login_and_go("/composer", authorized=True)

        b.wait_present("#main")

        # start and enable osbuild-composer from UI

        b.click("button:contains('Start')")

        # work around .pf-c-empty-state element disappear in one second

        b.wait_in_text("#main", "No Blueprints")

        b.wait_text(".pf-c-empty-state h1", "No Blueprints")

        # create blueprint with random name and description

        name_template = string.ascii_letters + string.digits + "-" + "_"

        bp_name = "".join(random.sample(name_template, 20))

        desc_template = string.ascii_letters + string.digits + " "

        bp_desc = "".join(random.sample(desc_template, 40))

        b.click(".pf-c-empty-state button:contains('Create Blueprint')")

        b.wait_present("div[role=dialog] #cmpsr-modal-crt-blueprint")

        # blueprint name

        b.set_input_text("#textInput-modal-markup", bp_name)

        # blueprint description

        b.set_input_text("#textInput2-modal-markup", bp_desc)

        # create

        b.click("#create-blueprint-modal-create-button")

        b.wait_not_present("div[role=dialog] #cmpsr-modal-crt-blueprint")

        # wait for all available compontents visible

        with b.wait_timeout(300):

            b.wait_present("ul[data-list=inputs]")

        # search for openssh-server pacakge

        b.set_input_text("#cmpsr-blueprint-input-filter", "openssh-server")

        b.key_press("\r")

        with b.wait_timeout(120):

            b.wait_text("#openssh-server-input", "openssh-server")

        # add it to blueprint

        b.click("li[data-input=openssh-server] .fa-plus")

        # wait for package added

        b.wait_present("#blueprint-tabs")

        b.wait_present("li[data-component=openssh-server]")

        # click commit button

        b.click(".cmpsr-header__actions .btn-primary")

        # click commit button

        b.click("#cmpsr-modal-pending-changes .modal-footer .btn-primary")

        b.wait_not_present("#cmpsr-modal-pending-changes")

        # wait for toast notification dialog

        b.wait_present("#cmpsr-toast-committed .pficon-ok")

        # click close button

        b.click("#cmpsr-toast-committed .pficon-close")

        b.wait_not_present("#cmpsr-toast-committed .pficon-ok")

        # click back to blueprint link

        b.click("a:contains('Back to Blueprints')")

        # new added blueprint should be there with correct name and description

        b.wait_visible("li[data-blueprint={}]".format(bp_name))

        b.wait_text("#{}-name".format(bp_name), bp_name)

        b.wait_text("div[data-description]", bp_desc)

        # delete blueprint, cancel first

        b.click("#{}-kebab".format(bp_name))

        b.wait_attr("#{}-kebab".format(bp_name), "aria-expanded", "true")

        b.click("ul[aria-labelledby={}-kebab] a:contains('Delete')".format(bp_name))

        b.wait_attr("#{}-kebab".format(bp_name), "aria-expanded", "false")

        b.click("#cmpsr-modal-delete button:contains('Cancel')")

        b.wait_not_present("#cmpsr-modal-delete")

        # delete here

        b.click("#{}-kebab".format(bp_name))

        b.wait_attr("#{}-kebab".format(bp_name), "aria-expanded", "true")

        b.click("ul[aria-labelledby={}-kebab] a:contains('Delete')".format(bp_name))

        b.wait_attr("#{}-kebab".format(bp_name), "aria-expanded", "false")

        b.click("#cmpsr-modal-delete button:contains('Delete')")

        b.wait_not_present("#cmpsr-modal-delete")

        b.wait_not_present("li[data-blueprint={}]".format(bp_name))

        # no blueprint here

        b.wait_in_text("#main", "No Blueprints")

        # collect code coverage result

        self.check_coverage()

    def testNoEnable(self):

        b = self.browser

        m = self.machine

        # stop and disable osbuild-composer.socket

        m.execute("systemctl disable --now osbuild-composer.socket")

        self.login_and_go("/composer", authorized=True)

        b.wait_present("#main")

        # do not enable osbuild-composer

        b.click(".checkbox input")

        # start and enable osbuild-composer from UI

        b.click("button:contains('Start')")

        # work around .pf-c-empty-state element disappear in one second

        b.wait_in_text("#main", "No Blueprints")

        b.wait_text(".pf-c-empty-state h1", "No Blueprints")

        # osbuild-composer should be disabled

        is_enabled = m.execute("systemctl is-enabled osbuild-composer.socket || true").rstrip()

        self.assertEqual(is_enabled, "disabled")

        # osbuild-composer should be started

        is_active = m.execute("systemctl is-active osbuild-composer.socket").rstrip()

        self.assertEqual(is_active, "active")

        # collect code coverage result

        self.check_coverage()

    def testNonAdminUser(self):

        b = self.browser

        m = self.machine

        # stop and disable osbuild-composer.socket

        m.execute("systemctl disable --now osbuild-composer.socket")

        self.login_and_go("/composer", authorized=False)

        b.wait_present("#main")

        # Start button can be disabled button or clickable button

        b.wait_present("button:contains('Start')")

        # disable this test due to https://github.com/osbuild/cockpit-composer/issues/941

        # if "disabled" not in b.attr("button:contains('Start')", "class"):

        #     b.click("button:contains('Start')")

        #     # error message

        #     b.wait_in_text(".cmpsr-alert-blank-slate", "Not permitted to perform this action.")

        #     # Start button still there

        #     b.wait_visible("button:contains('Start')")

        # collect code coverage result

        self.check_coverage()

if __name__ == '__main__':

    testlib.test_main()