mirror of
https://github.com/neogeek23/drawshare.git
synced 2026-02-04 11:08:21 +00:00
44 lines
1.0 KiB
Python
44 lines
1.0 KiB
Python
|
# -*- twisted.conch.test.test_mixin -*-
|
||
|
|
# Copyright (c) Twisted Matrix Laboratories.
|
||
|
|
# See LICENSE for details.
|
||
|
|
|
||
|
|
from twisted.trial import unittest
|
||
|
|
from twisted.test.proto_helpers import StringTransport
|
||
|
|
|
||
|
|
from twisted.conch import mixin
|
||
|
|
|
||
|
|
|
||
|
|
class TestBufferingProto(mixin.BufferingMixin):
|
||
|
|
scheduled = False
|
||
|
|
rescheduled = 0
|
||
|
|
def schedule(self):
|
||
|
|
self.scheduled = True
|
||
|
|
return object()
|
||
|
|
|
||
|
|
def reschedule(self, token):
|
||
|
|
self.rescheduled += 1
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
class BufferingTests(unittest.TestCase):
|
||
|
|
def testBuffering(self):
|
||
|
|
p = TestBufferingProto()
|
||
|
|
t = p.transport = StringTransport()
|
||
|
|
|
||
|
|
self.assertFalse(p.scheduled)
|
||
|
|
|
||
|
|
L = [b'foo', b'bar', b'baz', b'quux']
|
||
|
|
|
||
|
|
p.write(b'foo')
|
||
|
|
self.assertTrue(p.scheduled)
|
||
|
|
self.assertFalse(p.rescheduled)
|
||
|
|
|
||
|
|
for s in L:
|
||
|
|
n = p.rescheduled
|
||
|
|
p.write(s)
|
||
|
|
self.assertEqual(p.rescheduled, n + 1)
|
||
|
|
self.assertEqual(t.value(), b'')
|
||
|
|
|
||
|
|
p.flush()
|
||
|
|
self.assertEqual(t.value(), b'foo' + b''.join(L))
|