cic-stack/apps/cic-eth/tests/unit/api/test_callback.py

166 lines
3.7 KiB
Python
Raw Normal View History

2021-02-01 18:12:51 +01:00
# standard imports
import socket
import celery
import threading
import logging
import time
import json
# third-party imports
import pytest
import redis as redis_interface
# local imports
from cic_eth.callbacks import http
from cic_eth.callbacks import tcp
from cic_eth.callbacks import redis
celery_app = celery.current_app
logg = celery_app.log.get_default_logger()
class Response:
status = 200
def test_callback_http(
celery_session_worker,
mocker,
):
mocker.patch('cic_eth.callbacks.http.urlopen', return_value=Response())
s = celery.signature(
'cic_eth.callbacks.http.http',
[
'foo',
'http://localhost:65000',
1,
],
)
t = s.apply_async()
t.get()
def test_callback_tcp(
celery_session_worker,
):
timeout=2
data = {
'foo': 'bar',
'xyzzy': 42,
}
class Accept(threading.Thread):
def __init__(self, socket):
super(Accept, self).__init__()
self.socket = socket
self.exception = None
def run(self):
(c, sockaddr) = self.socket.accept()
echo = c.recv(1024)
c.close()
logg.debug('recived {} '.format(data))
o = json.loads(echo)
try:
2021-03-01 21:15:17 +01:00
assert o['result'] == data
2021-02-01 18:12:51 +01:00
except Exception as e:
self.exception = e
def join(self):
threading.Thread.join(self)
if self.exception != None:
raise self.exception
s = socket.socket(family=socket.AF_INET, type=socket.SOCK_STREAM)
s.bind(('localhost', 0))
s.settimeout(timeout)
s.listen(1)
sockaddr = s.getsockname()
a = Accept(s)
a.start()
s_cb = celery.signature(
'cic_eth.callbacks.tcp.tcp',
[
data,
'{}:{}'.format(sockaddr[0], sockaddr[1]),
'1',
],
queue=None,
)
s_cb.apply_async()
a.join()
s.close()
def test_callback_redis(
load_config,
celery_session_worker,
):
timeout=2
channel = 'barbarbar'
host = load_config.get('REDIS_HOST', 'localhost')
port = load_config.get('REDIS_PORT', '6379')
db = load_config.get('REDIS_DB', '0')
data = {
'foo': 'bar',
'xyzzy': 42,
}
class Accept(threading.Thread):
def __init__(self, pubsub):
super(Accept, self).__init__()
self.pubsub = pubsub
self.exception = None
def run(self):
self.pubsub.get_message() # subscribe message
echo = self.pubsub.get_message(timeout=timeout)
o = json.loads(echo['data'])
logg.debug('recived {} '.format(o))
try:
2021-03-01 21:15:17 +01:00
assert o['result'] == data
2021-02-01 18:12:51 +01:00
except Exception as e:
self.exception = e
def join(self):
threading.Thread.join(self)
if self.exception != None:
raise self.exception
ps = None
try:
r = redis_interface.Redis(host=host, port=int(port), db=int(db))
ps = r.pubsub(
)
ps.subscribe(channel)
except redis_interface.exceptions.ConnectionError as e:
pytest.skip('cannot connect to redis, skipping test: {}'.format(e))
a = Accept(ps)
a.start()
s_cb = celery.signature(
'cic_eth.callbacks.redis.redis',
[
data,
'{}:{}:{}:{}'.format(host, port, db, channel),
'1',
],
queue=None,
)
s_cb.apply_async()
a.join()
r.close()