# Copyright 2019 The ChromiumOS Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

"""Tests for the validate module."""

import os

from chromite.api import api_config
from chromite.api import validate
from chromite.api.gen.chromite.api import build_api_test_pb2
from chromite.api.gen.chromiumos import common_pb2
from chromite.lib import cros_build_lib
from chromite.lib import cros_test_lib
from chromite.lib import osutils


# These tests test the validators by defining a local `impl` function that
# has the same parameters as a controller function and the validator being
# tested. The validators don't care that they aren't actually controller
# functions, they just need the function to look like one, so it works
# to pass an arbitrary message; i.e. passing one of the Request messages
# we'd usually expect in a controller is not required. The validator
# just needs to be checking one of the fields on the message being used.
class ExistsTest(cros_test_lib.TempDirTestCase, api_config.ApiConfigMixin):
    """Tests for the exists validator."""

    def test_not_exists(self) -> None:
        """Test the validator fails when given a path that doesn't exist."""
        path = os.path.join(self.tempdir, "DOES_NOT_EXIST")

        @validate.exists("path")
        def impl(_request, _response, _config) -> None:
            self.fail("Incorrectly allowed method to execute.")

        with self.assertRaises(cros_build_lib.DieSystemExit):
            impl(common_pb2.Chroot(path=path), None, self.api_config)

    def test_exists(self) -> None:
        """Test the validator fails when given a path that doesn't exist."""
        path = os.path.join(self.tempdir, "chroot")
        osutils.SafeMakedirs(path)

        @validate.exists("path")
        def impl(_request, _response, _config) -> None:
            pass

        impl(common_pb2.Chroot(path=path), None, self.api_config)

    def test_skip_validation(self) -> None:
        """Test skipping validation case."""

        @validate.exists("path")
        def impl(_request, _response, _config) -> None:
            pass

        # This would otherwise raise an error for an invalid path.
        impl(common_pb2.Chroot(), None, self.no_validate_config)


class EqTest(cros_test_lib.TestCase, api_config.ApiConfigMixin):
    """Tests for the eq validator."""

    def test_eq(self) -> None:
        """Test a valid value."""

        @validate.eq("location", common_pb2.Path.Location.OUTSIDE)
        def impl(_request, _response, _config) -> None:
            pass

        impl(
            common_pb2.Path(
                path="/", location=common_pb2.Path.Location.OUTSIDE
            ),
            None,
            self.api_config,
        )

    def test_not_eq(self) -> None:
        """Test an invalid value."""

        @validate.eq("location", common_pb2.Path.Location.OUTSIDE)
        def impl(_request, _response, _config) -> None:
            pass

        # Should be failing on the invalid value.
        with self.assertRaises(cros_build_lib.DieSystemExit):
            impl(
                common_pb2.Path(
                    path="/", location=common_pb2.Path.Location.INSIDE
                ),
                None,
                self.api_config,
            )

    def test_not_set(self) -> None:
        """Test an unset value."""

        @validate.eq("location", common_pb2.Path.Location.OUTSIDE)
        def impl(_request, _response, _config) -> None:
            pass

        # Should be failing without a value set.
        with self.assertRaises(cros_build_lib.DieSystemExit):
            impl(common_pb2.Path(path="/"), None, self.api_config)

    def test_skip_validation(self) -> None:
        """Test skipping validation case."""

        @validate.eq("location", common_pb2.Path.Location.OUTSIDE)
        def impl(_request, _response, _config) -> None:
            pass

        # This would otherwise raise an error for an invalid path.
        impl(common_pb2.Path(path="/"), None, self.no_validate_config)


