One Hat Cyber Team

Your IP: 216.73.216.144
Server IP: 157.245.143.252
Server: Linux www 6.11.0-9-generic #9-Ubuntu SMP PREEMPT_DYNAMIC Mon Oct 14 13:19:59 UTC 2024 x86_64
Server Software: nginx/1.26.0
PHP Version: 8.3.11
Buat File | Buat Folder

Dir : ~/lib/python3/dist-packages/twisted/conch/test/

View File Name : test_mixin.py

# -*- twisted.conch.test.test_mixin -*-
# Copyright (c) Twisted Matrix Laboratories.
# See LICENSE for details.
from __future__ import annotations

from twisted.conch import mixin
from twisted.internet.testing import StringTransport
from twisted.trial import unittest


class TestBufferingProto(mixin.BufferingMixin):
    scheduled = False
    rescheduled = 0
    transport: StringTransport

    def schedule(self) -> object:
        self.scheduled = True
        return object()

    def reschedule(self, token: object) -> None:
        self.rescheduled += 1


class BufferingTests(unittest.TestCase):
    def testBuffering(self) -> None:
        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))