Skip to content

Commit 0362994

Browse files
Lingling PengLingling Peng
authored andcommitted
update to use self.registration id
1 parent a037c3b commit 0362994

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

tests/integration/synapseclient/models/async/test_agent_async.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,10 @@ class TestAgentSession:
6060
@pytest.fixture(autouse=True, scope="function")
6161
def init(self, syn: Synapse) -> None:
6262
self.syn = syn
63+
if syn.repoEndpoint == "https://repo-dev.dev.sagebase.org/repo/v1":
64+
self.AGENT_REGISTRATION_ID = "7"
65+
else:
66+
self.AGENT_REGISTRATION_ID = "29"
6367

6468
async def test_start(self) -> None:
6569
# GIVEN an agent session with a valid agent registration id
@@ -147,6 +151,10 @@ def get_test_agent(self) -> Agent:
147151
@pytest.fixture(autouse=True, scope="function")
148152
def init(self, syn: Synapse) -> None:
149153
self.syn = syn
154+
if syn.repoEndpoint == "https://repo-dev.dev.sagebase.org/repo/v1":
155+
self.AGENT_REGISTRATION_ID = "7"
156+
else:
157+
self.AGENT_REGISTRATION_ID = "29"
150158

151159
async def test_register(self) -> None:
152160
# GIVEN an Agent with a valid agent AWS id

0 commit comments

Comments
 (0)