class IsInTest(cros_test_lib.TestCase, api_config.ApiConfigMixin):
    """Tests for the is_in validator."""

    def test_in(self) -> None:
        """Test a valid value."""

        @validate.is_in("path", ["/chroot/path", "/other/chroot/path"])
        def impl(_request, _response, _config) -> None:
            pass

        # Make sure all of the values work.
        impl(common_pb2.Chroot(path="/chroot/path"), None, self.api_config)
        impl(
            common_pb2.Chroot(path="/other/chroot/path"), None, self.api_config
        )

    def test_not_in(self) -> None:
        """Test an invalid value."""

        @validate.is_in("path", ["/chroot/path", "/other/chroot/path"])
        def impl(_request, _response, _config) -> None:
            pass

        # Should be failing on the invalid value.
        with self.assertRaises(cros_build_lib.DieSystemExit):
            impl(common_pb2.Chroot(path="/bad/value"), None, self.api_config)

    def test_not_set(self) -> None:
        """Test an unset value."""

        @validate.is_in("path", ["/chroot/path", "/other/chroot/path"])
        def impl(_request, _response, _config) -> None:
            pass

        # Should be failing without a value set.
        with self.assertRaises(cros_build_lib.DieSystemExit):
            impl(common_pb2.Chroot(), None, self.api_config)

    def test_skip_validation(self) -> None:
        """Test skipping validation case."""

        @validate.is_in("path", ["/chroot/path", "/other/chroot/path"])
        def impl(_request, _response, _config) -> None:
            pass

        # This would otherwise raise an error for an invalid path.
        impl(common_pb2.Chroot(), None, self.no_validate_config)


