# 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 oslotest import base as test_base
import stevedore
from testtools import matchers

from oslo_middleware import base


class TestPasteDeploymentEntryPoints(test_base.BaseTestCase):
    def test_entry_points(self):
        factory_classes = {
            'catch_errors': 'CatchErrors',
            'correlation_id': 'CorrelationId',
            'cors': 'CORS',
            'debug': 'Debug',
            'http_proxy_to_wsgi': 'HTTPProxyToWSGI',
            'request_id': 'RequestId',
            'sizelimit': 'RequestBodySizeLimiter',
        }

        em: stevedore.ExtensionManager[base.ConfigurableMiddleware]
        em = stevedore.ExtensionManager('paste.filter_factory')

        # Ensure all the factories are defined by their names
        factory_names = [extension.name for extension in em]
        self.assertThat(factory_names, matchers.ContainsAll(factory_classes))

    def test_healthcheck_entry_points(self):
        healthcheck_plugins = {
            'disable_by_file': 'DisableByFileHealthcheck',
            'disable_by_files_ports': 'DisableByFilesPortsHealthcheck',
            'enable_by_files': 'EnableByFilesHealthcheck',
        }

        em: stevedore.ExtensionManager[base.ConfigurableMiddleware]
        em = stevedore.ExtensionManager('oslo.middleware.healthcheck')

        # Ensure all the healthcheck plugins are defined by their names
        plugin_names = [extension.name for extension in em]
        self.assertThat(
            plugin_names, matchers.ContainsAll(healthcheck_plugins)
        )