logos-messaging/logos-messaging-interop-tests
1
0
Fork 0
mirror of https://github.com/logos-messaging/logos-messaging-interop-tests.git synced 2026-01-09 09:23:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: node1 to act as light push receiving node

Browse Source 
- add setup lightpush node to spin off node2 with lightpush support
This commit is contained in:
Roman 2024-04-26 13:30:53 +08:00
parent 7f86deb0c3
commit b6e14e54d6
No known key found for this signature in database
GPG Key ID: B8FE070B54E11B75
2 changed files with 16 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
src/steps/rln.py
View File

@ -37,9 +37,6 @@ class StepsRLN(StepsCommon):
@allure.step
def setup_first_rln_relay_node(self, lightpush=None, **kwargs):
node_index = None
if lightpush == "true":
node_index = 1
self.node1 = WakuNode(DEFAULT_NWAKU, f"node1_{self.test_id}")
self.node1.start(
relay="true",
@ -47,7 +44,6 @@ class StepsRLN(StepsCommon):
rln_creds_source=RLN_CREDENTIALS,
rln_creds_id="1",
rln_relay_membership_index="1",
node_index=node_index,
lightpush=lightpush,
**kwargs,
)
@ -55,6 +51,9 @@ class StepsRLN(StepsCommon):
self.multiaddr_with_id = self.node1.get_multiaddr_with_id()
self.main_nodes.extend([self.node1])
self.add_node_peer(self.node1, self.multiaddr_list)
self.multiaddr_list.extend([self.node1.get_multiaddr_with_id()])
@allure.step
def setup_second_rln_relay_node(self, **kwargs):
self.node2 = WakuNode(DEFAULT_NWAKU, f"node2_{self.test_id}")
@ -66,18 +65,18 @@ class StepsRLN(StepsCommon):
rln_relay_membership_index="1",
**kwargs,
)
if self.node2.is_nwaku():
self.node2.add_peers([self.multiaddr_with_id])
self.add_node_peer(self.node2, [self.multiaddr_with_id])
self.main_nodes.extend([self.node2])
@allure.step
def setup_lightpush_node(self, relay=None, **kwargs):
self.light_push_node1 = WakuNode(DEFAULT_NWAKU, f"lightpush_node2_{self.test_id}")
self.light_push_node1.start(relay=relay, discv5_bootstrap_node=self.enr_uri, node_index=2, lightpushnode=self.multiaddr_list[0], **kwargs)
def setup_second_lightpush_node(self, relay="false", **kwargs):
self.light_push_node2 = WakuNode(DEFAULT_NWAKU, f"lightpush_node2_{self.test_id}")
self.light_push_node2.start(relay=relay, discv5_bootstrap_node=self.enr_uri, lightpush="true", lightpushnode=self.multiaddr_list[0], **kwargs)
if relay == "true":
self.main_nodes.extend([self.light_push_node1])
self.lightpush_nodes.extend([self.light_push_node1])
self.add_node_peer(self.light_push_node1, self.multiaddr_list)
self.main_nodes.extend([self.light_push_node2])
self.lightpush_nodes.extend([self.light_push_node2])
self.add_node_peer(self.light_push_node2, self.multiaddr_list)
@allure.step
def setup_first_relay_node(self, **kwargs):
@ -95,8 +94,7 @@ class StepsRLN(StepsCommon):
discv5_bootstrap_node=self.enr_uri,
**kwargs,
)
if self.node2.is_nwaku():
self.node2.add_peers([self.multiaddr_with_id])
self.add_node_peer(self.node2, [self.multiaddr_with_id])
self.main_nodes.extend([self.node2])
@allure.step

8
tests/relay/test_rln.py
View File

@ -171,10 +171,10 @@ class TestRelayRLN(StepsRLN, StepsRelay):
delay(n1_epoch_sec)
assert not failed_payloads, f"Payloads failed: {failed_payloads}"
@pytest.mark.skip(reason="Cannot start RLN relay with lightpush enabled")
@pytest.mark.skip(reason="Lightpush node is not adding RLN proofs yet - TBT with Gowaku")
def test_valid_payloads_lightpush_at_spam_rate(self):
self.setup_first_rln_relay_node(lightpush="true") # with lightpush enabled
self.setup_lightpush_node()
self.setup_first_rln_relay_node(lightpush="true")
self.setup_second_lightpush_node()
self.subscribe_main_relay_nodes()
previous = math.trunc(time())
for i, payload in enumerate(SAMPLE_INPUTS):
@ -182,7 +182,7 @@ class TestRelayRLN(StepsRLN, StepsRelay):
message = self.create_message(payload=to_base64(payload["value"]))
try:
now = math.trunc(time())
self.publish_light_push_message(message=message, sender=self.light_push_node1)
self.publish_light_push_message(message=message, sender=self.light_push_node2)
if i > 0 and (now - previous) == 0:
raise AssertionError("Publish with RLN enabled at spam rate worked!!!")
else: