Skip to content

Commit 3bec86b

Browse files
committed
Check that no further data was received in Client.is_omp()
1 parent 5faa8ec commit 3bec86b

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

samp_query/__init__.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -174,7 +174,8 @@ async def is_omp(self) -> bool:
174174
with trio.move_on_after(2 * ping):
175175
await self.send(b'o', payload)
176176
assert self.prefix
177-
await self.receive(header=self.prefix + b'o' + payload)
177+
data = await self.receive(header=self.prefix + b'o' + payload)
178+
assert not data # No data beyond expected header
178179
return True
179180

180181
return False

0 commit comments

Comments
 (0)