def setUp(self):
ParserUnitTestCase.setUp(self)
self.error_callback_values = []
self.state_callback_values = []
self.publish_callback_values = []
self.parser_config = {
ParserConfigKey.ORBNAME: ParserConfigKey.ORBNAME,
ParserConfigKey.SELECT: ParserConfigKey.SELECT,
ParserConfigKey.REJECT: ParserConfigKey.REJECT,
}
self.parser_state = None
self.PKT_ID = PKT_ID = 123
self.PKT_TYPE = PKT_TYPE = 'GENC'
self.PKT_DATA = PKT_DATA = 1, 2, 3, 4
self.PKT_TIME = PKT_TIME = 999
self.PKT_SAMPRATE = PKT_SAMPRATE = 666
self.PKT_NET = PKT_NET = 'net'
self.PKT_STA = PKT_STA = 'sta'
self.PKT_CHAN = PKT_CHAN = 'chan'
self.PKT_LOC = PKT_LOC = 'loc'
from mi.core.kudu import _pkt
pkt = _pkt._newPkt()
_pkt._Pkt_pkttype_set(pkt, PKT_TYPE)
pktchan = _pkt._newPktChannel()
_pkt._PktChannel_data_set(pktchan, PKT_DATA)
_pkt._PktChannel_samprate_set(pktchan, PKT_SAMPRATE)
_pkt._PktChannel_time_set(pktchan, PKT_TIME)
_pkt._PktChannel_net_set(pktchan, PKT_NET)
_pkt._PktChannel_sta_set(pktchan, PKT_STA)
_pkt._PktChannel_chan_set(pktchan, PKT_CHAN)
_pkt._PktChannel_loc_set(pktchan, PKT_LOC)
_pkt._Pkt_channels_set(pkt, [
pktchan,
])
pkttype, packet, srcname, time = _pkt._stuffPkt(pkt)
_pkt._freePkt(pkt)
with patch(
'mi.dataset.parser.antelope_orb.OrbReapThr') as MockOrbReapThr:
self.parser = AntelopeOrbParser(self.parser_config,
self.parser_state,
self.state_callback,
self.pub_callback,
self.error_callback)
self.parser._orbreapthr.get = MagicMock(return_value=(PKT_ID, srcname,
time, packet))
def _build_parser(self):
"""
Build and return the parser
"""
config = self._parser_config
config.update({
'particle_module': 'mi.dataset.parser.antelope_orb',
'particle_class': ['AntelopeOrbPacketParticle']
})
log.debug("My Config: %s", config)
log.debug("My parser state: %s", self._driver_state)
self._parser = AntelopeOrbParser(
config,
self._driver_state.get(DriverStateKey.PARSER_STATE),
self._save_parser_state,
self._data_callback,
self._sample_exception_callback,
)
return self._parser
