File: test_extensions.py

package info (click to toggle)
python-wsproto 1.2.0-1
  • links: PTS, VCS
  • area: main
  • in suites: bookworm, forky, sid, trixie
  • size: 436 kB
  • sloc: python: 4,195; makefile: 17; sh: 6
file content (44 lines) | stat: -rw-r--r-- 1,422 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
from wsproto import extensions as wpext, frame_protocol as fp


class TestExtension:
    def test_enabled(self) -> None:
        ext = wpext.Extension()
        assert not ext.enabled()

    def test_offer(self) -> None:
        ext = wpext.Extension()
        assert ext.offer() is None

    def test_accept(self) -> None:
        ext = wpext.Extension()
        offer = "myext"
        assert ext.accept(offer) is None

    def test_finalize(self) -> None:
        ext = wpext.Extension()
        offer = "myext"
        ext.finalize(offer)

    def test_frame_inbound_header(self) -> None:
        ext = wpext.Extension()
        result = ext.frame_inbound_header(None, None, None, None)  # type: ignore[arg-type]
        assert result == fp.RsvBits(False, False, False)

    def test_frame_inbound_payload_data(self) -> None:
        ext = wpext.Extension()
        data = b""
        assert ext.frame_inbound_payload_data(None, data) == data  # type: ignore[arg-type]

    def test_frame_inbound_complete(self) -> None:
        ext = wpext.Extension()
        assert ext.frame_inbound_complete(None, None) is None  # type: ignore[arg-type]

    def test_frame_outbound(self) -> None:
        ext = wpext.Extension()
        rsv = fp.RsvBits(True, True, True)
        data = b""
        assert ext.frame_outbound(None, None, rsv, data, None) == (  # type: ignore[arg-type]
            rsv,
            data,
        )