Skip to content

Commit c546bd0

Browse files
committed
edit unittest to work on windows
1 parent caeab0e commit c546bd0

File tree

1 file changed

+9
-8
lines changed

1 file changed

+9
-8
lines changed

test/test_logger.py

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,8 @@
88
from unittest import mock
99
from unittest.mock import Mock
1010
import gzip
11+
import os
1112
import sys
12-
import tempfile
1313
import can
1414
import can.logger
1515

@@ -118,24 +118,25 @@ def setUp(self) -> None:
118118
self.testmsg = can.Message(
119119
arbitration_id=0xC0FFEE, data=[0, 25, 0, 1, 3, 1, 4, 1], is_extended_id=True
120120
)
121-
122121
self.baseargs = [sys.argv[0], "-i", "virtual"]
123122

123+
self.testfile = open("coffee.log.gz", "w+")
124+
124125
def test_compressed_logfile(self):
125126
"""
126127
Basic test to verify Logger is able to write gzip files.
127128
"""
128129
self.mock_virtual_bus.recv = Mock(side_effect=[self.testmsg, KeyboardInterrupt])
129-
130-
with tempfile.NamedTemporaryFile(suffix=".log.gz", delete=True) as compressed:
131-
sys.argv = self.baseargs + ["--file_name", compressed.name]
132-
can.logger.main()
133-
with gzip.open(compressed.name, "rt") as decompressed:
134-
last_line = decompressed.readlines()[-1]
130+
sys.argv = self.baseargs + ["--file_name", self.testfile.name]
131+
can.logger.main()
132+
with gzip.open(self.testfile.name, "rt") as testlog:
133+
last_line = testlog.readlines()[-1]
135134

136135
self.assertEqual(last_line, "(0.000000) vcan0 00C0FFEE#0019000103010401\n")
137136

138137
def tearDown(self) -> None:
138+
self.testfile.close()
139+
os.remove(self.testfile.name)
139140
self.patcher_virtual_bus.stop()
140141

141142

0 commit comments

Comments
 (0)