1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
|
#!/usr/bin/env python
from __future__ import print_function
import unittest
from ruffus import pipeline_run, Pipeline, parallel, proxy_logger
import ruffus
"""
test_exceptions.py
"""
import os
import sys
import logging
# add grandparent to search path for testing
grandparent_dir = os.path.abspath(
os.path.join(os.path.dirname(__file__), "..", ".."))
sys.path.insert(0, grandparent_dir)
# module name = script name without extension
module_name = os.path.splitext(os.path.basename(__file__))[0]
# funky code to import by file name
parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))
def logging_factory(logger_name, listargs):
root_logger = logging.getLogger(logger_name)
root_logger.setLevel(logging.DEBUG)
handler = logging.StreamHandler(sys.stderr)
formatter_ = logging.Formatter("%(levelname)7s - %(message)s")
handler.setFormatter(formatter_)
handler.setLevel(logging.INFO)
root_logger.addHandler(handler)
return root_logger
log, log_mutex = proxy_logger.make_shared_logger_and_proxy(
logging_factory, __name__, [])
# 88888888888888888888888888888888888888888888888888888888888888888888888888888888888888888
# Tasks
# 88888888888888888888888888888888888888888888888888888888888888888888888888888888888888888
@parallel([['A', 1], ['B', 3], ['C', 3], ['D', 4], ['E', 4], ['F', 4]])
def parallel_task(name, param1):
sys.stderr.write(" Parallel task %s: \n" % name)
#raise task.JobSignalledBreak("Oops! I did it again!")
with log_mutex:
log.info(" Raising exception")
raise Exception("new")
try:
from StringIO import StringIO
except:
from io import StringIO
class Test_ruffus(unittest.TestCase):
def test_ruffus(self):
try:
pipeline_run(multiprocess=50, verbose=0, pipeline="main")
except ruffus.ruffus_exceptions.RethrownJobError:
return
raise Exception("Missing exception")
def test_exception_logging(self):
try:
pipeline_run(multiprocess=50, verbose=0, pipeline="main")
except ruffus.ruffus_exceptions.RethrownJobError as e:
log.info(e)
for exc in e.args:
task_name, job_name, exc_name, exc_value, exc_stack = exc
return
raise Exception("Missing exception")
def test_newstyle_ruffus(self):
test_pipeline = Pipeline("test")
test_pipeline.parallel(parallel_task, [['A', 1], ['B', 3], [
'C', 3], ['D', 4], ['E', 4], ['F', 4]])
try:
test_pipeline.run(multiprocess=50, verbose=0)
except ruffus.ruffus_exceptions.RethrownJobError:
return
raise Exception("Missing exception")
if __name__ == '__main__':
unittest.main()
|
