aboutsummaryrefslogtreecommitdiffstats
path: root/packages/meshbay-cli/tests/test_store_and_session.py
diff options
context:
space:
mode:
Diffstat (limited to 'packages/meshbay-cli/tests/test_store_and_session.py')
-rw-r--r--packages/meshbay-cli/tests/test_store_and_session.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/packages/meshbay-cli/tests/test_store_and_session.py b/packages/meshbay-cli/tests/test_store_and_session.py
index 42458e2..07ba77e 100644
--- a/packages/meshbay-cli/tests/test_store_and_session.py
+++ b/packages/meshbay-cli/tests/test_store_and_session.py
@@ -244,7 +244,7 @@ async def test_a_machine_waiting_for_approval_retries_until_it_is_approved(monke
monkeypatch.setattr(groups, "LINK_POLL", 0)
tries, shown = [], []
- async def connect(account, group_id, node):
+ async def connect(account, group_id, node, join_code=None):
tries.append(node["node_id"])
if len(tries) < 3:
raise DeviceLinkPending("ABCD-EFGH", time.time() + 600)
@@ -264,7 +264,7 @@ async def test_without_a_terminal_the_code_is_shown_and_the_run_ends(monkeypatch
from meshbay_cli.client import groups
from meshbay_cli.client.session import DeviceLinkPending
- async def connect(account, group_id, node):
+ async def connect(account, group_id, node, join_code=None):
raise DeviceLinkPending("ABCD-EFGH", time.time() + 600)
monkeypatch.setattr(groups, "connect", connect)