def __init__(self):
settings = load_settings()
super(FunctionClient,self).__init__(
settings.HOST, \
settings.CONTROLLER_PORT, \
key='%s/master.key' % settings.RUNTIME_FILES_DIR
)
def setup_test_environment():
"""
configures test environment:
* setup django environment
* load pydra settings
* silence logging
"""
# configure pydra and django environment
configure_django_settings()
settings = load_settings()
# silence logging
settings.LOG_LEVEL = 100
logger = logging.getLogger('root')
logger.level = 100
You should have received a copy of the GNU General Public License
along with Pydra. If not, see <http://www.gnu.org/licenses/>.
"""
import inspect
import os.path
import shutil
import tempfile
import time
import unittest
from datetime import datetime
#environment must be configured before loading tests
from pydra.config import configure_django_settings, load_settings
configure_django_settings()
load_settings()
import pydra_settings
from pydra.cluster.tasks import TaskNotFoundException, packaging
from pydra.cluster.tasks.task_manager import TaskManager
from pydra.models import TaskInstance
from pydra.util import makedirs
from pydra.tests.mixin_testcases import ModuleTestCaseMixIn
from pydra.tests import django_testcase
test_string = """
from pydra.cluster.tasks import Task
class TestTask(Task):
def work(self, **kwargs): pass
"""
#!/usr/bin/python
import os, sys, getopt, warnings
from pydra.config import load_settings
settings = load_settings()
from pydra.logs.logger import task_log_path
warnings.simplefilter('ignore', UserWarning)
def get_all_logs():
files = os.listdir(settings.LOG_ARCHIVE)
files.sort()
return files
def print_logs(ids):
ts = {}
for id in ids:
try:
# look up log dir and main task.log file
dir, logfile = task_log_path(id)
# get all extra workunit files in dir
files = os.listdir(dir)
files.sort()
files.remove('task.log')
for f in files:
file = open(dir+f, 'r')
line = file.readline()
line = line.split(' ', 2)
line = "%s %s" % (line[0], line[1])
file.seek(0)
ts[line] = file
