def test_handle_event():
httpretty.register_uri(httpretty.POST,
"http://example.com",
body='{"origin": "127.0.0.1"}')
with RedisRunnerContext():
subscriber = Subscriber(stream=uuid4().hex,
subscriber_name=uuid4().hex,
topics=["system"],
evt_types=["SYS"],
activities=["init", "verb"],
protocol=Protocol.HTTP,
endpoint="http://example.com")
event = Event(
event_id="test-event-",
topic="system",
evt_type="SYS",
activity="init",
ref_id="00",
)
ok = subscriber.handle_event(event, "consumer_group", "consumer_id")
assert ok
called = httpretty.last_request()
body = json.loads(called.body)
assert body["stream"] == subscriber.stream
assert body["event"]["event_id"] == event.event_id
assert body["event"]["topic"] == event.topic
assert body["event"]["evt_type"] == event.evt_type
assert body["event"]["activity"] == event.activity
assert body["event"]["ref_id"] == event.ref_id
assert body["consumer_group"] == "consumer_group"
assert body["consumer_id"] == "consumer_id"
assert body["subscriber_name"] == subscriber.subscriber_name
def test_create_and_list_streams():
with RedisRunnerContext():
subs = SubscriberManager.list_subscribers()
existing_len = len(subs)
subscriber = Subscriber(stream=uuid4().hex,
subscriber_name=uuid4().hex,
topics=["system"],
evt_types=["SYS"],
activities=["init", "verb"],
protocol=Protocol.HTTP,
endpoint="http://example.com")
ok = SubscriberManager.put_subscriber(subscriber)
assert ok is True
subs = SubscriberManager.list_subscribers()
assert existing_len + 1 == len(subs)
this_one = None
for sub in subs:
if sub.subscriber_name == subscriber.subscriber_name:
this_one = sub
break
assert this_one is not None
print(this_one.to_dict())
print(subscriber.to_dict())
assert this_one.stream == subscriber.stream
assert this_one.subscriber_name == subscriber.subscriber_name
assert set(this_one.topics) == set(subscriber.topics)
assert set(this_one.evt_types) == set(subscriber.evt_types)
assert set(this_one.activities) == set(subscriber.activities)
assert this_one.protocol == subscriber.protocol
assert this_one.endpoint == subscriber.endpoint
assert this_one == subscriber
def put_stream_subscriber(stream):
data = request.get_json(force=True)
subscriber = Subscriber.from_dict(data)
created = SubscriberManager.put_subscriber(subscriber)
return make_response(
json_dumps({
"status": "ok",
"stream": stream,
"subscriber": subscriber.subscriber_name,
"created": created
}), 200)
def test_handle_event_mismatch_activities():
httpretty.register_uri(httpretty.POST, "http://example.com")
with RedisRunnerContext():
subscriber = Subscriber(stream=uuid4().hex,
subscriber_name=uuid4().hex,
topics=["system"],
evt_types=["SYS"],
activities=["init", "verb"],
protocol=Protocol.HTTP,
endpoint="http://example.com")
event = Event(
event_id="test-event-",
topic="system",
evt_type="SYS",
activity="write",
ref_id="00",
)
ok = subscriber.matches(event)
assert ok is False
assert httpretty.has_request() is False
def test_workers_unite():
httpretty.register_uri(
httpretty.POST,
"http://example.com",
body='{"origin": "127.0.0.1"}',
status=200,
)
with RedisRunnerContext():
# -- create a subscriber that is mocked with httpretty
subscriber = Subscriber(stream=uuid4().hex,
subscriber_name=uuid4().hex,
topics=["system"],
evt_types=["SYS"],
activities=["init", "verb"],
protocol=Protocol.HTTP,
endpoint="http://example.com")
ok = SubscriberManager.put_subscriber(subscriber)
assert ok
# -- prepare an event to send
event = Event(
topic="system",
evt_type="SYS",
activity="init",
ref_id="00",
)
ok = ProtonPack.send_event(subscriber.stream, event)
assert ok
# -- start up a worker and have it consume the stream
gb = GhostBuster(subscriber.stream, "consumer_group", "consumer_id")
gb.start(False)
# -- assert the handler uri is called
called = httpretty.last_request()
print(dir(called))
assert called.body is not None
assert called.body != ""
body = json.loads(called.body)
assert body["stream"] == subscriber.stream
assert body["event"]["topic"] == event.topic
assert body["event"]["evt_type"] == event.evt_type
assert body["event"]["activity"] == event.activity
assert body["event"]["ref_id"] == event.ref_id
assert body["consumer_group"] == "consumer_group"
assert body["consumer_id"] == "consumer_id"
assert body["subscriber_name"] == subscriber.subscriber_name
def test_del_subscriber():
with RedisRunnerContext():
subscriber = Subscriber(stream=uuid4().hex,
subscriber_name=uuid4().hex,
topics=["system"],
evt_types=["SYS"],
activities=["init", "verb"],
protocol=Protocol.HTTP,
endpoint="http://example.com")
ok = SubscriberManager.put_subscriber(subscriber)
assert ok is True
subs = SubscriberManager.list_subscribers()
assert len(subs) == 1
for sub in subs:
SubscriberManager.del_subscriber(sub)
subs = SubscriberManager.list_subscribers()
assert len(subs) == 0