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()