Home
Diffusion Flamenco Worker (master) Browse

flamenco-worker/tests/test_commands_remove_tree.py
7441685963d9master

Last Change
Raw File

test_commands_remove_tree.py
View Options

from pathlib import Path
import os
from test_runner import AbstractCommandTest
class RemoveTreeTest(AbstractCommandTest):
def setUp(self):
super().setUp()
from flamenco_worker.commands import RemoveTreeCommand
import tempfile
self.tmpdir = tempfile.TemporaryDirectory()
self.tmppath = Path(self.tmpdir.name)
self.cmd = RemoveTreeCommand(
worker=self.fworker,
task_id='12345',
command_idx=0,
)
def tearDown(self):
super().tearDown()
self.tmpdir.cleanup()
def test_validate_settings(self):
self.assertIn('path', self.cmd.validate({'path': 12}))
self.assertIn('path', self.cmd.validate({'path': ''}))
self.assertIn('path', self.cmd.validate({}))
self.assertFalse(self.cmd.validate({'path': '/some/path'}))
def test_nonexistant_source(self):
path = self.tmppath / 'nonexisting'
task = self.cmd.run({'path': str(path)})
ok = self.loop.run_until_complete(task)
self.assertTrue(ok)
self.assertFalse(path.exists())
def test_source_file(self):
path = self.tmppath / 'existing'
path.touch()
task = self.cmd.run({'path': str(path)})
ok = self.loop.run_until_complete(task)
self.assertTrue(ok)
self.assertFalse(path.exists())
def test_soure_dir_with_files(self):
path = self.tmppath / 'dir'
path.mkdir()
(path / 'a.file').touch()
(path / 'b.file').touch()
(path / 'c.file').touch()
task = self.cmd.run({'path': str(path)})
ok = self.loop.run_until_complete(task)
self.assertTrue(ok)
self.assertFalse(path.exists())
def test_soure_dir_with_files_and_dirs(self):
path = self.tmppath / 'dir'
path.mkdir()
(path / 'subdir-a' / 'subsub-1').mkdir(parents=True)
(path / 'subdir-a' / 'subsub-2').mkdir()
(path / 'subdir-a' / 'subsub-3').mkdir()
(path / 'subdir-b' / 'subsub-1').mkdir(parents=True)
(path / 'subdir-c' / 'subsub-1').mkdir(parents=True)
(path / 'subdir-c' / 'subsub-2').mkdir()
(path / 'a.file').touch()
(path / 'b.file').touch()
(path / 'c.file').touch()
(path / 'subdir-a' / 'subsub-1' / 'a.file').touch()
(path / 'subdir-a' / 'subsub-1' / 'b.file').touch()
(path / 'subdir-a' / 'subsub-2' / 'a.file').touch()
(path / 'subdir-a' / 'subsub-2' / 'b.file').touch()
(path / 'subdir-a' / 'subsub-3' / 'a.file').touch()
(path / 'subdir-a' / 'subsub-3' / 'b.file').touch()
(path / 'subdir-b' / 'subsub-1' / 'a.file').touch()
(path / 'subdir-b' / 'subsub-1' / 'b.file').touch()
(path / 'subdir-c' / 'subsub-1' / 'a.file').touch()
(path / 'subdir-c' / 'subsub-1' / 'b.file').touch()
(path / 'subdir-c' / 'subsub-2' / 'a.file').touch()
(path / 'subdir-c' / 'subsub-2' / 'b.file').touch()
task = self.cmd.run({'path': str(path)})
ok = self.loop.run_until_complete(task)
self.assertTrue(ok)
self.assertFalse(path.exists())