diff options
Diffstat (limited to 'app/tests/test_audio.py')
| -rw-r--r-- | app/tests/test_audio.py | 19 |
1 files changed, 4 insertions, 15 deletions
diff --git a/app/tests/test_audio.py b/app/tests/test_audio.py index ef5e92a..6c67294 100644 --- a/app/tests/test_audio.py +++ b/app/tests/test_audio.py @@ -60,12 +60,7 @@ class CleanupChunksTests(unittest.TestCase): (chunks_dir / "chunk_0002.wav").write_bytes(b"stale") (chunks_dir / "keep.txt").write_bytes(b"keep") - original = audio.CHUNKS_FOLDER - audio.CHUNKS_FOLDER = chunks_dir - try: - cleanup_chunks() - finally: - audio.CHUNKS_FOLDER = original + cleanup_chunks(chunks_dir) self.assertFalse((chunks_dir / "chunk_0001.wav").exists()) self.assertFalse((chunks_dir / "chunk_0002.wav").exists()) @@ -77,12 +72,7 @@ class CleanupChunksTests(unittest.TestCase): (chunks_dir / "chapter_0001.m4b").write_bytes(b"stale") (chunks_dir / "chunk_0001.wav").write_bytes(b"stale") - original = audio.CHUNKS_FOLDER - audio.CHUNKS_FOLDER = chunks_dir - try: - cleanup_chunks() - finally: - audio.CHUNKS_FOLDER = original + cleanup_chunks(chunks_dir) self.assertFalse((chunks_dir / "chapter_0001.m4b").exists()) self.assertFalse((chunks_dir / "chunk_0001.wav").exists()) @@ -477,9 +467,7 @@ class CombineChunksPrintTests(unittest.TestCase): def setUp(self): self._tmp = tempfile.TemporaryDirectory() - self._chunks = patch.object(audio, "CHUNKS_FOLDER", Path(self._tmp.name)) - self._chunks.start() - self.addCleanup(self._chunks.stop) + self.addCleanup(self._tmp.cleanup) def _combine(self, total_chunks, chunk_results, intermediate=False): buf = io.StringIO() @@ -495,6 +483,7 @@ class CombineChunksPrintTests(unittest.TestCase): redirect_stdout(buf): ok = audio.combine_chunks( total_chunks, Path("out.m4b"), chunk_results, + chunks_dir=Path(self._tmp.name), output_format="m4b", intermediate=intermediate) self.assertTrue(ok) return buf.getvalue() |