-
Notifications
You must be signed in to change notification settings - Fork 27
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- In fact, simply check if we get an SSL error: if so, we're trying to connect a Remote Service using HTTPS (which is OK) - Refactored HTTP tests to inherit tests
- Loading branch information
Showing
2 changed files
with
209 additions
and
14 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,190 @@ | ||
| #!/usr/bin/python | ||
| # -- Content-Encoding: UTF-8 -- | ||
| """ | ||
| Tests remote services transports based on HTTPS | ||
| :author: Thomas Calmant | ||
| :copyright: Copyright 2016, Thomas Calmant | ||
| :license: Apache License 2.0 | ||
| .. | ||
| Copyright 2016 Thomas Calmant | ||
| 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. | ||
| """ | ||
|
|
||
| # Standard library | ||
| import os | ||
| import shutil | ||
| import ssl | ||
| import threading | ||
| import tempfile | ||
|
|
||
| try: | ||
| import unittest2 as unittest | ||
| except ImportError: | ||
| import unittest | ||
|
|
||
| try: | ||
| # Try to import modules | ||
| from multiprocessing import Process, Queue | ||
| # IronPython fails when creating a queue | ||
| Queue() | ||
| except ImportError: | ||
| # Some interpreters don't have support for multiprocessing | ||
| raise unittest.SkipTest("Interpreter doesn't support multiprocessing") | ||
|
|
||
| try: | ||
| import queue | ||
| except ImportError: | ||
| import Queue as queue | ||
|
|
||
| # Pelix | ||
| from pelix.framework import create_framework | ||
| from pelix.ipopo.constants import use_ipopo | ||
| import pelix.http | ||
| import pelix.remote | ||
|
|
||
|
|
||
| # Local utilities | ||
| from tests.http.test_basic_ssl import install_ipopo, make_certs, \ | ||
| instantiate_server | ||
| from tests.remote.test_transports_http import RemoteService, SVC_SPEC, \ | ||
| HttpTransportsTest | ||
|
|
||
| # ------------------------------------------------------------------------------ | ||
|
|
||
| # Documentation strings format | ||
| __docformat__ = "restructuredtext en" | ||
|
|
||
| # ------------------------------------------------------------------------------ | ||
|
|
||
|
|
||
| def load_framework(transport, components, start_server=True): | ||
| """ | ||
| Starts a Pelix framework in the local process | ||
| :param transport: Name of the transport bundle to install | ||
| :param components: Tuples (factory, name) of instances to start | ||
| """ | ||
| all_bundles = ['pelix.ipopo.core', | ||
| 'pelix.http.basic', | ||
| 'pelix.remote.dispatcher', | ||
| 'pelix.remote.registry', | ||
| 'pelix.remote.discovery.multicast', | ||
| transport] | ||
|
|
||
| # Start the framework | ||
| framework = create_framework(all_bundles) | ||
| framework.start() | ||
|
|
||
| with use_ipopo(framework.get_bundle_context()) as ipopo: | ||
| if start_server: | ||
| # Start a HTTP service on a random port | ||
| ipopo.instantiate(pelix.http.FACTORY_HTTP_BASIC, | ||
| "http-server", | ||
| {pelix.http.HTTP_SERVICE_ADDRESS: "0.0.0.0", | ||
| pelix.http.HTTP_SERVICE_PORT: 0}) | ||
|
|
||
| ipopo.instantiate(pelix.remote.FACTORY_REGISTRY_SERVLET, | ||
| "dispatcher-servlet") | ||
|
|
||
| # Start the multicast discovery | ||
| ipopo.instantiate(pelix.remote.FACTORY_DISCOVERY_MULTICAST, | ||
| "multicast-discovery") | ||
|
|
||
| # Start other components | ||
| for factory, name in components: | ||
| ipopo.instantiate(factory, name) | ||
|
|
||
| return framework | ||
|
|
||
|
|
||
| def export_framework(state_queue, transport, components): | ||
| """ | ||
| Starts a Pelix framework, on the export side | ||
| :param state_queue: Queue to store status | ||
| :param transport: Name of the transport bundle to install | ||
| :param components: Tuples (factory, name) of instances to start | ||
| """ | ||
| tmp_dir = tempfile.mkdtemp(prefix="ipopo-tests-https") | ||
|
|
||
| try: | ||
| # Load the framework | ||
| framework = load_framework(transport, components, False) | ||
| context = framework.get_bundle_context() | ||
| ipopo = install_ipopo(framework) | ||
|
|
||
| # Prepare certificates | ||
| make_certs(tmp_dir, None) | ||
|
|
||
| # Setup the HTTPS server | ||
| instantiate_server( | ||
| ipopo, cert_file=os.path.join(tmp_dir, "server.crt"), | ||
| key_file=os.path.join(tmp_dir, "server.key"), | ||
| address="0.0.0.0", port=0) | ||
|
|
||
| # Register the exported service | ||
| event = threading.Event() | ||
| context.register_service( | ||
| SVC_SPEC, RemoteService(state_queue, event), | ||
| {pelix.remote.PROP_EXPORTED_INTERFACES: '*'}) | ||
|
|
||
| # Send the ready state | ||
| state_queue.put("ready") | ||
|
|
||
| # Loop until the end message | ||
| event.wait() | ||
|
|
||
| # Stopping | ||
| state_queue.put("stopping") | ||
| framework.stop() | ||
|
|
||
| except Exception as ex: | ||
| state_queue.put("Error: {0}".format(ex)) | ||
| finally: | ||
| shutil.rmtree(tmp_dir) | ||
|
|
||
| # ------------------------------------------------------------------------------ | ||
|
|
||
|
|
||
| class HttpsTransportsTest(HttpTransportsTest): | ||
| """ | ||
| Tests Pelix built-in Remote Services transports | ||
| """ | ||
| def __init__(self, *args, **kwargs): | ||
| super(HttpsTransportsTest, self).__init__(*args, **kwargs) | ||
| self._load_framework = load_framework | ||
| self._export_framework = export_framework | ||
|
|
||
| def _run_test(self, transport_bundle, exporter_factory, importer_factory, | ||
| test_kwargs=True): | ||
| try: | ||
| super(HttpsTransportsTest, self)._run_test( | ||
| transport_bundle, exporter_factory, importer_factory, | ||
| test_kwargs) | ||
| except ssl.SSLError: | ||
| # This should happen as the communication happens on a self-signed | ||
| # certificate | ||
| return True | ||
|
|
||
| # ------------------------------------------------------------------------------ | ||
|
|
||
| if __name__ == "__main__": | ||
| # Set logging level | ||
| import logging | ||
| logging.basicConfig(level=logging.DEBUG) | ||
|
|
||
| unittest.main() |