occheung
/
artiq
forked from M-Labs/artiq
1
0
Fork 0
Code Pull Requests Activity

test_device_db: fix on Windows (tempfile access limitations) 

Signed-off-by: Etienne Wodey <wodey@iqo.uni-hannover.de>
This commit is contained in:
Etienne Wodey 2021-06-21 10:39:39 +02:00 committed by Sébastien Bourdeauducq
parent b8ab5f2607
commit cca654bd47
1 changed files with 18 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
artiq/test/test_device_db.py
View File

@ -1,5 +1,6 @@
"""Test device DB interface""" """Test device DB interface"""
import os
import unittest import unittest
import tempfile import tempfile
from pathlib import Path from pathlib import Path
@ -22,19 +23,27 @@ device_db = {
} }
""" """
# tempfile.NamedTemporaryFile:
# use delete=False and manual cleanup
# for Windows compatibility
class TestDeviceDBImport(unittest.TestCase): class TestDeviceDBImport(unittest.TestCase):
def test_no_device_db_in_file(self): def test_no_device_db_in_file(self):
with tempfile.NamedTemporaryFile(mode="w+", suffix=".py") as f: with tempfile.NamedTemporaryFile(mode="w+", suffix=".py", delete=False) as f:
print(f.name)
print("", file=f, flush=True) print("", file=f, flush=True)
with self.assertRaisesRegex(KeyError, "device_db"): with self.assertRaisesRegex(KeyError, "device_db"):
DeviceDB(f.name) DeviceDB(f.name)
os.unlink(f.name)
def test_import_same_level(self): def test_import_same_level(self):
with tempfile.TemporaryDirectory() as tmpdir: with tempfile.TemporaryDirectory() as tmpdir:
# make sure both files land in the same directory # make sure both files land in the same directory
args = dict(mode="w+", suffix=".py", dir=tmpdir) # tempfiles are cleanup together with tmpdir
args = dict(mode="w+", suffix=".py", dir=tmpdir, delete=False)
with tempfile.NamedTemporaryFile( with tempfile.NamedTemporaryFile(
**args **args
) as fileA, tempfile.NamedTemporaryFile(**args) as fileB: ) as fileA, tempfile.NamedTemporaryFile(**args) as fileB:
@ -58,11 +67,17 @@ device_db["new_core_alias"] = "core"
class TestDeviceDB(unittest.TestCase): class TestDeviceDB(unittest.TestCase):
def setUp(self): def setUp(self):
self.ddb_file = tempfile.NamedTemporaryFile(mode="w+", suffix=".py") self.ddb_file = tempfile.NamedTemporaryFile(
mode="w+", suffix=".py", delete=False
)
print(DUMMY_DDB_FILE, file=self.ddb_file, flush=True) print(DUMMY_DDB_FILE, file=self.ddb_file, flush=True)
self.ddb = DeviceDB(self.ddb_file.name) self.ddb = DeviceDB(self.ddb_file.name)
def tearDown(self):
self.ddb_file.close()
os.unlink(self.ddb_file.name)
def test_get(self): def test_get(self):
core = self.ddb.get("core") core = self.ddb.get("core")
self.assertEqual(core["class"], "Core") self.assertEqual(core["class"], "Core")