def test_rechunk(self):
f = websocket.Fragmentizer([b"foo"], False)
f.FRAGMENT_SIZE = 4
assert list(f(b"foobar")) == [
wsproto.events.BytesMessage(b"foob", message_finished=False),
wsproto.events.BytesMessage(b"ar", message_finished=True),
]
def test_empty(self):
f = websocket.Fragmentizer([b"foo"], False)
assert list(f(b"")) == [
wsproto.events.BytesMessage(b"", message_finished=True),
]
def test_keep_sizes(self):
f = websocket.Fragmentizer([b"foo", b"bar"], True)
assert list(f(b"foobaz")) == [
wsproto.events.TextMessage("foo", message_finished=False),
wsproto.events.TextMessage("baz", message_finished=True),
]
def test_empty(self):
f = websocket.Fragmentizer([b"foo"], False)
assert list(f(b"")) == []