本文整理匯總了Python中dulwich.protocol.ReceivableProtocol.recv方法的典型用法代碼示例。如果您正苦於以下問題：Python ReceivableProtocol.recv方法的具體用法？Python ReceivableProtocol.recv怎麽用？Python ReceivableProtocol.recv使用的例子？那麽, 這裏精選的方法代碼示例或許可以為您提供幫助。您也可以進一步了解該方法所在類dulwich.protocol.ReceivableProtocol的用法示例。
在下文中一共展示了ReceivableProtocol.recv方法的2個代碼示例，這些例子默認根據受歡迎程度排序。您可以為喜歡或者感覺有用的代碼點讚，您的評價將有助於係統推薦出更棒的Python代碼示例。
示例1: ReceivableProtocolTests
# 需要導入模塊: from dulwich.protocol import ReceivableProtocol [as 別名]
# 或者: from dulwich.protocol.ReceivableProtocol import recv [as 別名]
class ReceivableProtocolTests(BaseProtocolTests, TestCase):
def setUp(self):
TestCase.setUp(self)
self.rout = BytesIO()
self.rin = ReceivableBytesIO()
def _closeit():
self.rout.close()
self.rin.close()
self.proto = ReceivableProtocol(self.rin.recv, self.rout.write, _closeit)
self.proto._rbufsize = 8
self.addCleanup(self.proto.close)
def test_eof(self):
# Allow blocking reads past EOF just for this test. The only parts of
# the protocol that might check for EOF do not depend on the recv()
# semantics anyway.
self.rin.allow_read_past_eof = True
BaseProtocolTests.test_eof(self)
def test_recv(self):
all_data = b'1234567' * 10 # not a multiple of bufsize
self.rin.write(all_data)
self.rin.seek(0)
data = b''
# We ask for 8 bytes each time and actually read 7, so it should take
# exactly 10 iterations.
for _ in range(10):
data += self.proto.recv(10)
# any more reads would block
self.assertRaises(AssertionError, self.proto.recv, 10)
self.assertEqual(all_data, data)
def test_recv_read(self):
all_data = b'1234567' # recv exactly in one call
self.rin.write(all_data)
self.rin.seek(0)
self.assertEqual(b'1234', self.proto.recv(4))
self.assertEqual(b'567', self.proto.read(3))
self.assertRaises(AssertionError, self.proto.recv, 10)
def test_read_recv(self):
all_data = b'12345678abcdefg'
self.rin.write(all_data)
self.rin.seek(0)
self.assertEqual(b'1234', self.proto.read(4))
self.assertEqual(b'5678abc', self.proto.recv(8))
self.assertEqual(b'defg', self.proto.read(4))
self.assertRaises(AssertionError, self.proto.recv, 10)
def test_mixed(self):
# arbitrary non-repeating string
all_data = ','.join(str(i) for i in range(100)).encode('utf-8')
self.rin.write(all_data)
self.rin.seek(0)
data = b''
for i in range(1, 100):
data += self.proto.recv(i)
# if we get to the end, do a non-blocking read instead of blocking
if len(data) + i > len(all_data):
data += self.proto.recv(i)
# ReceivableBytesIO leaves off the last byte unless we ask
# nicely
data += self.proto.recv(1)
break
else:
data += self.proto.read(i)
else:
# didn't break, something must have gone wrong
self.fail()
self.assertEqual(all_data, data)示例2: ReceivableProtocolTests
# 需要導入模塊: from dulwich.protocol import ReceivableProtocol [as 別名]
# 或者: from dulwich.protocol.ReceivableProtocol import recv [as 別名]
class ReceivableProtocolTests(BaseProtocolTests, TestCase):
def setUp(self):
TestCase.setUp(self)
self.rout = StringIO()
self.rin = ReceivableStringIO()
self.proto = ReceivableProtocol(self.rin.recv, self.rout.write)
self.proto._rbufsize = 8
def test_recv(self):
all_data = "1234567" * 10 # not a multiple of bufsize
self.rin.write(all_data)
self.rin.seek(0)
data = ""
# We ask for 8 bytes each time and actually read 7, so it should take
# exactly 10 iterations.
for _ in xrange(10):
data += self.proto.recv(10)
# any more reads would block
self.assertRaises(AssertionError, self.proto.recv, 10)
self.assertEquals(all_data, data)
def test_recv_read(self):
all_data = "1234567" # recv exactly in one call
self.rin.write(all_data)
self.rin.seek(0)
self.assertEquals("1234", self.proto.recv(4))
self.assertEquals("567", self.proto.read(3))
self.assertRaises(AssertionError, self.proto.recv, 10)
def test_read_recv(self):
all_data = "12345678abcdefg"
self.rin.write(all_data)
self.rin.seek(0)
self.assertEquals("1234", self.proto.read(4))
self.assertEquals("5678abc", self.proto.recv(8))
self.assertEquals("defg", self.proto.read(4))
self.assertRaises(AssertionError, self.proto.recv, 10)
def test_mixed(self):
# arbitrary non-repeating string
all_data = ",".join(str(i) for i in xrange(100))
self.rin.write(all_data)
self.rin.seek(0)
data = ""
for i in xrange(1, 100):
data += self.proto.recv(i)
# if we get to the end, do a non-blocking read instead of blocking
if len(data) + i > len(all_data):
data += self.proto.recv(i)
# ReceivableStringIO leaves off the last byte unless we ask
# nicely
data += self.proto.recv(1)
break
else:
data += self.proto.read(i)
else:
# didn't break, something must have gone wrong
self.fail()
self.assertEquals(all_data, data)