class EachInTest(cros_test_lib.TestCase, api_config.ApiConfigMixin):
    """Tests for the each_in validator."""

    # Easier access to the enum values.
    ENUM_FOO = build_api_test_pb2.TEST_ENUM_FOO
    ENUM_BAR = build_api_test_pb2.TEST_ENUM_BAR
    ENUM_BAZ = build_api_test_pb2.TEST_ENUM_BAZ

    # pylint: disable=docstring-misnamed-args
    def _message_request(self, *messages):
        """Build a request instance, filling out the messages field.

        Args:
            messages: Each messages data (id, name, flag, enum) as lists. Only
                requires as many as are set. e.g. _request([1], [2]) will create
                two messages with only ids set. _request([1, 'name']) will
                create one with id and name set, but not flag or enum.
        """
        request = build_api_test_pb2.TestRequestMessage()
        for message in messages or []:
            msg = request.messages.add()
            try:
                msg.id = message[0]
                msg.name = message[1]
                msg.flag = message[2]
            except IndexError:
                pass

        return request

    def _enums_request(self, *enum_values):
        """Build a request instance, setting the test_enums field."""
        request = build_api_test_pb2.TestRequestMessage()
        for value in enum_values:
            request.test_enums.append(value)

        return request

    def _numbers_request(self, *numbers):
        """Build a request instance, setting the numbers field."""
        request = build_api_test_pb2.TestRequestMessage()
        request.numbers.extend(numbers)

        return request

    def test_message_in(self) -> None:
        """Test valid values."""

        @validate.each_in("messages", "name", ["foo", "bar"])
        def impl(_request, _response, _config) -> None:
            pass

        impl(self._message_request([1, "foo"]), None, self.api_config)
        impl(
            self._message_request([1, "foo"], [2, "bar"]), None, self.api_config
        )

    def test_enum_in(self) -> None:
        """Test valid enum values."""

        @validate.each_in("test_enums", None, [self.ENUM_FOO, self.ENUM_BAR])
        def impl(_request, _response, _config) -> None:
            pass

        impl(self._enums_request(self.ENUM_FOO), None, self.api_config)
        impl(
            self._enums_request(self.ENUM_FOO, self.ENUM_BAR),
            None,
            self.api_config,
        )

    def test_scalar_in(self) -> None:
        """Test valid scalar values."""

        @validate.each_in("numbers", None, [1, 2])
        def impl(_request, _response, _config) -> None:
            pass

        impl(self._numbers_request(1), None, self.api_config)
        impl(self._numbers_request(1, 2), None, self.api_config)

    def test_message_not_in(self) -> None:
        """Test an invalid value."""

        @validate.each_in("messages", "name", ["foo", "bar"])
        def impl(_request, _response, _config) -> None:
            pass

        # Should be failing on the invalid value.
        with self.assertRaises(cros_build_lib.DieSystemExit):
            impl(self._message_request([1, "invalid"]), None, self.api_config)
        with self.assertRaises(cros_build_lib.DieSystemExit):
            impl(
                self._message_request([1, "invalid"], [2, "invalid"]),
                None,
                self.api_config,
            )
        with self.assertRaises(cros_build_lib.DieSystemExit):
            impl(
                self._message_request([1, "foo"], [2, "invalid"]),
                None,
                self.api_config,
            )

    def test_enum_not_in(self) -> None:
        """Test an invalid enum value."""

        @validate.each_in("test_enums", None, [self.ENUM_FOO, self.ENUM_BAR])
        def impl(_request, _response, _config) -> None:
            pass

        # Only invalid values.
        with self.assertRaises(cros_build_lib.DieSystemExit):
            impl(self._enums_request(self.ENUM_BAZ), None, self.api_config)
        # Mixed valid/invalid values.
        with self.assertRaises(cros_build_lib.DieSystemExit):
            impl(
                self._enums_request(self.ENUM_FOO, self.ENUM_BAZ),
                None,
                self.api_config,
            )

    def test_scalar_not_in(self) -> None:
        """Test invalid scalar value."""

        @validate.each_in("numbers", None, [1, 2])
        def impl(_request, _response, _config) -> None:
            pass

        # Only invalid values.
        with self.assertRaises(cros_build_lib.DieSystemExit):
            impl(self._numbers_request(3), None, self.api_config)
        # Mixed valid/invalid values.
        with self.assertRaises(cros_build_lib.DieSystemExit):
            impl(self._numbers_request(1, 2, 3), None, self.api_config)

    def test_not_set(self) -> None:
        """Test an unset value."""

        @validate.each_in("messages", "name", ["foo", "bar"])
        def impl(_request, _response, _config) -> None:
            pass

        # Should be failing without a value set.
        # No entries in the field.
        with self.assertRaises(cros_build_lib.DieSystemExit):
            impl(self._message_request(), None, self.api_config)
        # No value set on lone entry.
        with self.assertRaises(cros_build_lib.DieSystemExit):
            impl(self._message_request([1]), None, self.api_config)
        # No value set on multiple entries.
        with self.assertRaises(cros_build_lib.DieSystemExit):
            impl(self._message_request([1], [2]), None, self.api_config)
        # Some valid and some invalid entries.
        with self.assertRaises(cros_build_lib.DieSystemExit):
            impl(self._message_request([1, "foo"], [2]), None, self.api_config)

    def test_optional(self) -> None:
        """Test optional argument."""

        @validate.each_in("messages", "name", ["foo", "bar"], optional=True)
        @validate.each_in(
            "test_enums", None, [self.ENUM_FOO, self.ENUM_BAR], optional=True
        )
        @validate.each_in("numbers", None, [1, 2], optional=True)
        def impl(_request, _response, _config) -> None:
            pass

        # No entries in the field succeeds.
        impl(self._message_request(), None, self.api_config)

        # Still fails when entries exist but value unset cases.
        # No value set on lone entry.
        with self.assertRaises(cros_build_lib.DieSystemExit):
            impl(self._message_request([1]), None, self.api_config)
        # No value set on multiple entries.
        with self.assertRaises(cros_build_lib.DieSystemExit):
            impl(self._message_request([1], [2]), None, self.api_config)
        # Some valid and some invalid entries.
        with self.assertRaises(cros_build_lib.DieSystemExit):
            impl(self._message_request([1, "foo"], [2]), None, self.api_config)

    def test_skip_validation(self) -> None:
        """Test skipping validation case."""

        @validate.each_in("messages", "name", ["foo", "bar"])
        @validate.each_in("test_enums", None, [self.ENUM_FOO, self.ENUM_BAR])
        @validate.each_in("numbers", None, [1, 2])
        def impl(_request, _response, _config) -> None:
            pass

        # This would otherwise raise an error for multiple invalid fields.
        impl(
            self._message_request([1, "invalid"]), None, self.no_validate_config
        )


