api/controller/api_unittest.py

# 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.

"""API controller tests."""

from chromite.api import api_config
from chromite.api import router
from chromite.api.controller import api as api_controller
from chromite.api.gen.chromite.api import api_pb2
from chromite.lib import constants
from chromite.lib import cros_test_lib


class CompileProtoTest(
    cros_test_lib.RunCommandTestCase, api_config.ApiConfigMixin
):
    """CompileProto tests."""

    def setUp(self):
        self.request = api_pb2.CompileProtoRequest()
        self.response = api_pb2.CompileProtoResponse()

    def testCompileProto(self):
        """Quick CompileProto functional check."""
        self.rc.SetDefaultCmdResult(stdout=" M foo/bar.py")
        expected = [str(constants.CHROMITE_DIR / "foo" / "bar.py")]

        api_controller.CompileProto(
            self.request, self.response, self.api_config
        )
        returned = [f.path for f in self.response.modified_files]

        self.assertCountEqual(expected, returned)
        self.assertTrue(all(f in returned for f in expected))
        self.assertTrue(all(f in expected for f in returned))

    def testValidateOnly(self):
        """Verify validate only calls do not execute logic."""
        api_controller.CompileProto(
            self.request, self.response, self.validate_only_config
        )

        self.assertFalse(self.rc.call_count)

    def testMockSuccess(self):
        """Verify mock success calls do not execute logic."""
        api_controller.CompileProto(
            self.request, self.response, self.mock_call_config
        )
        self.assertTrue(len(self.response.modified_files))
        self.assertFalse(self.rc.call_count)


class GetMethodsTest(cros_test_lib.MockTestCase, api_config.ApiConfigMixin):
    """GetMethods tests."""

    def setUp(self):
        self.request = api_pb2.MethodGetRequest()
        self.response = api_pb2.MethodGetResponse()

    def testGetMethods(self):
        """Simple GetMethods check."""
        methods = ["foo", "bar"]
        self.PatchObject(router.Router, "ListMethods", return_value=methods)

        api_controller.GetMethods(self.request, self.response, self.api_config)

        self.assertCountEqual(
            methods, [m.method for m in self.response.methods]
        )

    def testValidateOnly(self):
        """Check validate_only_config calls only validate."""
        patch = self.PatchObject(router.Router, "ListMethods")

        api_controller.GetMethods(
            self.request, self.response, self.validate_only_config
        )

        patch.assert_not_called()


class GetVersionTest(cros_test_lib.MockTestCase, api_config.ApiConfigMixin):
    """GetVersion tests."""

    def setUp(self):
        self.PatchObject(api_controller, "VERSION_MAJOR", new=1)
        self.PatchObject(api_controller, "VERSION_MINOR", new=2)
        self.PatchObject(api_controller, "VERSION_BUG", new=3)

        self.request = api_pb2.VersionGetRequest()
        self.response = api_pb2.VersionGetResponse()

    def testGetVersion(self):
        """Simple GetVersion check."""
        api_controller.GetVersion(self.request, self.response, self.api_config)

        self.assertEqual(self.response.version.major, 1)
        self.assertEqual(self.response.version.minor, 2)
        self.assertEqual(self.response.version.bug, 3)

    def testValidateOnly(self):
        """Check validate_only_config calls only validate."""
        api_controller.GetVersion(
            self.request, self.response, self.validate_only_config
        )

        self.assertFalse(self.response.version.major)
        self.assertFalse(self.response.version.minor)
        self.assertFalse(self.response.version.bug)