def register_conversation(self, response_port=None, response_data=None):
from mocket import Mocket, MocketEntry
response_port = response_port or self.LORA_RESPONSE_PORT
response_data = response_data or []
lora_entry = MocketEntry(location=(None, None),
responses=response_data)
lora_entry.request_cls = bytearray
Mocket.reset()
Mocket.register(lora_entry)
def recvfrom(self, buffersize):
if response_data:
buffer = self.recv(buffersize)
return buffer, response_port
else:
return None, response_port
self.monkeypatch.setattr(MocketSocket,
'recvfrom',
recvfrom,
raising=False)
def test_has_requests(self):
Mocket._requests.extend([1, 2, 3])
self.assertTrue(Mocket.has_requests())
Mocket.reset()
self.assertFalse(Mocket.has_requests())
def test_reset(self):
Mocket._requests.extend([1, 2, 3])
self.assertEqual(Mocket._requests, [1, 2, 3])
Mocket.reset()
self.assertEqual(Mocket._requests, [])
def setUp(self):
Mocket.reset()
def mocketize_setup(self):
Entry.register_response('FLUSHDB', OK)
self.rclient.flushdb()
self.assertEqual(len(Mocket._requests), 1)
Mocket.reset()
def mocketize_setup(self):
Entry.register_response('FLUSHDB', OK)
self.rclient.flushdb()
self.assertEqual(len(Mocket._requests), 1)
Mocket.reset()
def test_has_requests(self):
Mocket._requests.extend([1, 2, 3])
self.assertTrue(Mocket.has_requests())
Mocket.reset()
self.assertFalse(Mocket.has_requests())
def test_reset(self):
Mocket._requests.extend([1, 2, 3])
self.assertEqual(Mocket._requests, [1, 2, 3])
Mocket.reset()
self.assertEqual(Mocket._requests, [])
def setUp(self):
Mocket.reset()
