starts refactoring tests

This commit is contained in:
artus
2018-11-18 21:56:58 +01:00
parent 0a6dd690c7
commit 0bbc61d386

View File

@@ -3,36 +3,39 @@
import os import os
import unittest import unittest
from pathlib import Path from pathlib import Path
from ..utils import Content, get_content, copy_files, move_files from ..store import Content, Store, Backup
class TestContent(unittest.TestCase): class TestContent(unittest.TestCase):
_test_files = set(("test1", "test2", "test3")) _test_files = set(("test1", "test2", "test3"))
test_folder = Path("./sandbox/")
test_subdir = test_folder / "subdir"
def _get_test_files_paths(self): @classmethod
for f in self._test_files: def _get_test_files_paths(cls):
for f in cls._test_files:
if f == "test1": if f == "test1":
yield self.test_subdir / f yield cls.test_subdir / f
else: else:
yield self.test_folder / f yield cls.test_folder / f
def setUp(self): @classmethod
self.test_folder = Path("./sandbox/") def setUpClass(cls):
if not self.test_folder.exists(): if not cls.test_folder.exists():
os.mkdir(self.test_folder) os.mkdir(cls.test_folder)
self.test_subdir = self.test_folder / "subdir" if not cls.test_subdir.exists():
if not self.test_subdir.exists(): os.mkdir(cls.test_subdir)
os.mkdir(self.test_subdir) for path in cls._get_test_files_paths():
for path in self._get_test_files_paths():
# Create an empty file # Create an empty file
open(path, 'w').close() open(path, 'w').close()
def tearDown(self): @classmethod
for path in self._get_test_files_paths(): def tearDownClass(cls):
for path in cls._get_test_files_paths():
os.remove(path) os.remove(path)
self.test_subdir.rmdir() cls.test_subdir.rmdir()
self.test_folder.rmdir() cls.test_folder.rmdir()
def test_content_difference(self): def test_content_difference(self):
first = Content(((1, 0), (2, 0), (3, 0), (4, 0))) first = Content(((1, 0), (2, 0), (3, 0), (4, 0)))
@@ -41,11 +44,11 @@ class TestContent(unittest.TestCase):
self.assertEqual(result, Content(((2, 0), (4, 0)))) self.assertEqual(result, Content(((2, 0), (4, 0))))
def test_get_content_without_exclude(self): def test_get_content_without_exclude(self):
content = get_content(self.test_folder, exclude=None) content = Store(self.test_folder).read_content(exclude=None)
self.assertEqual(set(content.keys()), self._test_files) self.assertEqual(set(content.keys()), self._test_files)
def test_get_content_with_exclude(self): def test_get_content_with_exclude(self):
content = get_content(self.test_folder, exclude="subdir") content = Store(self.test_folder).read_content(exclude="subdir")
self.assertEqual(set(content.keys()), set(("test2", "test3"))) self.assertEqual(set(content.keys()), set(("test2", "test3")))