Method assertFrameSent

tests/test_protocol.py:31–55

Outgoing data for ``connection`` contains the given frame. ``frame`` may be ``None`` if no frame is expected. When ``eof`` is ``True``, the end of the stream is also expected.

(self, connection, frame, eof=False)

class ProtocolTestCase(FramesTestCase):
    def assertFrameSent(self, connection, frame, eof=False):
        """
        Outgoing data for ``connection`` contains the given frame.

        ``frame`` may be ``None`` if no frame is expected.

        When ``eof`` is ``True``, the end of the stream is also expected.

        """
        frames_sent = [
            (
                None
                if write is SEND_EOF
                else self.parse(
                    write,
                    mask=connection.side is CLIENT,
                    extensions=connection.extensions,
                )
            )
            for write in connection.data_to_send()
        ]
        frames_expected = [] if frame is None else [frame]
        if eof:
            frames_expected += [None]
        self.assertEqual(frames_sent, frames_expected)

    def assertFrameReceived(self, connection, frame):
        """