tursom/memgraph
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update e2e tests

Browse Source
This commit is contained in:
János Benjamin Antal 2022-01-21 16:19:17 +01:00
parent 5c576b95af
commit 635fe8b42f
1 changed files with 14 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
tests/e2e/streams/kafka_streams_tests.py
View File

@ -421,10 +421,13 @@ def test_set_offset(kafka_producer, kafka_topics, connection, transformation):
assert comparison_check("Final Message", res[0]) assert comparison_check("Final Message", res[0])
common.execute_and_fetch_all(cursor, "MATCH (n) DETACH DELETE n") common.execute_and_fetch_all(cursor, "MATCH (n) DETACH DELETE n")
def test_info_procedure(kafka_topics, connection): def test_info_procedure(kafka_topics, connection):
cursor = connection.cursor() cursor = connection.cursor()
stream_name = 'test_stream' stream_name = 'test_stream'
configs = {"sasl.username": "michael.scott"}
local = "localhost:9092" local = "localhost:9092"
credentials = {"sasl.password": "S3cr3tP4ssw0rd"}
consumer_group = "ConsumerGr" consumer_group = "ConsumerGr"
common.execute_and_fetch_all( common.execute_and_fetch_all(
cursor, cursor,
@ -432,13 +435,21 @@ def test_info_procedure(kafka_topics, connection):
f"TOPICS {','.join(kafka_topics)} " f"TOPICS {','.join(kafka_topics)} "
f"TRANSFORM pulsar_transform.simple " f"TRANSFORM pulsar_transform.simple "
f"CONSUMER_GROUP {consumer_group} " f"CONSUMER_GROUP {consumer_group} "
f"BOOTSTRAP_SERVERS '{local}'" f"BOOTSTRAP_SERVERS '{local}' "
f"CONFIGS {configs} "
f"CREDENTIALS {credentials}"
) )
stream_info = common.execute_and_fetch_all(cursor, f"CALL mg.kafka_stream_info('{stream_name}') YIELD *") stream_info = common.execute_and_fetch_all(
cursor, f"CALL mg.kafka_stream_info('{stream_name}') YIELD *")
expected_stream_info = [(local, consumer_group, kafka_topics)] reducted_credentials = {key: "<REDUCTED>" for (
key, value) in credentials.items()}
expected_stream_info = [
(local, configs, consumer_group, reducted_credentials, kafka_topics)]
common.validate_info(stream_info, expected_stream_info) common.validate_info(stream_info, expected_stream_info)
if __name__ == "__main__": if __name__ == "__main__":
sys.exit(pytest.main([__file__, "-rA"])) sys.exit(pytest.main([__file__, "-rA"]))