class RequireTest(cros_test_lib.TestCase, api_config.ApiConfigMixin):
    """Tests for the require validator."""

    def test_invalid_field(self) -> None:
        """Test validator fails when given an unset value."""

        @validate.require("does.not.exist")
        def impl(_request, _response, _config) -> None:
            self.fail("Incorrectly allowed method to execute.")

        with self.assertRaises(cros_build_lib.DieSystemExit):
            impl(common_pb2.Chroot(), None, self.api_config)

    def test_not_set(self) -> None:
        """Test validator fails when given an unset value."""

        @validate.require("env.use_flags")
        def impl(_request, _response, _config) -> None:
            self.fail("Incorrectly allowed method to execute.")

        with self.assertRaises(cros_build_lib.DieSystemExit):
            impl(common_pb2.Chroot(), None, self.api_config)

    def test_set(self) -> None:
        """Test validator passes when given set values."""

        @validate.require("path", "env.use_flags")
        def impl(_request, _response, _config) -> None:
            pass

        in_proto = common_pb2.Chroot(
            path="/chroot/path", env={"use_flags": [{"flag": "test"}]}
        )
        impl(in_proto, None, self.api_config)

    def test_mixed(self) -> None:
        """Test validator fails when given a set value and an unset value."""

        @validate.require("path", "env.use_flags")
        def impl(_request, _response, _config) -> None:
            pass

        with self.assertRaises(cros_build_lib.DieSystemExit):
            impl(common_pb2.Chroot(path="/chroot/path"), None, self.api_config)

    def test_skip_validation(self) -> None:
        """Test skipping validation case."""

        @validate.require("path", "env.use_flags")
        def impl(_request, _response, _config) -> None:
            pass

        # This would otherwise raise an error for an invalid path.
        impl(common_pb2.Chroot(), None, self.no_validate_config)


class RequireAnyTest(cros_test_lib.TestCase, api_config.ApiConfigMixin):
    """Tests for the require_any validator."""

    def _get_request(
        self, mid: int = None, name: str = None, flag: bool = None
    ):
        """Build a request instance from the given data."""
        request = build_api_test_pb2.MultiFieldMessage()

        if mid:
            request.id = mid
        if name:
            request.name = name
        if flag:
            request.flag = flag

        return request

    def test_invalid_field(self) -> None:
        """Test validator fails when given an invalid field."""

        @validate.require_any("does.not.exist", "also.invalid")
        def impl(_request, _response, _config) -> None:
            self.fail("Incorrectly allowed method to execute.")

        with self.assertRaises(cros_build_lib.DieSystemExit):
            impl(self._get_request(), None, self.api_config)

    def test_not_set(self) -> None:
        """Test validator fails when given unset values."""

        @validate.require_any("id", "name")
        def impl(_request, _response, _config) -> None:
            self.fail("Incorrectly allowed method to execute.")

        with self.assertRaises(cros_build_lib.DieSystemExit):
            impl(self._get_request(flag=True), None, self.api_config)

    def test_set(self) -> None:
        """Test validator passes when given set values."""

        @validate.require_any("id", "name")
        def impl(_request, _response, _config) -> None:
            pass

        impl(self._get_request(1), None, self.api_config)
        impl(self._get_request(name="foo"), None, self.api_config)
        impl(self._get_request(1, name="foo"), None, self.api_config)


