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
|
import sys
import importlib
from io import StringIO
from tests import TestCase
def get_var(tool_name, entry="main"):
mod = importlib.import_module(
"mutagen._tools.%s" % tool_name.replace("-", "_"))
return getattr(mod, entry)
class _TTools(TestCase):
TOOL_NAME = None
def setUp(self):
self.assertTrue(isinstance(self.TOOL_NAME, str))
self._main = get_var(self.TOOL_NAME)
def get_var(self, name):
return get_var(self.TOOL_NAME, name)
def call2(self, *args):
for arg in args:
self.assertTrue(isinstance(arg, str))
old_stdout = sys.stdout
old_stderr = sys.stderr
try:
out = StringIO()
err = StringIO()
sys.stdout = out
sys.stderr = err
try:
ret = self._main([self.TOOL_NAME] + list(args))
except SystemExit as e:
ret = e.code
ret = ret or 0
out_val = out.getvalue()
err_val = err.getvalue()
return (ret, out_val, err_val)
finally:
sys.stdout = old_stdout
sys.stderr = old_stderr
def call(self, *args):
return self.call2(*args)[:2]
def tearDown(self):
del self._main
|
