def test_get_empty(self): node = Mock(address='test') url = 'test-url' # test reader = Reader(node, url=url) reader.receiver = Mock() reader.receiver.receive.side_effect = Timeout message = reader.get(10) # validation reader.receiver.receive.assert_called_once_with(10) self.assertEqual(message, None)
def test_get(self): node = Mock(address='test') received = Mock(body='<body/>') url = 'test-url' # test reader = Reader(node, url=url) reader.receiver = Mock() reader.receiver.receive.return_value = received message = reader.get(10) # validation reader.receiver.receive.assert_called_once_with(10) self.assertTrue(isinstance(message, Message)) self.assertEqual(message._reader, reader) self.assertEqual(message._impl, received) self.assertEqual(message._body, received.body)