Skip to content

Commit 442aaaf

Browse files
committed
Add getblocking() method.
1 parent c4240b7 commit 442aaaf

2 files changed

Lines changed: 16 additions & 0 deletions

File tree

mocket/mocket.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -235,6 +235,9 @@ def getpeername(self):
235235
def setblocking(self, block):
236236
self.settimeout(None) if block else self.settimeout(0.0)
237237

238+
def getblocking(self):
239+
return self.gettimeout() is None
240+
238241
def getsockname(self):
239242
return socket.gethostbyname(self._address[0]), self._address[1]
240243

tests/main/test_socket.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,13 @@
1+
import socket
2+
3+
import pytest
4+
5+
from mocket.mocket import MocketSocket
6+
7+
8+
@pytest.mark.parametrize("blocking", (False, True))
9+
def test_blocking_socket(blocking):
10+
sock = MocketSocket(socket.AF_INET, socket.SOCK_STREAM)
11+
sock.connect(("locahost", 1234))
12+
sock.setblocking(blocking)
13+
assert sock.getblocking() is blocking

0 commit comments

Comments
 (0)