def test_01_check_cache_for_egg(self):
"""
Test _check_cache_for_egg checks the cache for the egg,
returns path if present locally, or None if not.
"""
launcher = ZMQEggDriverProcess("DUMMY_VAL")
self.assertEqual(launcher._check_cache_for_egg("NOT_FOUND_EGG"), None)
self.assertEqual(launcher._check_cache_for_egg(EGG), None)
def test_01_check_cache_for_egg(self):
"""
Test _check_cache_for_egg checks the cache for the egg,
returns path if present locally, or None if not.
"""
launcher = ZMQEggDriverProcess("DUMMY_VAL")
self.assertEqual(launcher._check_cache_for_egg("NOT_FOUND_EGG"), None)
self.assertEqual(launcher._check_cache_for_egg(EGG), None)
def test_03_check_cache_for_egg(self):
"""
Test _check_cache_for_egg checks the cache for the egg,
returns path if present locally, or None if not.
"""
# Cleanup on isle one!
launcher = ZMQEggDriverProcess("DUMMY_VAL")
self.assertEqual(launcher._get_remote_egg(EGG), "/tmp/%s" % EGG)
self.assertEqual(launcher._check_cache_for_egg("NOT_FOUND_EGG"), None)
self.assertEqual(launcher._check_cache_for_egg(EGG), "/tmp/%s" % EGG)
def test_03_check_cache_for_egg(self):
"""
Test _check_cache_for_egg checks the cache for the egg,
returns path if present locally, or None if not.
"""
# Cleanup on isle one!
launcher = ZMQEggDriverProcess("DUMMY_VAL")
self.assertEqual(launcher._get_remote_egg(EGG), "/tmp/%s" % EGG)
self.assertEqual(launcher._check_cache_for_egg("NOT_FOUND_EGG"), None)
self.assertEqual(launcher._check_cache_for_egg(EGG), "/tmp/%s" % EGG)
def test_04_get_egg(self):
"""
Test _get_egg should return a path to a local egg for existing
eggs, and exception for non-existing in the repo.
"""
launcher = ZMQEggDriverProcess("DUMMY_VAL")
got_exception = False
try:
self.assertEqual(launcher._get_egg("NOT_FOUND_EGG"), None)
except DriverLaunchException:
got_exception = True
self.assertTrue(got_exception)
def test_04_get_egg(self):
"""
Test _get_egg should return a path to a local egg for existing
eggs, and exception for non-existing in the repo.
"""
launcher = ZMQEggDriverProcess("DUMMY_VAL")
got_exception = False
try:
self.assertEqual(launcher._get_egg("NOT_FOUND_EGG"), None)
except DriverLaunchException:
got_exception = True
self.assertTrue(got_exception)
def test_02_get_remote_egg(self):
"""
Test _get_remote_egg should return path for cached egg if present,
path for cached egg if not present locally, but in repo, or exception if not present locally or in repo.
"""
launcher = ZMQEggDriverProcess("DUMMY_VAL")
got_exception = False
try:
self.assertEqual(launcher._get_remote_egg("NOT_FOUND_EGG"), None)
except DriverLaunchException:
got_exception = True
self.assertTrue(got_exception)
self.assertEqual(launcher._get_remote_egg(EGG), "/tmp/%s" % EGG)
def test_02_get_remote_egg(self):
"""
Test _get_remote_egg should return path for cached egg if present,
path for cached egg if not present locally, but in repo, or exception if not present locally or in repo.
"""
launcher = ZMQEggDriverProcess("DUMMY_VAL")
got_exception = False
try:
self.assertEqual(launcher._get_remote_egg("NOT_FOUND_EGG"), None)
except DriverLaunchException:
got_exception = True
self.assertTrue(got_exception)
self.assertEqual(launcher._get_remote_egg(EGG), "/tmp/%s" % EGG)
def load_egg():
# Dynamically load the egg into the test path
dvr_config = make_config()
dvr_egg = dvr_config['dvr_egg']
dvr_sha = CFG.device.sbe37.dvr_sha
launcher = ZMQEggDriverProcess(dvr_config)
egg = launcher._get_egg(dvr_egg)
from hashlib import sha1
with open(egg,'r') as f:
doc = f.read()
sha = sha1(doc).hexdigest()
if sha != dvr_sha:
raise ImportError('Failed to load driver %s: incorrect checksum. (%s!=%s)' % (dvr_egg, dvr_sha, sha))
if not egg in sys.path: sys.path.insert(0, egg)
dvr_config['process_type'] = (DriverProcessType.EGG,)
return dvr_config
def load_egg(dvr_config=None):
# Dynamically load the egg into the test path
if dvr_config is None:
dvr_config = make_config()
dvr_egg = dvr_config['dvr_egg']
#dvr_sha = CFG.device.sbe37.dvr_sha
launcher = ZMQEggDriverProcess(dvr_config)
egg = launcher._get_egg(dvr_egg)
#from hashlib import sha1
with open(egg, 'r') as f:
doc = f.read()
#This does not work uniformly
#sha = sha1(doc).hexdigest()
#if sha != dvr_sha:
# raise ImportError('Failed to load driver %s: incorrect checksum. (%s!=%s)' % (dvr_egg, dvr_sha, sha))
if not egg in sys.path: sys.path.insert(0, egg)
dvr_config['process_type'] = (DriverProcessType.EGG, )
return dvr_config
# Driver config.
# DVR_CONFIG['comms_config']['port'] is set by the setup.
DVR_CONFIG = {
'dvr_egg': DRV_URI,
'dvr_mod': DRV_MOD,
'dvr_cls': DRV_CLS,
'workdir': WORK_DIR,
'process_type': None
}
# Launch from egg or a local MI repo.
LAUNCH_FROM_EGG = True
if LAUNCH_FROM_EGG:
# Dynamically load the egg into the test path
launcher = ZMQEggDriverProcess(DVR_CONFIG)
egg = launcher._get_egg(DRV_URI)
from hashlib import sha1
with open(egg, 'r') as f:
doc = f.read()
sha = sha1(doc).hexdigest()
if not egg in sys.path: sys.path.insert(0, egg)
DVR_CONFIG['process_type'] = (DriverProcessType.EGG, )
else:
mi_repo = os.getcwd() + os.sep + 'extern' + os.sep + 'mi_repo'
if not mi_repo in sys.path: sys.path.insert(0, mi_repo)
DVR_CONFIG['process_type'] = (DriverProcessType.PYTHON_MODULE, )
DVR_CONFIG['mi_repo'] = mi_repo
DRV_SHA = DRV_SHA_0_1_5
DRV_MOD = 'mi.instrument.seabird.sbe37smb.ooicore.driver'
DRV_CLS = 'SBE37Driver'
# Driver config.
# DVR_CONFIG['comms_config']['port'] is set by the setup.
DVR_CONFIG = {
'dvr_egg' : DRV_URI,
'dvr_mod' : DRV_MOD,
'dvr_cls' : DRV_CLS,
'workdir' : WORK_DIR,
'process_type' : (DriverProcessType.EGG,)
}
# Dynamically load the egg into the test path
launcher = ZMQEggDriverProcess(DVR_CONFIG)
egg = launcher._get_egg(DRV_URI)
from hashlib import sha1
with open(egg,'r') as f:
doc = f.read()
sha = sha1(doc).hexdigest()
if sha != DRV_SHA:
raise ImportError('Failed to load driver %s: incorrect checksum. (%s!=%s)' % (DRV_URI, DRV_SHA, sha))
if not egg in sys.path: sys.path.insert(0, egg)
# Load MI modules from the egg
from mi.core.instrument.instrument_driver import DriverProtocolState
from mi.core.instrument.instrument_driver import DriverConnectionState
from mi.core.exceptions import InstrumentParameterException
from mi.instrument.seabird.sbe37smb.ooicore.driver import SBE37ProtocolEvent
from mi.instrument.seabird.sbe37smb.ooicore.driver import SBE37Parameter
def _set_up_pre_environment_for_instrument(self, instr_info):
"""
Based on test_instrument_agent.py
Basically, this method launches a port agent and then completes the
instrument driver configuration used to properly set up a particular
instrument agent.
@param instr_info A value in instruments_dict
@return instrument_driver_config
"""
import sys
from ion.agents.instrument.driver_process import DriverProcessType
from ion.agents.instrument.driver_process import ZMQEggDriverProcess
# A seabird driver.
DRV_URI = SBE37_EGG
DRV_MOD = 'mi.instrument.seabird.sbe37smb.ooicore.driver'
DRV_CLS = 'SBE37Driver'
WORK_DIR = '/tmp/'
DELIM = ['<<', '>>']
instrument_driver_config = {
'dvr_egg' : DRV_URI,
'dvr_mod' : DRV_MOD,
'dvr_cls' : DRV_CLS,
'workdir' : WORK_DIR,
'process_type' : None
}
# Launch from egg or a local MI repo.
LAUNCH_FROM_EGG=True
if LAUNCH_FROM_EGG:
# Dynamically load the egg into the test path
launcher = ZMQEggDriverProcess(instrument_driver_config)
egg = launcher._get_egg(DRV_URI)
if not egg in sys.path: sys.path.insert(0, egg)
instrument_driver_config['process_type'] = (DriverProcessType.EGG,)
else:
mi_repo = os.getcwd() + os.sep + 'extern' + os.sep + 'mi_repo'
if not mi_repo in sys.path: sys.path.insert(0, mi_repo)
instrument_driver_config['process_type'] = (DriverProcessType.PYTHON_MODULE,)
instrument_driver_config['mi_repo'] = mi_repo
DEV_ADDR = instr_info['DEV_ADDR']
DEV_PORT = instr_info['DEV_PORT']
DATA_PORT = instr_info['DATA_PORT']
CMD_PORT = instr_info['CMD_PORT']
PA_BINARY = instr_info['PA_BINARY']
support = DriverIntegrationTestSupport(None,
None,
DEV_ADDR,
DEV_PORT,
DATA_PORT,
CMD_PORT,
PA_BINARY,
DELIM,
WORK_DIR)
# Start port agent, add stop to cleanup.
port = support.start_pagent()
log.info('Port agent started at port %i', port)
self.addCleanup(support.stop_pagent)
# Configure instrument driver to use port agent port number.
instrument_driver_config['comms_config'] = {
'addr': 'localhost',
'port': port,
'cmd_port': CMD_PORT
}
return instrument_driver_config
def _set_up_pre_environment_for_instrument(self):
"""
From test_instrument_agent.py
Basically, this method launches the port agent and the completes the
instrument driver configuration used to properly set up the
instrument agent.
@return instrument_driver_config
"""
import sys
from ion.agents.instrument.driver_process import DriverProcessType
from ion.agents.instrument.driver_process import ZMQEggDriverProcess
# A seabird driver.
DRV_URI = 'http://sddevrepo.oceanobservatories.org/releases/seabird_sbe37smb_ooicore-0.0.7-py2.7.egg'
DRV_MOD = 'mi.instrument.seabird.sbe37smb.ooicore.driver'
DRV_CLS = 'SBE37Driver'
WORK_DIR = '/tmp/'
DELIM = ['<<', '>>']
instrument_driver_config = {
'dvr_egg' : DRV_URI,
'dvr_mod' : DRV_MOD,
'dvr_cls' : DRV_CLS,
'workdir' : WORK_DIR,
'process_type' : None
}
# Launch from egg or a local MI repo.
LAUNCH_FROM_EGG=True
if LAUNCH_FROM_EGG:
# Dynamically load the egg into the test path
launcher = ZMQEggDriverProcess(instrument_driver_config)
egg = launcher._get_egg(DRV_URI)
if not egg in sys.path: sys.path.insert(0, egg)
instrument_driver_config['process_type'] = (DriverProcessType.EGG,)
else:
mi_repo = os.getcwd() + os.sep + 'extern' + os.sep + 'mi_repo'
if not mi_repo in sys.path: sys.path.insert(0, mi_repo)
instrument_driver_config['process_type'] = (DriverProcessType.PYTHON_MODULE,)
instrument_driver_config['mi_repo'] = mi_repo
DEV_ADDR = CFG.device.sbe37.host
DEV_PORT = CFG.device.sbe37.port
DATA_PORT = CFG.device.sbe37.port_agent_data_port
CMD_PORT = CFG.device.sbe37.port_agent_cmd_port
PA_BINARY = CFG.device.sbe37.port_agent_binary
self._support = DriverIntegrationTestSupport(None,
None,
DEV_ADDR,
DEV_PORT,
DATA_PORT,
CMD_PORT,
PA_BINARY,
DELIM,
WORK_DIR)
# Start port agent, add stop to cleanup.
port = self._support.start_pagent()
log.info('Port agent started at port %i', port)
self.addCleanup(self._support.stop_pagent)
# Configure instrument driver to use port agent port number.
instrument_driver_config['comms_config'] = {
'addr': 'localhost',
'port': port,
'cmd_port': CMD_PORT
}
return instrument_driver_config
# A seabird driver.
DRV_MOD = 'mi.instrument.seabird.sbe37smb.ooicore.driver'
DRV_CLS = 'SBE37Driver'
WORK_DIR = '/tmp/'
DVR_CONFIG = {
'dvr_egg' : DRV_URI_GOOD,
'dvr_mod' : DRV_MOD,
'dvr_cls' : DRV_CLS,
'workdir' : WORK_DIR,
'process_type' : None
}
# Dynamically load the egg into the test path
launcher = ZMQEggDriverProcess(DVR_CONFIG)
egg = launcher._get_egg(DRV_URI_GOOD)
if not egg in sys.path:
sys.path.insert(0, egg)
# now we can import SBE37ProtocolEvent
from mi.instrument.seabird.sbe37smb.ooicore.driver import SBE37ProtocolEvent
# ------------------------------------------------------------------------
@patch.dict(CFG, {'endpoint': {'receive': {'timeout': 180}}})
class TestPlatformInstrument(BaseIntTestPlatform):
# def setUp(self):
# # Start container
