# -*- coding: utf-8 -*-
# Copyright 2018 New Vector Ltd
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.


from tests import unittest

from synapse.config._base import ConfigError
from synapse.rulecheck.domain_rule_checker import DomainRuleChecker


class DomainRuleCheckerTestCase(unittest.TestCase):

    def test_allowed(self):
        config = {
          "default": False,
          "domain_mapping": {
              "source_one": [ "target_one", "target_two" ],
              "source_two": [ "target_two" ]
          }
        }
        check = DomainRuleChecker(config)
        self.assertTrue(check.user_may_invite("test:source_one","test:target_one", "room"))
        self.assertTrue(check.user_may_invite("test:source_one","test:target_two", "room"))
        self.assertTrue(check.user_may_invite("test:source_two","test:target_two", "room"))


    def test_disallowed(self):
        config = {
            "default": True,
            "domain_mapping": {
                "source_one": [ "target_one", "target_two" ],
                "source_two": [ "target_two" ]
            }
        }
        check = DomainRuleChecker(config)
        self.assertFalse(check.user_may_invite("test:source_one","test:target_three", "room"))
        self.assertFalse(check.user_may_invite("test:source_two","test:target_three", "room"))
        self.assertFalse(check.user_may_invite("test:source_two","test:target_one", "room"))


    def test_default_allow(self):
        config = {
          "default": True,
          "domain_mapping": {
              "source_one": [ "target_one", "target_two" ],
              "source_two": [ "target_two" ]
          }
        }
        check = DomainRuleChecker(config)
        self.assertTrue(check.user_may_invite("test:source_three","test:target_one", "room"))

    def test_default_deny(self):
        config = {
          "default": False,
          "domain_mapping": {
             "source_one": [ "target_one", "target_two" ],
             "source_two": [ "target_two" ]
          }
        }
        check = DomainRuleChecker(config)
        self.assertFalse(check.user_may_invite("test:source_three","test:target_one", "room"))


    def test_config_parse(self):
        config = {
          "default": False,
          "domain_mapping": {
              "source_one": [ "target_one", "target_two" ],
              "source_two": [ "target_two" ]
          }
        }
        self.assertEquals(config, DomainRuleChecker.parse_config(config))


    def test_config_parse_failure(self):
        config = {
          "domain_mapping": {
              "source_one": [ "target_one", "target_two" ],
              "source_two": [ "target_two" ]
          }
        }
        self.assertRaises(ConfigError, DomainRuleChecker.parse_config, config)