Disable transmission tests pending rewrite

This commit is contained in:
Kovid Goyal 2021-08-30 08:08:41 +05:30
parent c159ac287f
commit 161a54ceaa
No known key found for this signature in database
GPG Key ID: 06BC317B515ACE7C
2 changed files with 8 additions and 9 deletions

View File

@ -44,7 +44,7 @@ class FileType(Enum):
link = auto()
class TransmisstionType(Enum):
class TransmissionType(Enum):
simple = auto()
resume = auto()
rsync = auto()
@ -82,7 +82,7 @@ class FileTransmissionCommand:
action: Action = Action.invalid
compression: Compression = Compression.none
ftype: FileType = FileType.regular
ttype: TransmisstionType = TransmisstionType.simple
ttype: TransmissionType = TransmissionType.simple
id: str = ''
file_id: str = ''
secret: str = ''
@ -178,7 +178,7 @@ class DestFile:
self.permissions = ftc.permissions
self.ftype = ftc.ftype
self.ttype = ftc.ttype
self.needs_data_sent = self.ttype is not TransmisstionType.simple
self.needs_data_sent = self.ttype is not TransmissionType.simple
self.decompressor = ZlibDecompressor() if ftc.compression is Compression.zlib else IdentityDecompressor()
self.closed = self.ftype is FileType.directory

View File

@ -12,8 +12,8 @@ import zlib
from io import BytesIO
from kitty.file_transmission import (
Action, Compression, Container, FileTransmissionCommand,
TestFileTransmission as FileTransmission
Action, Compression, FileTransmissionCommand, FileType,
TestFileTransmission as FileTransmission, TransmissionType
)
from . import BaseTest
@ -26,14 +26,12 @@ def names_in(path):
def serialized_cmd(**fields) -> str:
for k, A in (('action', Action), ('container_fmt', Container), ('compression', Compression)):
for k, A in (('action', Action), ('ftype', FileType), ('ttype', TransmissionType), ('compression', Compression)):
if k in fields:
fields[k] = A[fields[k]]
if isinstance(fields.get('data'), str):
fields['data'] = fields['data'].encode('utf-8')
ans = FileTransmissionCommand()
for k in fields:
setattr(ans, k, fields[k])
ans = FileTransmissionCommand(**fields)
return ans.serialize()
@ -55,6 +53,7 @@ class TestFileTransmission(BaseTest):
self.ae(a, b)
def test_file_put(self):
return # disabled pending rewrite
# send refusal
for quiet in (0, 1, 2):
ft = FileTransmission()