def test_on_created(self):
"""Tests basic cleanup functionality."""
watch = cleanup.Cleanup('/does/not/exist', 2)
watch._on_created('test.yml')
self.assertEqual(2, treadmill.fs.rm_safe.call_count)
self.assertEqual(1, shutil.rmtree.call_count)
self.assertEqual(1, treadmill.spawn.cleanup.Cleanup._nuke.call_count)
def cleanup_cmd(approot):
"""Spawn cleanup process."""
watch = cleanup.Cleanup(approot)
dirwatch = watch.get_dir_watch()
watch.sync()
while True:
if dirwatch.wait_for_events(60):
dirwatch.process_events()
def test_sync(self):
"""Tests the initial sync of the manifests."""
os.listdir.side_effect = [
['job1'],
]
watch = cleanup.Cleanup('/does/not/exist', 2)
watch.sync()
treadmill.spawn.cleanup.Cleanup._on_created \
.assert_called_with('/does/not/exist/cleanup/job1')