eris/tests/worker_test.py
Andrew Hamilton 4935c585aa Made work on Ubuntu 17.10.
- Changed to python3.6.
- Accepted minor differences in golden files.
- php is now 7.1.
- Updated README.
- Need to test that the AppImage still works.
2017-10-27 12:39:50 +10:00

42 lines
1.1 KiB
Python
Executable file

#!/usr/bin/env python3.6
# Copyright (C) 2017 Andrew Hamilton. All rights reserved.
# Licensed under the Artistic License 2.0.
import asyncio
import os
import shutil
import tempfile
import unittest
import vigil.tools as tools
import vigil.worker as worker
class WorkerTestCase(unittest.TestCase):
def setUp(self):
self.temp_dir = tempfile.mkdtemp()
self.original_working_dir = os.getcwd()
os.chdir(self.temp_dir)
os.mkdir(tools.CACHE_PATH)
open("foo", "w").close()
def tearDown(self):
shutil.rmtree(self.temp_dir)
os.chdir(self.original_working_dir)
def test_run_job(self):
loop = asyncio.get_event_loop()
worker_ = worker.Worker(False, False)
loop.run_until_complete(worker_.create_process())
future = worker_.run_tool("foo", tools.metadata)
status = loop.run_until_complete(future)
self.assertEqual(status, tools.Status.normal)
result_path = os.path.join(tools.CACHE_PATH, "foo-metadata")
self.assertTrue(os.path.exists(result_path))
if __name__ == "__main__":
unittest.main()