Permalink
Cannot retrieve contributors at this time
#!/usr/bin/env python | |
#-*-Mode:python;coding:utf-8;tab-width:4;c-basic-offset:4;indent-tabs-mode:()-*- | |
# ex: set ft=python fenc=utf-8 sts=4 ts=4 sw=4 et nomod: | |
# | |
# MIT License | |
# | |
# Copyright (c) 2013-2020 Michael Truog <mjtruog at protonmail dot com> | |
# | |
# Permission is hereby granted, free of charge, to any person obtaining a | |
# copy of this software and associated documentation files (the "Software"), | |
# to deal in the Software without restriction, including without limitation | |
# the rights to use, copy, modify, merge, publish, distribute, sublicense, | |
# and/or sell copies of the Software, and to permit persons to whom the | |
# Software is furnished to do so, subject to the following conditions: | |
# | |
# The above copyright notice and this permission notice shall be included in | |
# all copies or substantial portions of the Software. | |
# | |
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR | |
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, | |
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE | |
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER | |
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING | |
# FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER | |
# DEALINGS IN THE SOFTWARE. | |
# | |
""" | |
Websockets Integration Test with Python/C | |
""" | |
from __future__ import print_function | |
import sys | |
import threading | |
import time | |
import traceback | |
from cloudi_c import API, TerminateException | |
class _Task(threading.Thread): | |
def __init__(self, api): | |
threading.Thread.__init__(self) | |
self.__api = api | |
def run(self): | |
try: | |
self.__api.subscribe('bounce/get', self.__request) | |
self.__api.subscribe('bounce/delay', self.__delay) | |
self.__api.subscribe('bounce/websocket/connect', | |
self.__connect) | |
self.__api.subscribe('bounce/websocket/disconnect', | |
self.__disconnect) | |
result = self.__api.poll() | |
assert result is False | |
except TerminateException: | |
pass | |
except Exception: | |
traceback.print_exc(file=sys.stderr) | |
print('terminate websockets python_c') | |
def __connect(self, request_type, name, pattern, request_info, request, | |
timeout, priority, trans_id, pid): | |
# pylint: disable=unused-argument | |
# pylint: disable=too-many-arguments | |
assert request == b'CONNECT' | |
print('connect: %s' % | |
str(API.info_key_value_parse(request_info))) | |
return 'got connect! yay!' | |
def __disconnect(self, request_type, name, pattern, request_info, request, | |
timeout, priority, trans_id, pid): | |
# pylint: disable=unused-argument | |
# pylint: disable=too-many-arguments | |
assert request == b'DISCONNECT' | |
print('disconnect: %s' % | |
str(API.info_key_value_parse(request_info))) | |
return '' | |
def __request(self, request_type, name, pattern, request_info, request, | |
timeout, priority, trans_id, pid): | |
# pylint: disable=unused-argument | |
# pylint: disable=too-many-arguments | |
# send the request to self | |
self.__api.send_async(self.__api.prefix() + 'bounce/delay', | |
request) | |
return request | |
def __delay(self, request_type, name, pattern, request_info, request, | |
timeout, priority, trans_id, pid): | |
# pylint: disable=unused-argument | |
# pylint: disable=too-many-arguments | |
time.sleep(1.0) | |
assert name[-6:] == '/delay' | |
trans_ids = self.__api.mcast_async(name[:-6] + '/websocket', | |
b'notification: got "' + | |
request + b'" 1 second ago') | |
if trans_ids == []: | |
print('websockets: (no websockets connected?)') | |
else: | |
for check in trans_ids: | |
(_, response, _) = self.__api.recv_async(trans_id=check) | |
print('websockets: %s' % str(response)) | |
def _main(): | |
thread_count = API.thread_count() | |
assert thread_count >= 1 | |
threads = [_Task(API(thread_index)) | |
for thread_index in range(thread_count)] | |
for thread in threads: | |
thread.start() | |
for thread in threads: | |
thread.join() | |
if __name__ == '__main__': | |
_main() |