class RequireEachTest(cros_test_lib.TestCase, api_config.ApiConfigMixin):
    """Tests for the require_each validator."""

    def _multi_field_message(self, msg_id=None, name=None, flag=None):
        msg = build_api_test_pb2.MultiFieldMessage()
        if msg_id is not None:
            msg.id = int(msg_id)
        if name is not None:
            msg.name = str(name)
        if flag is not None:
            msg.flag = bool(flag)
        return msg

    def _request(self, messages=None, count=0):
        """Build the request."""
        if messages is None:
            messages = [self._multi_field_message() for _ in range(count)]

        request = build_api_test_pb2.TestRequestMessage()
        for message in messages:
            msg = request.messages.add()
            msg.CopyFrom(message)

        return request

    def test_invalid_field(self) -> None:
        """Test validator fails when given an invalid field."""

        @validate.require_each("does.not", ["exist"])
        def impl(_request, _response, _config) -> None:
            self.fail("Incorrectly allowed method to execute.")

        with self.assertRaises(cros_build_lib.DieSystemExit):
            impl(self._request(), None, self.api_config)

    def test_invalid_call_no_subfields(self) -> None:
        """Test validator fails when given no subfields."""

        with self.assertRaises(AssertionError):

            @validate.require_each("does.not", [])
            def _(_request, _response, _config) -> None:
                pass

    def test_invalid_call_invalid_subfields(self) -> None:
        """Test validator fails when given subfields incorrectly."""

        with self.assertRaises(AssertionError):

            @validate.require_each("does.not", "exist")
            def _(_request, _response, _config) -> None:
                pass

    def test_not_set(self) -> None:
        """Test validator fails when given an unset value."""

        @validate.require_each("messages", ["id"])
        def impl(_request, _response, _config) -> None:
            self.fail("Incorrectly allowed method to execute.")

        with self.assertRaises(cros_build_lib.DieSystemExit):
            impl(self._request(count=2), None, self.api_config)

    def test_no_elements_success(self) -> None:
        """Test validator fails when given no messages in the repeated field."""

        @validate.require_each("messages", ["id"])
        def impl(_request, _response, _config) -> None:
            pass

        impl(self._request(), None, self.api_config)

    def test_no_elements_failure(self) -> None:
        """Test validator fails when given no messages in the repeated field."""

        @validate.require_each("messages", ["id"], allow_empty=False)
        def impl(_request, _response, _config) -> None:
            self.fail("Incorrectly allowed method to execute.")

        with self.assertRaises(cros_build_lib.DieSystemExit):
            impl(self._request(), None, self.api_config)

    def test_set(self) -> None:
        """Test validator passes when given set values."""

        @validate.require_each("messages", ["id"])
        def impl(_request, _response, _config) -> None:
            pass

        messages = [self._multi_field_message(msg_id=i) for i in range(1, 5)]
        impl(self._request(messages=messages), None, self.api_config)

    def test_one_set_fails(self) -> None:
        """Test validator passes when given set values."""

        @validate.require_each("messages", ["id", "name"])
        def impl(_request, _response, _config) -> None:
            pass

        messages = [self._multi_field_message(msg_id=i) for i in range(1, 5)]
        with self.assertRaises(cros_build_lib.DieSystemExit):
            impl(self._request(messages=messages), None, self.api_config)

    def test_multi_set(self) -> None:
        """Test validator passes when all values set."""

        @validate.require_each("messages", ["id", "name"])
        def impl(_request, _response, _config) -> None:
            pass

        messages = [
            self._multi_field_message(msg_id=i, name=i) for i in range(1, 5)
        ]
        impl(self._request(messages=messages), None, self.api_config)

    def test_skip_validation(self) -> None:
        """Test skipping validation case."""

        @validate.require_each("messages", ["id"], allow_empty=False)
        def impl(_request, _response, _config) -> None:
            pass

        impl(self._request(), None, self.no_validate_config)


class ValidateOnlyTest(cros_test_lib.TestCase, api_config.ApiConfigMixin):
    """validate_only decorator tests."""

    def test_validate_only(self) -> None:
        """Test validate only."""

        @validate.require("path")
        @validate.validation_complete
        def impl(_request, _response, _config):
            self.fail("Implementation was called.")
            return 1

        # Just using arbitrary messages, we just need the
        # (request, response, config) arguments so it can check the config.
        rc = impl(
            common_pb2.Chroot(path="/chroot/path"),
            common_pb2.Chroot(),
            self.validate_only_config,
        )

        self.assertEqual(0, rc)

    def test_no_validate_only(self) -> None:
        """Test no use of validate only."""

        @validate.validation_complete
        def impl(_request, _response, _config) -> None:
            self.fail("Incorrectly allowed method to execute.")

        # We will get an assertion error unless validate_only prevents the
        # function from being called.
        with self.assertRaises(AssertionError):
            impl(common_pb2.Chroot(), common_pb2.Chroot(), self.api_config)