Skip to content

Commit e4a1108

Browse files
authored
feat: Register shared client with the test (#10)
1 parent 582703e commit e4a1108

File tree

2 files changed

+12
-2
lines changed

2 files changed

+12
-2
lines changed

src/hive/client.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -113,6 +113,7 @@ class Client:
113113
config: ClientConfig
114114
id: str
115115
ip: str
116+
shared: bool = False
116117

117118
@classmethod
118119
def start(

src/hive/testing.py

Lines changed: 11 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,10 +24,12 @@ def start(cls, url: str, name: str, description: str) -> "HiveTestSuite":
2424
def end(self):
2525
response = requests.delete(self.url)
2626
response.raise_for_status()
27-
27+
2828
def start_client(self, **kwargs) -> Client | None:
2929
kwargs["url"] = f"{self.url}/node"
30-
return Client.start(**kwargs)
30+
client = Client.start(**kwargs)
31+
client.shared = True
32+
return client
3133

3234
def start_test(self, name: str, description: str) -> "HiveTest":
3335
url = f"{self.url}/test"
@@ -71,3 +73,10 @@ def end(self, *, result: HiveTestResult):
7173
def start_client(self, **kwargs) -> Client | None:
7274
kwargs["url"] = f"{self.url}/node"
7375
return Client.start(**kwargs)
76+
77+
def register_shared_client(self, client: Client):
78+
if not client.shared:
79+
return
80+
url = f"{client.url}/{client.id}/test/{self.id}"
81+
response = requests.post(url)
82+
response.raise_for_status()

0 commit comments

Comments
 (0)