This commit is contained in:
Kovid Goyal
2024-02-15 15:01:50 +05:30
parent 76381f5cdd
commit e7d6101bd4
3 changed files with 10 additions and 7 deletions

View File

@@ -6,6 +6,7 @@ import io
import os import os
import select import select
import shlex import shlex
import shutil
import signal import signal
import struct import struct
import sys import sys
@@ -179,6 +180,12 @@ class BaseTest(TestCase):
maxDiff = 2048 maxDiff = 2048
is_ci = os.environ.get('CI') == 'true' is_ci = os.environ.get('CI') == 'true'
def rmtree_ignoring_errors(self, tdir):
try:
shutil.rmtree(tdir)
except FileNotFoundError as err:
print('Failed to delete the directory:', tdir, 'with error:', err, file=sys.stderr)
def tearDown(self): def tearDown(self):
set_options(None) set_options(None)

View File

@@ -7,7 +7,7 @@ import shutil
import stat import stat
import tempfile import tempfile
from collections import namedtuple from collections import namedtuple
from contextlib import contextmanager, suppress from contextlib import contextmanager
from pathlib import Path from pathlib import Path
from kittens.transfer.rsync import Differ, Hasher, Patcher, parse_ftc from kittens.transfer.rsync import Differ, Hasher, Patcher, parse_ftc
@@ -188,8 +188,7 @@ class TestFileTransmission(BaseTest):
self.orig_home = os.environ.get('HOME') self.orig_home = os.environ.get('HOME')
def tearDown(self): def tearDown(self):
with suppress(FileNotFoundError): self.rmtree_ignoring_errors(self.tdir)
shutil.rmtree(self.tdir)
self.responses = [] self.responses = []
if self.orig_home is None: if self.orig_home is None:
os.environ.pop('HOME', None) os.environ.pop('HOME', None)

View File

@@ -2,11 +2,9 @@
# License: GPL v3 Copyright: 2017, Kovid Goyal <kovid at kovidgoyal.net> # License: GPL v3 Copyright: 2017, Kovid Goyal <kovid at kovidgoyal.net>
import os import os
import shutil
import sys import sys
import tempfile import tempfile
import unittest import unittest
from contextlib import suppress
from functools import partial from functools import partial
from kitty.constants import is_macos, read_kitty_resource from kitty.constants import is_macos, read_kitty_resource
@@ -35,8 +33,7 @@ class Rendering(BaseTest):
def tearDown(self): def tearDown(self):
self.test_ctx.__exit__() self.test_ctx.__exit__()
del self.sprites, self.cell_width, self.cell_height, self.test_ctx del self.sprites, self.cell_width, self.cell_height, self.test_ctx
with suppress(FileNotFoundError): self.rmtree_ignoring_errors(self.tdir)
shutil.rmtree(self.tdir)
super().tearDown() super().tearDown()
def test_sprite_map(self): def test_sprite_map(self):