13c9bf76af
* Add base of e2e tests * Add python dependencies * Explicitly close customer in destructor * Parametrize tests and add test for CHECK STREAM * Add tests for SHOW STREAMS * Add test for concurrent start/stop during check * Add test for calling check with an already started stream * Run streams e2e tests on CI servers Co-authored-by: antonio2368 <antonio2368@users.noreply.github.com> Co-authored-by: Jure Bajic <jbajic@users.noreply.github.com>
38 lines
1.2 KiB
Python
38 lines
1.2 KiB
Python
import mgp
|
|
|
|
|
|
@mgp.transformation
|
|
def simple(context: mgp.TransCtx,
|
|
messages: mgp.Messages
|
|
) -> mgp.Record(query=str, parameters=mgp.Map):
|
|
|
|
result_queries = []
|
|
|
|
for i in range(0, messages.total_messages()):
|
|
message = messages.message_at(i)
|
|
payload_as_str = message.payload().decode("utf-8")
|
|
result_queries.append(mgp.Record(
|
|
query=f"CREATE (n:MESSAGE {{timestamp: '{message.timestamp()}', payload: '{payload_as_str}', topic: '{message.topic_name()}'}})",
|
|
parameters=None))
|
|
|
|
return result_queries
|
|
|
|
|
|
@mgp.transformation
|
|
def with_parameters(context: mgp.TransCtx,
|
|
messages: mgp.Messages
|
|
) -> mgp.Record(query=str, parameters=mgp.Map):
|
|
|
|
result_queries = []
|
|
|
|
for i in range(0, messages.total_messages()):
|
|
message = messages.message_at(i)
|
|
payload_as_str = message.payload().decode("utf-8")
|
|
result_queries.append(mgp.Record(
|
|
query="CREATE (n:MESSAGE {timestamp: $timestamp, payload: $payload, topic: $topic})",
|
|
parameters={"timestamp": message.timestamp(),
|
|
"payload": payload_as_str,
|
|
"topic": message.topic_name()}))
|
|
|
|
return result_queries
|