from .dummy import DummyAction
from .dummy import DummyCondition
from plone.app.contentrules.browser.rule import RuleEditFormView
from plone.app.contentrules.rule import Rule
from plone.app.contentrules.tests.base import ContentRulesTestCase
from plone.contentrules.engine.interfaces import IRuleStorage
from zope.component import getMultiAdapter
from zope.component import getUtility
from zope.lifecycleevent.interfaces import IObjectModifiedEvent


class DummyModifiedRule(Rule):

    title = "My test rule"
    description = "Test my rule"
    event = IObjectModifiedEvent
    enabled = True


class TestRuleManagementViews(ContentRulesTestCase):
    def testRuleAdding(self):
        adding = getMultiAdapter((self.portal, self.portal.REQUEST), name="+rule")
        storage = getUtility(IRuleStorage)
        self.assertEqual(0, len(storage))
        r = Rule()
        adding.add(r)
        self.assertEqual(1, len(storage))
        self.assertTrue(list(storage.values())[0] is r)

    def testRuleAddView(self):
        adding = getMultiAdapter((self.portal, self.portal.REQUEST), name="+rule")
        addview = getMultiAdapter(
            (adding, self.portal.REQUEST), name="plone.ContentRule"
        )
        storage = getUtility(IRuleStorage)
        self.assertEqual(0, len(storage))
        addview.form_instance.update()
        content = addview.form_instance.create(
            {"title": "foo", "description": "bar", "event": None}
        )  # noqa
        addview.form_instance.add(content)
        self.assertEqual(1, len(storage))
        self.assertEqual("foo", list(storage.values())[0].title)

    def testRuleEditView(self):
        r = Rule()
        editview = getMultiAdapter((r, self.portal.REQUEST), name="edit")
        self.assertTrue(isinstance(editview, RuleEditFormView))


class TestRuleElementManagementViews(ContentRulesTestCase):
    def afterSetUp(self):
        self.setRoles(("Manager",))

    def testRuleStopModification(self):
        storage = getUtility(IRuleStorage)
        storage["foo"] = Rule()

        rule = self.portal.restrictedTraverse("++rule++foo")
        view = rule.restrictedTraverse("manage-elements")
        view.template = lambda: "No template thanks"

        self.portal.REQUEST.form["stopExecuting"] = "on"
        self.portal.REQUEST.form["form.button.Save"] = True
        self.addAuthToRequest()

        self.assertEqual(False, rule.stop)
        view()
        self.assertEqual(True, rule.stop)

    def testRuleConditionAdding(self):
        storage = getUtility(IRuleStorage)
        storage["foo"] = Rule()
        rule = self.portal.restrictedTraverse("++rule++foo")
        adding = getMultiAdapter((rule, self.portal.REQUEST), name="+condition")
        d = DummyCondition()
        self.assertEqual(0, len(rule.conditions))
        adding.add(d)
        self.assertEqual(1, len(rule.conditions))
        self.assertTrue(rule.conditions[0] is d)

    def testRuleActionAdding(self):
        storage = getUtility(IRuleStorage)
        storage["foo"] = Rule()
        rule = self.portal.restrictedTraverse("++rule++foo")
        adding = getMultiAdapter((rule, self.portal.REQUEST), name="+action")
        d = DummyAction()
        self.assertEqual(0, len(rule.actions))
        adding.add(d)
        self.assertEqual(1, len(rule.actions))
        self.assertTrue(rule.actions[0] is d)

    def testRulesControlPanel(self):
        portal = self.portal
        storage = getUtility(IRuleStorage)
        storage["foo"] = DummyModifiedRule()
        controlpanel = portal.restrictedTraverse("@@rules-controlpanel")
        registered_rules = controlpanel.registeredRules()
        self.assertEqual(1, len(registered_rules))
        registered_rule = registered_rules[0]
        self.assertEqual(
            registered_rule["row_class"],
            "trigger-iobjectmodifiedevent state-enabled assignment-unassigned",
        )
        self.assertEqual(registered_rule["trigger"], "Object modified")
        self.assertTrue(registered_rule["enabled"])
        self.assertFalse(registered_rule["assigned"])

        rule_types = controlpanel.ruleTypesToShow()
        rule_types_ids = [r["id"] for r in rule_types]
        self.assertTrue("trigger-iobjectmodifiedevent" in rule_types_ids)
        self.assertFalse("trigger-iobjectaddedevent" in rule_types_ids)

        rule_states = controlpanel.statesToShow()
        rule_states_ids = [r["id"] for r in rule_states]
        self.assertTrue("state-enabled" in rule_states_ids)
        self.assertTrue("state-disabled" in rule_states_ids)

        # enable rule
        portal.REQUEST["rule-id"] = "foo"
        self.addAuthToRequest()
        portal.restrictedTraverse("@@contentrule-disable").disable_rule()
        registered_rules = controlpanel.registeredRules()
        self.assertFalse(registered_rules[0]["enabled"])

        portal.restrictedTraverse("@@contentrule-enable").enable_rule()
        registered_rules = controlpanel.registeredRules()
        self.assertTrue(registered_rules[0]["enabled"])

        # works without ajax
        portal.REQUEST.form["rule-id"] = "foo"
        portal.REQUEST.form["form.button.DisableRule"] = "1"
        portal.restrictedTraverse("@@rules-controlpanel")()
        registered_rules = controlpanel.registeredRules()
        self.assertFalse(registered_rules[0]["enabled"])

        portal.restrictedTraverse("@@contentrule-delete").delete_rule()
        registered_rules = controlpanel.registeredRules()
        self.assertEqual(0, len(registered_rules))

    def testChangeGloballyEnable(self):
        storage = getUtility(IRuleStorage)
        portal = self.portal
        self.addAuthToRequest()

        portal.restrictedTraverse("@@contentrule-globally-enable").globally_enable()
        self.assertTrue(storage.active)

        portal.restrictedTraverse("@@contentrule-globally-disable").globally_disable()
        self.assertFalse(storage.active)

        portal.restrictedTraverse("@@contentrule-globally-enable").globally_enable()
        self.assertTrue(storage.active)

        # without ajax
        self.portal.REQUEST.form