parser.add_argument('--ip_addr', dest='ip_addr', default=None,
help='IP address of this client')
parser.add_argument('--secret', dest='secret', default='',
help='authentication secret for handshake with nodes')
parser.add_argument('--certfile', dest='certfile', default='',
help='file containing SSL certificate')
parser.add_argument('--keyfile', dest='keyfile', default='',
help='file containing SSL key')
parser.add_argument('--scheduler_node', dest='scheduler_node', default=None,
help='name or IP address where dispyscheduler is running to which '
'jobs are submitted')
config = vars(parser.parse_args(sys.argv[1:]))
if config['loglevel']:
logger.setLevel(logger.DEBUG)
pycos.logger.setLevel(logger.DEBUG)
else:
logger.setLevel(logger.INFO)
del config['loglevel']
if config['certfile']:
config['certfile'] = os.path.abspath(config['certfile'])
else:
config['certfile'] = None
if config['keyfile']:
config['keyfile'] = os.path.abspath(config['keyfile'])
else:
config['keyfile'] = None
args = config.pop('args')
import unittest
import os
import shutil
from dispy import logger as dispy_logger
import fission.core.base as base
from fission.core.nodes import BaseNode
from fission.core.jobs import BaseJob
dispy_logger.setLevel(dispy_logger.ERROR)
class TestJob(unittest.TestCase):
def setUp(self):
self.Node1 = BaseNode("192.168.4.2")
self.Node2 = BaseNode("192.168.4.3")
self.Job = BaseJob()
self.Job2 = BaseJob()
def tearDown(self):
base.NODES = {}
base.JOBS = {}
def test_allocate(self):
self.assertEqual(self.Job.allocated, None)
self.assertEqual(self.Job.status, self.Job.STOPPED)
self.Job.allocate(self.Node1)
with self.assertRaises(RuntimeError):
self.Job.allocate(self.Node2)
with self.assertRaises(RuntimeError):
self.Job2.allocate("NotANode")