def test_long_stream(self):
stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_shorter.dat'))
self.parser = DostadParser(self.config, None, stream_handle,
self.state_callback, self.pub_callback, self.exception_callback)
result = self.parser.get_records(7)
stream_handle.close()
self.assertEqual(result[0], self.particle_metadata)
self.assertEqual(result[1], self.particle_a)
self.assertEqual(result[2], self.particle_b)
self.assertEqual(result[3], self.particle_c)
self.assertEqual(result[4], self.particle_d)
self.assertEqual(result[5], self.particle_e)
self.assertEqual(result[6], self.particle_f)
# 0-69 contains an incomplete block (end of a sample)
# 1329-1332 there are 3 extra \n's between sio blocks
# 2294-2363, and 4092-4161 contains an error text string in between two sio blocks
# 4351-4927 has a bad AD then CT message where the size from the header does not line up with
# the final \x03
self.assert_state([],
[[0, 69], [1329,1332], [2294,2363], [4092,4161], [4351, 4927], [9020, 9374]])
self.assertEqual(self.publish_callback_value[5], self.particle_e)
self.assertEqual(self.publish_callback_value[6], self.particle_f)
self.assertEqual(self.exception_callback_value, None)
def test_set_state(self):
"""
test changing the state after initializing
"""
self.state = {
StateKey.UNPROCESSED_DATA: [[0, 1000]],
StateKey.IN_PROCESS_DATA: [],
StateKey.TIMESTAMP: 0.0
}
new_state = {
StateKey.UNPROCESSED_DATA: [[0, 69], [944, 6300]],
StateKey.IN_PROCESS_DATA: [],
StateKey.TIMESTAMP: self.timestamp2
}
self.stream_handle = open(
os.path.join(RESOURCE_PATH, 'node59p1_shorter.dat'))
self.parser = DostadParser(
self.config, self.state, self.stream_handle, self.state_callback,
self.pub_callback) # last one is the link to the data source
# there should only be 3 records, make sure we stop there
result = self.parser.get_records(3)
self.assert_state([], [[0, 69], [944, 1000]], self.timestamp3)
result = self.parser.get_records(1)
self.assertEqual(result, [])
self.parser.set_state(new_state)
result = self.parser.get_records(1)
self.stream_handle.close()
self.assert_result(result, [],
[[0, 69], [944, 2370], [2560, 2947], [3137, 4173],
[4363, 5437], [5683, 6072], [6267, 6300]],
self.timestamp4, self.particle_d)
def test_in_process_start(self):
"""
test starting a parser with a state in the middle of processing
"""
new_state = {StateKey.IN_PROCESS_DATA:[[390, 507, 1, 0], [637, 754, 1, 0], [6131, 6248, 1, 0]],
StateKey.UNPROCESSED_DATA:[ [390,6300]],
StateKey.METADATA_SENT: True,
StateKey.FILE_SIZE: 9400}
stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_shorter.dat'))
self.parser = DostadParser(self.config, new_state, stream_handle,
self.state_callback, self.pub_callback, self.exception_callback)
result = self.parser.get_records(1)
self.assert_result(result, [[637, 754, 1, 0], [6131, 6248, 1, 0]],
[[507,6300]],
self.particle_b)
result = self.parser.get_records(2)
self.assertEqual(result[0], self.particle_c)
self.assertEqual(result[1], self.particle_d)
self.assert_state([],
[[507,637], [754,6131], [6248, 6300]])
self.assertEqual(self.publish_callback_value[-1], self.particle_d)
self.assertEqual(self.exception_callback_value, None)
def test_long_stream(self):
self.stream_handle = open(
os.path.join(RESOURCE_PATH, 'node59p1_shorter.dat'))
data = self.stream_handle.read()
data_len = len(data)
self.stream_handle.seek(0)
self.state = {
StateKey.UNPROCESSED_DATA: [[0, data_len]],
StateKey.IN_PROCESS_DATA: [],
StateKey.TIMESTAMP: 0.0
}
self.parser = DostadParser(
self.config, self.state, self.stream_handle, self.state_callback,
self.pub_callback) # last one is the link to the data source
result = self.parser.get_records(6)
self.stream_handle.close()
self.assertEqual(result[0], self.particle_a)
self.assertEqual(result[1], self.particle_b)
self.assertEqual(result[2], self.particle_c)
self.assertEqual(result[-3], self.particle_d)
self.assertEqual(result[-2], self.particle_e)
self.assertEqual(result[-1], self.particle_f)
self.assert_state([],
[[0, 69], [944, 2370], [2560, 2947], [3137, 4173],
[4363, 5437], [5683, 6072], [8273, 9400]],
self.timestamp6)
self.assertEqual(self.publish_callback_value[-2], self.particle_e)
self.assertEqual(self.publish_callback_value[-1], self.particle_f)
def test_in_process_start(self):
"""
test starting a parser with a state in the middle of processing
"""
new_state = {
StateKey.IN_PROCESS_DATA: [[390, 507, 1, 0, 0],
[637, 754, 1, 0, 0]],
StateKey.UNPROCESSED_DATA: [[0, 69], [390, 507], [637, 754],
[944, 6300]],
StateKey.TIMESTAMP:
self.timestamp3
}
self.stream_handle = open(
os.path.join(RESOURCE_PATH, 'node59p1_shorter.dat'))
self.parser = DostadParser(
self.config, new_state, self.stream_handle, self.state_callback,
self.pub_callback) # last one is the link to the data source
result = self.parser.get_records(1)
# even though the state says this particle is not a new sequence, since it is the
# first after setting the state it will be new
self.assert_result(result, [[637, 754, 1, 0, 0]],
[[0, 69], [637, 754], [944, 6300]], self.timestamp2,
self.particle_b_new)
result = self.parser.get_records(2)
self.assertEqual(result[0], self.particle_c)
self.assertEqual(result[1], self.particle_d)
self.assert_state([],
[[0, 69], [944, 2370], [2560, 2947], [3137, 4173],
[4363, 5437], [5683, 6072], [6267, 6300]],
self.timestamp4)
self.assertEqual(self.publish_callback_value[-1], self.particle_d)
def test_mid_state_start(self):
"""
test starting a parser with a state in the middle of processing
"""
new_state = {StateKey.IN_PROCESS_DATA:[],
StateKey.UNPROCESSED_DATA:[[0,69], [314,6300]],
StateKey.METADATA_SENT: True,
StateKey.FILE_SIZE: 9400}
stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_shorter.dat'))
self.parser = DostadParser(self.config, new_state, stream_handle,
self.state_callback, self.pub_callback, self.exception_callback)
result = self.parser.get_records(1)
# 0-69 contains an incomplete block (end of a sample)
# 1329-1332 there are 3 extra \n's between sio blocks
# 2294-2363, and 4092-4161 contains an error text string in between two sio blocks
# 4351-4927 has a bad AD then CT message where the size from the header does not line up with
# the final \x03
self.assert_result(result, [[637, 754, 1, 0],[6131, 6248, 1, 0]],
[[0,69],[637,754],[1329,1332], [2294,2363],
[4092,4161], [4351, 4927], [6131,6300]],
self.particle_b)
result = self.parser.get_records(2)
self.assertEqual(result[0], self.particle_c)
self.assertEqual(result[1], self.particle_d)
self.assert_state([], [[0,69],[1329,1332], [2294,2363],
[4092,4161], [4351, 4927], [6248,6300]])
stream_handle.close()
self.assertEqual(self.exception_callback_value, None)
def test_update(self):
"""
Test a file which has had a section of data replaced by 0s, as if a block of data has not been received yet,
then using the returned state make a new parser with the test data that has the 0s filled in
"""
state = {StateKey.UNPROCESSED_DATA:[[0, 6300]],
StateKey.IN_PROCESS_DATA:[], StateKey.METADATA_SENT: False, StateKey.FILE_SIZE: 9400}
# this file has a block of DO data replaced by 0s
stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_replaced.dat'))
self.parser = DostadParser(self.config, state, stream_handle,
self.state_callback, self.pub_callback, self.exception_callback)
result = self.parser.get_records(1)
# 0-69 contains an incomplete block (end of a sample)
# 390-570 is the zeroed block
# 1329-1332 there are 3 extra \n's between sio blocks
# 2294-2363, and 4092-4161 contains an error text string in between two sio blocks
# 4351-4927 has a bad AD then CT message where the size from the header does not line up with
# the final \x03
self.assert_result(result, [[197,314,2,1], [637,754,1,0], [6131,6248,1,0]],
[[0,69], [197,314], [390,507], [637,754], [1329,1332], [2294,2363],
[4092,4161], [4351, 4927], [6131,6300]],
self.particle_metadata)
result = self.parser.get_records(1)
self.assert_result(result, [[637,754,1,0], [6131,6248,1,0]],
[[0,69], [390,507], [637,754], [1329,1332], [2294,2363],
[4092,4161], [4351, 4927], [6131,6300]],
self.particle_a)
result = self.parser.get_records(1)
self.assert_result(result, [[6131,6248,1,0]],
[[0,69], [390,507], [1329,1332], [2294,2363],
[4092,4161], [4351, 4927], [6131,6300]],
self.particle_c)
stream_handle.close()
next_state = self.parser._state
# this file has the block of data that was missing in the previous file
stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_shorter.dat'))
self.parser = DostadParser(self.config, next_state, stream_handle,
self.state_callback, self.pub_callback, self.exception_callback)
# first get the old 'in process' records from 6131-6248
# Once those are done, the un processed data will be checked
result = self.parser.get_records(1)
self.assert_result(result, [],
[[0,69], [390,507], [1329,1332], [2294,2363],
[4092,4161], [4351, 4927], [6248,6300]],
self.particle_d)
# this should be the first of the newly filled in particles from 390-507
result = self.parser.get_records(1)
self.assert_result(result, [],
[[0,69], [1329,1332], [2294,2363],
[4092,4161], [4351, 4927], [6248,6300]],
self.particle_b)
stream_handle.close()
self.assertEqual(self.exception_callback_value, None)
def test_update(self):
"""
Test a file which has had a section of data replaced by 0s, as if a block of data has not been received yet,
then using the returned state make a new parser with the test data that has the 0s filled in
"""
self.state = {
StateKey.UNPROCESSED_DATA: [[0, 6300]],
StateKey.IN_PROCESS_DATA: [],
StateKey.TIMESTAMP: 0.0
}
# this file has a block of FL data replaced by 0s
self.stream_handle = open(
os.path.join(RESOURCE_PATH, 'node59p1_replaced.dat'))
self.parser = DostadParser(
self.config, self.state, self.stream_handle, self.state_callback,
self.pub_callback) # last one is the link to the data source
result = self.parser.get_records(1)
self.assert_result(
result, [[637, 754, 1, 0, 1], [6150, 6267, 1, 0, 1]],
[[0, 69], [390, 507], [637, 754], [944, 2370], [2560, 2947],
[3137, 4173], [4363, 5437], [5683, 6072], [6150, 6300]],
self.timestamp4, self.particle_a)
result = self.parser.get_records(1)
self.assert_result(
result, [[6150, 6267, 1, 0, 1]],
[[0, 69], [390, 507], [944, 2370], [2560, 2947], [3137, 4173],
[4363, 5437], [5683, 6072], [6150, 6300]], self.timestamp4,
self.particle_c_new)
self.stream_handle.close()
next_state = self.parser._state
# this file has the block of data that was missing in the previous file
self.stream_handle = open(
os.path.join(RESOURCE_PATH, 'node59p1_shorter.dat'))
self.parser = DostadParser(
self.config, next_state, self.stream_handle, self.state_callback,
self.pub_callback) # last one is the link to the data source
# first get the old 'in process' records
# Once those are done, the un processed data will be checked
result = self.parser.get_records(1)
self.assert_result(
result, [],
[[0, 69], [390, 507], [944, 2370], [2560, 2947], [3137, 4173],
[4363, 5437], [5683, 6072], [6267, 6300]], self.timestamp4,
self.particle_d)
# this should be the first of the newly filled in particles from
result = self.parser.get_records(1)
self.assert_result(result, [],
[[0, 69], [944, 2370], [2560, 2947], [3137, 4173],
[4363, 5437], [5683, 6072], [6267, 6300]],
self.timestamp2, self.particle_b_new)
self.stream_handle.close()
def test_simple(self):
"""
Read test data from the file and pull out data particles one at a time.
Assert that the results are those we expected.
"""
stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_shorter.dat'))
# NOTE: using the unprocessed data state of 0,6300 limits the file to reading
# just 6300 bytes, so even though the file is longer it only reads the first
# 6300
state = {StateKey.UNPROCESSED_DATA:[[0, 6300]],
StateKey.IN_PROCESS_DATA:[], StateKey.METADATA_SENT: False, StateKey.FILE_SIZE: 9400}
self.parser = DostadParser(self.config, state, stream_handle,
self.state_callback, self.pub_callback, self.exception_callback)
# the metadata particle and first particle come from the same block of data, first get
# the metadata particle (this is why there are 2 particles parsed in the first in process state)
# after that there are 3 more dosta samples parsed from 390-507, 637-754, and 6131-6248
# 0-69 contains an incomplete block (end of a sample)
# 1329-1332 there are 3 extra \n's between sio blocks
# 2294-2363, and 4092-4161 contains an error text string in between two sio blocks
# 4351-4927 has a bad AD then CT message where the size from the header does not line up with
# the final \x03
result = self.parser.get_records(1)
in_process = [[197,314,2,1], [390, 507, 1, 0], [637, 754, 1, 0], [6131, 6248, 1, 0]]
unprocessed = [[0,69], [197,314], [390,507], [637,754], [1329,1332], [2294,2363],
[4092,4161], [4351, 4927], [6131,6300]]
self.assert_result(result, in_process, unprocessed, self.particle_metadata)
self.assertEqual(self.parser._state[StateKey.METADATA_SENT], True)
# then get the first dosta data particle, this clears out the block from 197-314
result = self.parser.get_records(1)
in_process = [[390, 507, 1, 0], [637, 754, 1, 0], [6131, 6248, 1, 0]]
unprocessed = [[0,69], [390,507], [637,754], [1329,1332], [2294,2363],
[4092,4161], [4351, 4927], [6131,6300]]
self.assert_result(result, in_process, unprocessed, self.particle_a)
self.assertEqual(self.parser._state[StateKey.METADATA_SENT], True)
result = self.parser.get_records(1)
in_process = [[637, 754, 1, 0], [6131, 6248, 1, 0]]
unprocessed = [[0,69], [637,754], [1329,1332], [2294,2363],
[4092,4161], [4351, 4927], [6131,6300]]
self.assert_result(result, in_process, unprocessed, self.particle_b)
result = self.parser.get_records(1)
in_process = [[6131, 6248, 1, 0]]
unprocessed = [[0,69], [1329,1332], [2294,2363], [4092,4161], [4351, 4927], [6131,6300]]
self.assert_result(result, in_process, unprocessed, self.particle_c)
result = self.parser.get_records(1)
in_process = []
unprocessed = [[0,69], [1329,1332], [2294,2363],[4092,4161], [4351, 4927], [6248,6300]]
self.assert_result(result, in_process, unprocessed, self.particle_d)
stream_handle.close()
self.assertEqual(self.exception_callback_value, None)
def test_update(self):
"""
Test a file which has had a section of data replaced by 0s, as if a block of data has not been received yet,
then using the returned state make a new parser with the test data that has the 0s filled in
"""
self.state = {StateKey.UNPROCESSED_DATA:[[0, 6300]],
StateKey.IN_PROCESS_DATA:[]}
# this file has a block of DO data replaced by 0s
self.stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_replaced.dat'))
self.parser = DostadParser(self.config, self.state, self.stream_handle,
self.state_callback, self.pub_callback, self.exception_callback)
result = self.parser.get_records(1)
self.assert_result(result, [[637,754,1,0], [6131,6248,1,0]],
[[0,69], [390,507], [637,754], [1329,1332], [2294,2363],
[4092,4161], [4351, 4927], [6131,6300]],
self.particle_a)
result = self.parser.get_records(1)
self.assert_result(result, [[6131,6248,1,0]],
[[0,69], [390,507], [1329,1332], [2294,2363],
[4092,4161], [4351, 4927], [6131,6300]],
self.particle_c)
self.stream_handle.close()
next_state = self.parser._state
# this file has the block of data that was missing in the previous file
self.stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_shorter.dat'))
self.parser = DostadParser(self.config, next_state, self.stream_handle,
self.state_callback, self.pub_callback, self.exception_callback)
# first get the old 'in process' records
# Once those are done, the un processed data will be checked
result = self.parser.get_records(1)
self.assert_result(result, [],
[[0,69], [390,507], [1329,1332], [2294,2363],
[4092,4161], [4351, 4927], [6248,6300]],
self.particle_d)
# this should be the first of the newly filled in particles from
result = self.parser.get_records(1)
self.assert_result(result, [],
[[0,69], [1329,1332], [2294,2363],
[4092,4161], [4351, 4927], [6248,6300]],
self.particle_b)
self.stream_handle.close()
self.assertEqual(self.exception_callback_value, None)
def test_set_state(self):
"""
test changing the state after initializing
"""
self.state = {StateKey.UNPROCESSED_DATA:[[0, 1000]], StateKey.IN_PROCESS_DATA:[],
StateKey.TIMESTAMP:0.0}
new_state = {StateKey.UNPROCESSED_DATA:[[0,69],[944,6300]],
StateKey.IN_PROCESS_DATA:[],
StateKey.TIMESTAMP:self.timestamp2}
self.stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_shorter.dat'))
self.parser = DostadParser(self.config, self.state, self.stream_handle,
self.state_callback, self.pub_callback) # last one is the link to the data source
# there should only be 3 records, make sure we stop there
result = self.parser.get_records(3)
self.assert_state([],
[[0,69],[944,1000]],
self.timestamp3)
result = self.parser.get_records(1)
self.assertEqual(result, [])
self.parser.set_state(new_state)
result = self.parser.get_records(1)
self.stream_handle.close()
self.assert_result(result, [],
[[0,69],[944, 2370], [2560, 2947], [3137, 4173],
[4363, 5437], [5683, 6072], [6267, 6300]],
self.timestamp4, self.particle_d)
def test_in_process_start(self):
"""
test starting a parser with a state in the middle of processing
"""
new_state = {StateKey.IN_PROCESS_DATA:[[390, 507, 1, 0, 0], [637, 754, 1, 0, 0]],
StateKey.UNPROCESSED_DATA:[[0,69], [390,507], [637,754], [944,6300]],
StateKey.TIMESTAMP:self.timestamp3}
self.stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_shorter.dat'))
self.parser = DostadParser(self.config, new_state, self.stream_handle,
self.state_callback, self.pub_callback) # last one is the link to the data source
result = self.parser.get_records(1)
# even though the state says this particle is not a new sequence, since it is the
# first after setting the state it will be new
self.assert_result(result, [[637, 754, 1, 0, 0]],
[[0,69],[637,754],[944,6300]],
self.timestamp2, self.particle_b_new)
result = self.parser.get_records(2)
self.assertEqual(result[0], self.particle_c)
self.assertEqual(result[1], self.particle_d)
self.assert_state([],
[[0,69],[944, 2370], [2560, 2947], [3137, 4173],
[4363, 5437], [5683, 6072], [6267, 6300]],
self.timestamp4)
self.assertEqual(self.publish_callback_value[-1], self.particle_d)
def test_set_state(self):
"""
test changing the state after initializing
"""
self.state = {StateKey.UNPROCESSED_DATA:[[0, 1000]], StateKey.IN_PROCESS_DATA:[]}
new_state = {StateKey.UNPROCESSED_DATA:[[0,69],[944,6300]],
StateKey.IN_PROCESS_DATA:[]}
self.stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_shorter.dat'))
self.parser = DostadParser(self.config, self.state, self.stream_handle,
self.state_callback, self.pub_callback, self.exception_callback)
# there should only be 3 records, make sure we stop there
result = self.parser.get_records(3)
self.assert_state([],
[[0,69],[944,1000]])
result = self.parser.get_records(1)
self.assertEqual(result, [])
self.parser.set_state(new_state)
result = self.parser.get_records(1)
self.stream_handle.close()
self.assert_result(result, [],
[[0,69],[1329,1332], [2294,2363], [4092,4161], [4351, 4927],
[6248,6300]],
self.particle_d)
self.assertEqual(self.exception_callback_value, None)
def test_in_process_start(self):
"""
test starting a parser with a state in the middle of processing
"""
new_state = {StateKey.IN_PROCESS_DATA:[[390, 507, 1, 0], [637, 754, 1, 0]],
StateKey.UNPROCESSED_DATA:[[0,69], [390,507], [637,754], [944,6300]]}
self.stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_shorter.dat'))
self.parser = DostadParser(self.config, new_state, self.stream_handle,
self.state_callback, self.pub_callback, self.exception_callback)
result = self.parser.get_records(1)
# even though the state says this particle is not a new sequence, since it is the
# first after setting the state it will be new
self.assert_result(result, [[637, 754, 1, 0]],
[[0,69],[637,754],[944,6300]],
self.particle_b)
result = self.parser.get_records(2)
self.assertEqual(result[0], self.particle_c)
self.assertEqual(result[1], self.particle_d)
self.assert_state([],
[[0,69], [1329,1332], [2294,2363], [4092,4161], [4351, 4927],
[6248,6300]])
self.assertEqual(self.publish_callback_value[-1], self.particle_d)
self.assertEqual(self.exception_callback_value, None)
def test_get_many(self):
"""
Read test data from the file and pull out multiple data particles at one time.
Assert that the results are those we expected.
"""
state = {
StateKey.UNPROCESSED_DATA: [[0, 1000]],
StateKey.IN_PROCESS_DATA: [],
StateKey.METADATA_SENT: False,
StateKey.FILE_SIZE: 9400
}
stream_handle = open(
os.path.join(RESOURCE_PATH, 'node59p1_shorter.dat'))
self.parser = DostadParser(
self.config.get(DataTypeKey.DOSTA_ABCDJM_SIO_TELEMETERED), state,
stream_handle, self.state_callback, self.pub_callback,
self.exception_callback)
result = self.parser.get_records(4)
stream_handle.close()
self.assertEqual(result, [
self.particle_metadata, self.particle_a, self.particle_b,
self.particle_c
])
# 0-69 contains an incomplete block (end of a sample)
self.assert_state([], [[0, 69], [944, 1000]])
self.assertEqual(self.publish_callback_value[0],
self.particle_metadata)
self.assertEqual(self.publish_callback_value[1], self.particle_a)
self.assertEqual(self.publish_callback_value[2], self.particle_b)
self.assertEqual(self.publish_callback_value[3], self.particle_c)
self.assertEqual(self.exception_callback_value, None)
def test_update(self):
"""
Test a file which has had a section of data replaced by 0s, as if a block of data has not been received yet,
then using the returned state make a new parser with the test data that has the 0s filled in
"""
self.state = {StateKey.UNPROCESSED_DATA:[[0, 6300]],
StateKey.IN_PROCESS_DATA:[], StateKey.TIMESTAMP:0.0}
# this file has a block of FL data replaced by 0s
self.stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_replaced.dat'))
self.parser = DostadParser(self.config, self.state, self.stream_handle,
self.state_callback, self.pub_callback) # last one is the link to the data source
result = self.parser.get_records(1)
self.assert_result(result, [[637,754,1,0,1], [6150,6267,1,0,1]],
[[0,69], [390,507], [637,754], [944, 2370], [2560, 2947],
[3137, 4173], [4363, 5437], [5683, 6072], [6150, 6300]],
self.timestamp4, self.particle_a)
result = self.parser.get_records(1)
self.assert_result(result, [[6150,6267,1,0,1]],
[[0,69], [390,507], [944, 2370], [2560, 2947],
[3137, 4173], [4363, 5437], [5683, 6072], [6150, 6300]],
self.timestamp4, self.particle_c_new)
self.stream_handle.close()
next_state = self.parser._state
# this file has the block of data that was missing in the previous file
self.stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_shorter.dat'))
self.parser = DostadParser(self.config, next_state, self.stream_handle,
self.state_callback, self.pub_callback) # last one is the link to the data source
# first get the old 'in process' records
# Once those are done, the un processed data will be checked
result = self.parser.get_records(1)
self.assert_result(result, [],
[[0,69], [390,507], [944, 2370], [2560, 2947], [3137, 4173],
[4363, 5437], [5683, 6072], [6267, 6300]],
self.timestamp4, self.particle_d)
# this should be the first of the newly filled in particles from
result = self.parser.get_records(1)
self.assert_result(result, [],
[[0,69], [944, 2370], [2560, 2947], [3137, 4173], [4363, 5437],
[5683, 6072], [6267, 6300]],
self.timestamp2, self.particle_b_new)
self.stream_handle.close()
def _build_parser(self, parser_state, stream_in, data_key=None):
"""
Build and return the parser
"""
config = self._parser_config.get(data_key)
#
# If the key is DOSTA_ABCDJM_SIO_RECOVERED, build the WFP parser.
#
if data_key == DataTypeKey.DOSTA_ABCDJM_SIO_RECOVERED:
config.update({
DataSetDriverConfigKeys.PARTICLE_MODULE:
'mi.dataset.parser.dostad',
DataSetDriverConfigKeys.PARTICLE_CLASS: None,
DataSetDriverConfigKeys.PARTICLE_CLASSES_DICT: {
METADATA_PARTICLE_CLASS_KEY:
DostadParserRecoveredMetadataDataParticle,
DATA_PARTICLE_CLASS_KEY: DostadParserRecoveredDataParticle
}
})
parser = DostadRecoveredParser(
config, parser_state, stream_in,
lambda state, ingested: self._save_parser_state(
state, data_key, ingested), self._data_callback,
self._sample_exception_callback)
return parser
#
# If the key is DOSTA_ABCDJM_SIO_TELEMETERED, build the WFP SIO Mule parser.
#
elif data_key == DataTypeKey.DOSTA_ABCDJM_SIO_TELEMETERED:
config.update({
DataSetDriverConfigKeys.PARTICLE_MODULE:
'mi.dataset.parser.dostad',
DataSetDriverConfigKeys.PARTICLE_CLASS: None,
DataSetDriverConfigKeys.PARTICLE_CLASSES_DICT: {
METADATA_PARTICLE_CLASS_KEY:
DostadParserTelemeteredMetadataDataParticle,
DATA_PARTICLE_CLASS_KEY:
DostadParserTelemeteredDataParticle
}
})
parser = DostadParser(
config, parser_state, stream_in,
lambda state: self._save_parser_state(state, data_key),
self._data_callback, self._sample_exception_callback)
return parser
#
# If the key is one that we're not expecting, don't build any parser.
#
else:
raise ConfigurationException(
"Invalid data_key supplied to build parser")
def test_simple(self):
"""
Read test data from the file and pull out data particles one at a time.
Assert that the results are those we expected.
"""
self.stream_handle = open(
os.path.join(RESOURCE_PATH, 'node59p1_shorter.dat'))
# NOTE: using the unprocessed data state of 0,6300 limits the file to reading
# just 6300 bytes, so even though the file is longer it only reads the first
# 6300
self.state = {
StateKey.UNPROCESSED_DATA: [[0, 6300]],
StateKey.IN_PROCESS_DATA: [],
StateKey.TIMESTAMP: 0.0
}
self.parser = DostadParser(self.config, self.state, self.stream_handle,
self.state_callback, self.pub_callback)
result = self.parser.get_records(1)
self.assert_result(
result,
[[390, 507, 1, 0, 0], [637, 754, 1, 0, 0], [6150, 6267, 1, 0, 1]],
[[0, 69], [390, 507], [637, 754], [944, 2370], [2560, 2947],
[3137, 4173], [4363, 5437], [5683, 6072], [6150, 6300]],
self.timestamp4, self.particle_a)
result = self.parser.get_records(1)
self.assert_result(
result, [[637, 754, 1, 0, 0], [6150, 6267, 1, 0, 1]],
[[0, 69], [637, 754], [944, 2370], [2560, 2947], [3137, 4173],
[4363, 5437], [5683, 6072], [6150, 6300]], self.timestamp4,
self.particle_b)
result = self.parser.get_records(1)
self.assert_result(result, [[6150, 6267, 1, 0, 1]],
[[0, 69], [944, 2370], [2560, 2947], [3137, 4173],
[4363, 5437], [5683, 6072], [6150, 6300]],
self.timestamp4, self.particle_c)
result = self.parser.get_records(1)
self.assert_result(result, [],
[[0, 69], [944, 2370], [2560, 2947], [3137, 4173],
[4363, 5437], [5683, 6072], [6267, 6300]],
self.timestamp4, self.particle_d)
self.stream_handle.close()
def test_mid_state_start(self):
"""
test starting a parser with a state in the middle of processing
"""
new_state = {
StateKey.IN_PROCESS_DATA: [],
StateKey.UNPROCESSED_DATA: [[0, 69], [314, 1000]],
StateKey.TIMESTAMP: self.timestamp1
}
self.stream_handle = open(
os.path.join(RESOURCE_PATH, 'node59p1_shorter.dat'))
self.parser = DostadParser(
self.config, new_state, self.stream_handle, self.state_callback,
self.pub_callback) # last one is the link to the data source
result = self.parser.get_records(1)
self.assert_result(result, [
[637, 754, 1, 0, 0],
], [[0, 69], [637, 754], [944, 1000]], self.timestamp3,
self.particle_b_new)
result = self.parser.get_records(1)
self.assert_result(result, [], [[0, 69], [944, 1000]], self.timestamp3,
self.particle_c)
self.stream_handle.close()
def test_set_state(self):
"""
test changing the state after initializing
"""
state = {StateKey.UNPROCESSED_DATA:[[0, 1000]], StateKey.IN_PROCESS_DATA:[],
StateKey.METADATA_SENT: False,
StateKey.FILE_SIZE: 9400}
new_state = {StateKey.UNPROCESSED_DATA:[[0,69],[944,6300]],
StateKey.IN_PROCESS_DATA:[], StateKey.METADATA_SENT: True,
StateKey.FILE_SIZE: 9400}
stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_shorter.dat'))
self.parser = DostadParser(self.config, state, stream_handle,
self.state_callback, self.pub_callback, self.exception_callback)
# there should only be 4 records, make sure we stop there
result = self.parser.get_records(4)
# 0-69 contains an incomplete block (end of a sample)
self.assert_state([],
[[0,69],[944,1000]])
result = self.parser.get_records(1)
self.assertEqual(result, [])
self.parser.set_state(new_state)
result = self.parser.get_records(1)
stream_handle.close()
# 0-69 contains an incomplete block (end of a sample)
# 1329-1332 there are 3 extra \n's between sio blocks
# 2294-2363, and 4092-4161 contains an error text string in between two sio blocks
# 4351-4927 has a bad AD then CT message where the size from the header does not line up with
# the final \x03
self.assert_result(result, [],
[[0,69],[1329,1332], [2294,2363], [4092,4161], [4351, 4927],
[6248,6300]],
self.particle_d)
self.assertEqual(self.exception_callback_value, None)
def _build_parser(self, parser_state, infile):
"""
Build and return the parser
"""
config = self._parser_config
# Fill in blanks with particle info
config.update({
'particle_module': 'mi.dataset.parser.dostad',
'particle_class': 'DostadParserDataParticle'
})
log.debug("My Config: %s", config)
self._parser = DostadParser(config, parser_state, infile,
self._save_parser_state,
self._data_callback)
return self._parser
def test_get_many(self):
"""
Read test data from the file and pull out multiple data particles at one time.
Assert that the results are those we expected.
"""
self.state = {
StateKey.UNPROCESSED_DATA: [[0, 1000]],
StateKey.IN_PROCESS_DATA: [],
StateKey.TIMESTAMP: 0.0
}
self.stream_handle = open(
os.path.join(RESOURCE_PATH, 'node59p1_shorter.dat'))
self.parser = DostadParser(
self.config, self.state, self.stream_handle, self.state_callback,
self.pub_callback) # last one is the link to the data source
result = self.parser.get_records(3)
self.stream_handle.close()
self.assertEqual(result,
[self.particle_a, self.particle_b, self.particle_c])
self.assert_state([], [[0, 69], [944, 1000]], self.timestamp3)
self.assertEqual(self.publish_callback_value[0], self.particle_a)
self.assertEqual(self.publish_callback_value[1], self.particle_b)
self.assertEqual(self.publish_callback_value[2], self.particle_c)
def test_long_stream(self):
self.stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_shorter.dat'))
self.parser = DostadParser(self.config, None, self.stream_handle,
self.state_callback, self.pub_callback, self.exception_callback)
result = self.parser.get_records(6)
self.stream_handle.close()
self.assertEqual(result[0], self.particle_a)
self.assertEqual(result[1], self.particle_b)
self.assertEqual(result[2], self.particle_c)
self.assertEqual(result[-3], self.particle_d)
self.assertEqual(result[-2], self.particle_e)
self.assertEqual(result[-1], self.particle_f)
self.assert_state([],
[[0, 69], [1329,1332], [2294,2363], [4092,4161], [4351, 4927], [9020, 9374]])
self.assertEqual(self.publish_callback_value[-2], self.particle_e)
self.assertEqual(self.publish_callback_value[-1], self.particle_f)
self.assertEqual(self.exception_callback_value, None)
def test_mid_state_start(self):
"""
test starting a parser with a state in the middle of processing
"""
new_state = {StateKey.IN_PROCESS_DATA:[],
StateKey.UNPROCESSED_DATA:[[0,69], [314,1000]]}
self.stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_shorter.dat'))
self.parser = DostadParser(self.config, new_state, self.stream_handle,
self.state_callback, self.pub_callback, self.exception_callback)
result = self.parser.get_records(1)
self.assert_result(result, [[637, 754, 1, 0],],
[[0,69],[637,754],[944,1000]],
self.particle_b)
result = self.parser.get_records(1)
self.assert_result(result, [],
[[0,69],[944,1000]],
self.particle_c)
self.stream_handle.close()
self.assertEqual(self.exception_callback_value, None)
def test_mid_state_start(self):
"""
test starting a parser with a state in the middle of processing
"""
new_state = {StateKey.IN_PROCESS_DATA:[],
StateKey.UNPROCESSED_DATA:[[0,69], [314,1000]],
StateKey.TIMESTAMP:self.timestamp1}
self.stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_shorter.dat'))
self.parser = DostadParser(self.config, new_state, self.stream_handle,
self.state_callback, self.pub_callback) # last one is the link to the data source
result = self.parser.get_records(1)
self.assert_result(result, [[637, 754, 1, 0, 0],],
[[0,69],[637,754],[944,1000]],
self.timestamp3, self.particle_b_new)
result = self.parser.get_records(1)
self.assert_result(result, [],
[[0,69],[944,1000]],
self.timestamp3, self.particle_c)
self.stream_handle.close()
def test_get_many(self):
"""
Read test data from the file and pull out multiple data particles at one time.
Assert that the results are those we expected.
"""
self.state = {StateKey.UNPROCESSED_DATA:[[0, 1000]],
StateKey.IN_PROCESS_DATA:[], StateKey.TIMESTAMP:0.0}
self.stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_shorter.dat'))
self.parser = DostadParser(self.config, self.state, self.stream_handle,
self.state_callback, self.pub_callback) # last one is the link to the data source
result = self.parser.get_records(3)
self.stream_handle.close()
self.assertEqual(result,
[self.particle_a, self.particle_b, self.particle_c])
self.assert_state([],
[[0,69],[944,1000]],
self.timestamp3)
self.assertEqual(self.publish_callback_value[0], self.particle_a)
self.assertEqual(self.publish_callback_value[1], self.particle_b)
self.assertEqual(self.publish_callback_value[2], self.particle_c)
def test_simple(self):
"""
Read test data from the file and pull out data particles one at a time.
Assert that the results are those we expected.
"""
self.stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_shorter.dat'))
# NOTE: using the unprocessed data state of 0,6300 limits the file to reading
# just 6300 bytes, so even though the file is longer it only reads the first
# 6300
self.state = {StateKey.UNPROCESSED_DATA:[[0, 6300]],
StateKey.IN_PROCESS_DATA:[]}
self.parser = DostadParser(self.config, self.state, self.stream_handle,
self.state_callback, self.pub_callback, self.exception_callback)
result = self.parser.get_records(1)
self.assert_result(result,
[[390, 507, 1, 0], [637, 754, 1, 0], [6131, 6248, 1, 0]],
[[0,69], [390,507], [637,754], [1329,1332], [2294,2363],
[4092,4161], [4351, 4927], [6131,6300]],
self.particle_a)
result = self.parser.get_records(1)
self.assert_result(result, [[637, 754, 1, 0], [6131, 6248, 1, 0]],
[[0,69], [637,754], [1329,1332], [2294,2363],
[4092,4161], [4351, 4927], [6131,6300]],
self.particle_b)
result = self.parser.get_records(1)
self.assert_result(result, [[6131, 6248, 1, 0]],
[[0,69], [1329,1332], [2294,2363],
[4092,4161], [4351, 4927], [6131,6300]],
self.particle_c)
result = self.parser.get_records(1)
self.assert_result(result, [],
[[0,69], [1329,1332], [2294,2363],
[4092,4161], [4351, 4927], [6248,6300]],
self.particle_d)
self.stream_handle.close()
self.assertEqual(self.exception_callback_value, None)
def _build_telemetered_parser(self, parser_state, stream_in):
"""
Build and return the telemetered parser
@param parser_state starting parser state to pass to parser
@param stream_in Handle of open file to pass to parser
"""
config = self._parser_config[DataSourceKey.DOSTA_ABCDJM_SIO_TELEMETERED]
# Fill in blanks with particle info
config.update({
'particle_module': 'mi.dataset.parser.dostad',
'particle_class': ['DostadParserDataParticle',
'DostadMetadataDataParticle']
})
log.debug("My Config: %s", config)
parser = DostadParser(
config,
parser_state,
stream_in,
lambda state: self._save_parser_state(state, DataSourceKey.DOSTA_ABCDJM_SIO_TELEMETERED),
self._data_callback,
self._sample_exception_callback
)
return parser
def test_simple(self):
"""
Read test data from the file and pull out data particles one at a time.
Assert that the results are those we expected.
"""
self.stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_shorter.dat'))
# NOTE: using the unprocessed data state of 0,6300 limits the file to reading
# just 6300 bytes, so even though the file is longer it only reads the first
# 6300
self.state = {StateKey.UNPROCESSED_DATA:[[0, 6300]],
StateKey.IN_PROCESS_DATA:[], StateKey.TIMESTAMP:0.0}
self.parser = DostadParser(self.config, self.state, self.stream_handle,
self.state_callback, self.pub_callback)
result = self.parser.get_records(1)
self.assert_result(result,
[[390, 507, 1, 0, 0], [637, 754, 1, 0, 0], [6150, 6267, 1, 0, 1]],
[[0,69], [390,507], [637,754], [944, 2370], [2560, 2947],
[3137, 4173], [4363, 5437], [5683, 6072], [6150, 6300]],
self.timestamp4, self.particle_a)
result = self.parser.get_records(1)
self.assert_result(result, [[637, 754, 1, 0, 0], [6150, 6267, 1, 0, 1]],
[[0,69], [637,754], [944, 2370], [2560, 2947],
[3137, 4173], [4363, 5437], [5683, 6072], [6150, 6300]],
self.timestamp4, self.particle_b)
result = self.parser.get_records(1)
self.assert_result(result, [[6150, 6267, 1, 0, 1]],
[[0,69], [944, 2370], [2560, 2947], [3137, 4173],
[4363, 5437], [5683, 6072], [6150, 6300]],
self.timestamp4, self.particle_c)
result = self.parser.get_records(1)
self.assert_result(result, [],
[[0,69], [944, 2370], [2560, 2947], [3137, 4173],
[4363, 5437], [5683, 6072], [6267, 6300]],
self.timestamp4, self.particle_d)
self.stream_handle.close()
def test_get_many(self):
"""
Read test data from the file and pull out multiple data particles at one time.
Assert that the results are those we expected.
"""
state = {StateKey.UNPROCESSED_DATA:[[0, 1000]],
StateKey.IN_PROCESS_DATA:[], StateKey.METADATA_SENT: False, StateKey.FILE_SIZE: 9400}
stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_shorter.dat'))
self.parser = DostadParser(self.config, state, stream_handle,
self.state_callback, self.pub_callback, self.exception_callback)
result = self.parser.get_records(4)
stream_handle.close()
self.assertEqual(result,
[self.particle_metadata, self.particle_a, self.particle_b, self.particle_c])
# 0-69 contains an incomplete block (end of a sample)
self.assert_state([],
[[0,69],[944,1000]])
self.assertEqual(self.publish_callback_value[0], self.particle_metadata)
self.assertEqual(self.publish_callback_value[1], self.particle_a)
self.assertEqual(self.publish_callback_value[2], self.particle_b)
self.assertEqual(self.publish_callback_value[3], self.particle_c)
self.assertEqual(self.exception_callback_value, None)
def test_long_stream(self):
self.stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_shorter.dat'))
data = self.stream_handle.read()
data_len = len(data)
self.stream_handle.seek(0)
self.state = {StateKey.UNPROCESSED_DATA:[[0, data_len]],
StateKey.IN_PROCESS_DATA:[], StateKey.TIMESTAMP:0.0}
self.parser = DostadParser(self.config, self.state, self.stream_handle,
self.state_callback, self.pub_callback) # last one is the link to the data source
result = self.parser.get_records(6)
self.stream_handle.close()
self.assertEqual(result[0], self.particle_a)
self.assertEqual(result[1], self.particle_b)
self.assertEqual(result[2], self.particle_c)
self.assertEqual(result[-3], self.particle_d)
self.assertEqual(result[-2], self.particle_e)
self.assertEqual(result[-1], self.particle_f)
self.assert_state([],
[[0, 69], [944, 2370], [2560, 2947], [3137, 4173], [4363, 5437], [5683, 6072], [8273, 9400]],
self.timestamp6)
self.assertEqual(self.publish_callback_value[-2], self.particle_e)
self.assertEqual(self.publish_callback_value[-1], self.particle_f)
class DostadParserUnitTestCase(ParserUnitTestCase):
def state_callback(self, state):
""" Call back method to watch what comes in via the position callback """
self.state_callback_value = state
def pub_callback(self, pub):
""" Call back method to watch what comes in via the publish callback """
self.publish_callback_value = pub
def exception_callback(self, exception):
""" Call back method to watch what comes in via the exception callback """
self.exception_callback_value = exception
def setUp(self):
ParserUnitTestCase.setUp(self)
self.config = {
DataSetDriverConfigKeys.PARTICLE_MODULE: 'mi.dataset.parser.dostad',
DataSetDriverConfigKeys.PARTICLE_CLASS: 'DostadParserDataParticle'
}
# first DO tag
self.particle_a = DostadParserDataParticle('51EF0E75\xff\x11\x25\x11' \
'4831\t128\t302.636\t98.918\t16.355\t30.771\t' \
'30.771\t36.199\t5.429\t1309.9\t1175.7\t299.6\x0d\x0a')
self.particle_b = DostadParserDataParticle('51EF1901\xff\x11\x25\x11' \
'4831\t128\t317.073\t98.515\t14.004\t31.299\t31.299' \
'\t36.647\t5.349\t1338.5\t1212.4\t375.8\x0d\x0a')
self.particle_c = DostadParserDataParticle('51EF6D61\xff\x11\x25\x11' \
'4831\t128\t326.544\t96.726\t11.873\t31.965\t31.965' \
'\t37.134\t5.169\t1370.7\t1245.9\t444.4\x0d\x0a')
self.particle_d = DostadParserDataParticle('51EFC1C2\xff\x11\x25\x11' \
'4831\t128\t332.060\t99.617\t12.432\t31.490\t31.490' \
'\t36.812\t5.323\t1026.0\t1100.0\t426.5\x0d\x0a')
self.particle_e = DostadParserDataParticle('51F01622\xff\x11\x25\x11' \
'4831\t128\t354.515\t101.592\t10.440\t31.666\t31.666' \
'\t36.930\t5.263\t1020.1\t1123.0\t490.0\x0d\x0a')
self.particle_f = DostadParserDataParticle('51F06A82\xff\x11\x25\x11' \
'4831\t128\t337.540\t100.172\t11.955\t31.521\t31.521' \
'\t36.805\t5.284\t983.8\t1092.6\t441.7\x0d\x0a')
self.state_callback_value = None
self.publish_callback_value = None
self.exception_callback_value = None
def assert_result(self, result, in_process_data, unprocessed_data, particle):
self.assertEqual(result, [particle])
self.assert_state(in_process_data, unprocessed_data)
self.assert_(isinstance(self.publish_callback_value, list))
self.assertEqual(self.publish_callback_value[0], particle)
def assert_state(self, in_process_data, unprocessed_data):
self.assertEqual(self.parser._state[StateKey.IN_PROCESS_DATA], in_process_data)
self.assertEqual(self.parser._state[StateKey.UNPROCESSED_DATA], unprocessed_data)
self.assertEqual(self.state_callback_value[StateKey.IN_PROCESS_DATA], in_process_data)
self.assertEqual(self.state_callback_value[StateKey.UNPROCESSED_DATA], unprocessed_data)
def test_simple(self):
"""
Read test data from the file and pull out data particles one at a time.
Assert that the results are those we expected.
"""
self.stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_shorter.dat'))
# NOTE: using the unprocessed data state of 0,6300 limits the file to reading
# just 6300 bytes, so even though the file is longer it only reads the first
# 6300
self.state = {StateKey.UNPROCESSED_DATA:[[0, 6300]],
StateKey.IN_PROCESS_DATA:[]}
self.parser = DostadParser(self.config, self.state, self.stream_handle,
self.state_callback, self.pub_callback, self.exception_callback)
result = self.parser.get_records(1)
self.assert_result(result,
[[390, 507, 1, 0], [637, 754, 1, 0], [6131, 6248, 1, 0]],
[[0,69], [390,507], [637,754], [1329,1332], [2294,2363],
[4092,4161], [4351, 4927], [6131,6300]],
self.particle_a)
result = self.parser.get_records(1)
self.assert_result(result, [[637, 754, 1, 0], [6131, 6248, 1, 0]],
[[0,69], [637,754], [1329,1332], [2294,2363],
[4092,4161], [4351, 4927], [6131,6300]],
self.particle_b)
result = self.parser.get_records(1)
self.assert_result(result, [[6131, 6248, 1, 0]],
[[0,69], [1329,1332], [2294,2363],
[4092,4161], [4351, 4927], [6131,6300]],
self.particle_c)
result = self.parser.get_records(1)
self.assert_result(result, [],
[[0,69], [1329,1332], [2294,2363],
[4092,4161], [4351, 4927], [6248,6300]],
self.particle_d)
self.stream_handle.close()
self.assertEqual(self.exception_callback_value, None)
def test_get_many(self):
"""
Read test data from the file and pull out multiple data particles at one time.
Assert that the results are those we expected.
"""
self.state = {StateKey.UNPROCESSED_DATA:[[0, 1000]],
StateKey.IN_PROCESS_DATA:[]}
self.stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_shorter.dat'))
self.parser = DostadParser(self.config, self.state, self.stream_handle,
self.state_callback, self.pub_callback, self.exception_callback)
result = self.parser.get_records(3)
self.stream_handle.close()
self.assertEqual(result,
[self.particle_a, self.particle_b, self.particle_c])
self.assert_state([],
[[0,69],[944,1000]])
self.assertEqual(self.publish_callback_value[0], self.particle_a)
self.assertEqual(self.publish_callback_value[1], self.particle_b)
self.assertEqual(self.publish_callback_value[2], self.particle_c)
self.assertEqual(self.exception_callback_value, None)
def test_long_stream(self):
self.stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_shorter.dat'))
self.parser = DostadParser(self.config, None, self.stream_handle,
self.state_callback, self.pub_callback, self.exception_callback)
result = self.parser.get_records(6)
self.stream_handle.close()
self.assertEqual(result[0], self.particle_a)
self.assertEqual(result[1], self.particle_b)
self.assertEqual(result[2], self.particle_c)
self.assertEqual(result[-3], self.particle_d)
self.assertEqual(result[-2], self.particle_e)
self.assertEqual(result[-1], self.particle_f)
self.assert_state([],
[[0, 69], [1329,1332], [2294,2363], [4092,4161], [4351, 4927], [9020, 9374]])
self.assertEqual(self.publish_callback_value[-2], self.particle_e)
self.assertEqual(self.publish_callback_value[-1], self.particle_f)
self.assertEqual(self.exception_callback_value, None)
def test_mid_state_start(self):
"""
test starting a parser with a state in the middle of processing
"""
new_state = {StateKey.IN_PROCESS_DATA:[],
StateKey.UNPROCESSED_DATA:[[0,69], [314,1000]]}
self.stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_shorter.dat'))
self.parser = DostadParser(self.config, new_state, self.stream_handle,
self.state_callback, self.pub_callback, self.exception_callback)
result = self.parser.get_records(1)
self.assert_result(result, [[637, 754, 1, 0],],
[[0,69],[637,754],[944,1000]],
self.particle_b)
result = self.parser.get_records(1)
self.assert_result(result, [],
[[0,69],[944,1000]],
self.particle_c)
self.stream_handle.close()
self.assertEqual(self.exception_callback_value, None)
def test_in_process_start(self):
"""
test starting a parser with a state in the middle of processing
"""
new_state = {StateKey.IN_PROCESS_DATA:[[390, 507, 1, 0], [637, 754, 1, 0]],
StateKey.UNPROCESSED_DATA:[[0,69], [390,507], [637,754], [944,6300]]}
self.stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_shorter.dat'))
self.parser = DostadParser(self.config, new_state, self.stream_handle,
self.state_callback, self.pub_callback, self.exception_callback)
result = self.parser.get_records(1)
# even though the state says this particle is not a new sequence, since it is the
# first after setting the state it will be new
self.assert_result(result, [[637, 754, 1, 0]],
[[0,69],[637,754],[944,6300]],
self.particle_b)
result = self.parser.get_records(2)
self.assertEqual(result[0], self.particle_c)
self.assertEqual(result[1], self.particle_d)
self.assert_state([],
[[0,69], [1329,1332], [2294,2363], [4092,4161], [4351, 4927],
[6248,6300]])
self.assertEqual(self.publish_callback_value[-1], self.particle_d)
self.assertEqual(self.exception_callback_value, None)
def test_set_state(self):
"""
test changing the state after initializing
"""
self.state = {StateKey.UNPROCESSED_DATA:[[0, 1000]], StateKey.IN_PROCESS_DATA:[]}
new_state = {StateKey.UNPROCESSED_DATA:[[0,69],[944,6300]],
StateKey.IN_PROCESS_DATA:[]}
self.stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_shorter.dat'))
self.parser = DostadParser(self.config, self.state, self.stream_handle,
self.state_callback, self.pub_callback, self.exception_callback)
# there should only be 3 records, make sure we stop there
result = self.parser.get_records(3)
self.assert_state([],
[[0,69],[944,1000]])
result = self.parser.get_records(1)
self.assertEqual(result, [])
self.parser.set_state(new_state)
result = self.parser.get_records(1)
self.stream_handle.close()
self.assert_result(result, [],
[[0,69],[1329,1332], [2294,2363], [4092,4161], [4351, 4927],
[6248,6300]],
self.particle_d)
self.assertEqual(self.exception_callback_value, None)
def test_update(self):
"""
Test a file which has had a section of data replaced by 0s, as if a block of data has not been received yet,
then using the returned state make a new parser with the test data that has the 0s filled in
"""
self.state = {StateKey.UNPROCESSED_DATA:[[0, 6300]],
StateKey.IN_PROCESS_DATA:[]}
# this file has a block of DO data replaced by 0s
self.stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_replaced.dat'))
self.parser = DostadParser(self.config, self.state, self.stream_handle,
self.state_callback, self.pub_callback, self.exception_callback)
result = self.parser.get_records(1)
self.assert_result(result, [[637,754,1,0], [6131,6248,1,0]],
[[0,69], [390,507], [637,754], [1329,1332], [2294,2363],
[4092,4161], [4351, 4927], [6131,6300]],
self.particle_a)
result = self.parser.get_records(1)
self.assert_result(result, [[6131,6248,1,0]],
[[0,69], [390,507], [1329,1332], [2294,2363],
[4092,4161], [4351, 4927], [6131,6300]],
self.particle_c)
self.stream_handle.close()
next_state = self.parser._state
# this file has the block of data that was missing in the previous file
self.stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_shorter.dat'))
self.parser = DostadParser(self.config, next_state, self.stream_handle,
self.state_callback, self.pub_callback, self.exception_callback)
# first get the old 'in process' records
# Once those are done, the un processed data will be checked
result = self.parser.get_records(1)
self.assert_result(result, [],
[[0,69], [390,507], [1329,1332], [2294,2363],
[4092,4161], [4351, 4927], [6248,6300]],
self.particle_d)
# this should be the first of the newly filled in particles from
result = self.parser.get_records(1)
self.assert_result(result, [],
[[0,69], [1329,1332], [2294,2363],
[4092,4161], [4351, 4927], [6248,6300]],
self.particle_b)
self.stream_handle.close()
self.assertEqual(self.exception_callback_value, None)
class DostadParserUnitTestCase(ParserUnitTestCase):
def state_callback(self, state):
""" Call back method to watch what comes in via the position callback """
self.state_callback_value = state
def pub_callback(self, pub):
""" Call back method to watch what comes in via the publish callback """
self.publish_callback_value = pub
def exception_callback(self, exception):
""" Call back method to watch what comes in via the exception callback """
self.exception_callback_value = exception
def setUp(self):
ParserUnitTestCase.setUp(self)
self.config = {
DataSetDriverConfigKeys.PARTICLE_MODULE: 'mi.dataset.parser.dostad',
DataSetDriverConfigKeys.PARTICLE_CLASS: 'DostadParserDataParticle'
}
# first DO tag
self.particle_a = DostadParserDataParticle('51EF0E75\xff\x11\x25\x11' \
'4831\t128\t302.636\t98.918\t16.355\t30.771\t' \
'30.771\t36.199\t5.429\t1309.9\t1175.7\t299.6\x0d\x0a')
self.particle_metadata = DostadMetadataDataParticle('51EF0E75\xff\x11\x25\x11' \
'4831\t128\t302.636\t98.918\t16.355\t30.771\t' \
'30.771\t36.199\t5.429\t1309.9\t1175.7\t299.6\x0d\x0a')
self.particle_b = DostadParserDataParticle('51EF1901\xff\x11\x25\x11' \
'4831\t128\t317.073\t98.515\t14.004\t31.299\t31.299' \
'\t36.647\t5.349\t1338.5\t1212.4\t375.8\x0d\x0a')
self.particle_c = DostadParserDataParticle('51EF6D61\xff\x11\x25\x11' \
'4831\t128\t326.544\t96.726\t11.873\t31.965\t31.965' \
'\t37.134\t5.169\t1370.7\t1245.9\t444.4\x0d\x0a')
self.particle_d = DostadParserDataParticle('51EFC1C2\xff\x11\x25\x11' \
'4831\t128\t332.060\t99.617\t12.432\t31.490\t31.490' \
'\t36.812\t5.323\t1026.0\t1100.0\t426.5\x0d\x0a')
self.particle_e = DostadParserDataParticle('51F01622\xff\x11\x25\x11' \
'4831\t128\t354.515\t101.592\t10.440\t31.666\t31.666' \
'\t36.930\t5.263\t1020.1\t1123.0\t490.0\x0d\x0a')
self.particle_f = DostadParserDataParticle('51F06A82\xff\x11\x25\x11' \
'4831\t128\t337.540\t100.172\t11.955\t31.521\t31.521' \
'\t36.805\t5.284\t983.8\t1092.6\t441.7\x0d\x0a')
self.state_callback_value = None
self.publish_callback_value = None
self.exception_callback_value = None
def assert_result(self, result, in_process_data, unprocessed_data, particle):
self.assertEqual(result, [particle])
self.assert_state(in_process_data, unprocessed_data)
self.assert_(isinstance(self.publish_callback_value, list))
self.assertEqual(self.publish_callback_value[0], particle)
def assert_state(self, in_process_data, unprocessed_data):
self.assertEqual(self.parser._state[StateKey.IN_PROCESS_DATA], in_process_data)
self.assertEqual(self.parser._state[StateKey.UNPROCESSED_DATA], unprocessed_data)
self.assertEqual(self.state_callback_value[StateKey.IN_PROCESS_DATA], in_process_data)
self.assertEqual(self.state_callback_value[StateKey.UNPROCESSED_DATA], unprocessed_data)
def test_simple(self):
"""
Read test data from the file and pull out data particles one at a time.
Assert that the results are those we expected.
"""
stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_shorter.dat'))
# NOTE: using the unprocessed data state of 0,6300 limits the file to reading
# just 6300 bytes, so even though the file is longer it only reads the first
# 6300
state = {StateKey.UNPROCESSED_DATA:[[0, 6300]],
StateKey.IN_PROCESS_DATA:[], StateKey.METADATA_SENT: False, StateKey.FILE_SIZE: 9400}
self.parser = DostadParser(self.config, state, stream_handle,
self.state_callback, self.pub_callback, self.exception_callback)
# the metadata particle and first particle come from the same block of data, first get
# the metadata particle (this is why there are 2 particles parsed in the first in process state)
# after that there are 3 more dosta samples parsed from 390-507, 637-754, and 6131-6248
# 0-69 contains an incomplete block (end of a sample)
# 1329-1332 there are 3 extra \n's between sio blocks
# 2294-2363, and 4092-4161 contains an error text string in between two sio blocks
# 4351-4927 has a bad AD then CT message where the size from the header does not line up with
# the final \x03
result = self.parser.get_records(1)
in_process = [[197,314,2,1], [390, 507, 1, 0], [637, 754, 1, 0], [6131, 6248, 1, 0]]
unprocessed = [[0,69], [197,314], [390,507], [637,754], [1329,1332], [2294,2363],
[4092,4161], [4351, 4927], [6131,6300]]
self.assert_result(result, in_process, unprocessed, self.particle_metadata)
self.assertEqual(self.parser._state[StateKey.METADATA_SENT], True)
# then get the first dosta data particle, this clears out the block from 197-314
result = self.parser.get_records(1)
in_process = [[390, 507, 1, 0], [637, 754, 1, 0], [6131, 6248, 1, 0]]
unprocessed = [[0,69], [390,507], [637,754], [1329,1332], [2294,2363],
[4092,4161], [4351, 4927], [6131,6300]]
self.assert_result(result, in_process, unprocessed, self.particle_a)
self.assertEqual(self.parser._state[StateKey.METADATA_SENT], True)
result = self.parser.get_records(1)
in_process = [[637, 754, 1, 0], [6131, 6248, 1, 0]]
unprocessed = [[0,69], [637,754], [1329,1332], [2294,2363],
[4092,4161], [4351, 4927], [6131,6300]]
self.assert_result(result, in_process, unprocessed, self.particle_b)
result = self.parser.get_records(1)
in_process = [[6131, 6248, 1, 0]]
unprocessed = [[0,69], [1329,1332], [2294,2363], [4092,4161], [4351, 4927], [6131,6300]]
self.assert_result(result, in_process, unprocessed, self.particle_c)
result = self.parser.get_records(1)
in_process = []
unprocessed = [[0,69], [1329,1332], [2294,2363],[4092,4161], [4351, 4927], [6248,6300]]
self.assert_result(result, in_process, unprocessed, self.particle_d)
stream_handle.close()
self.assertEqual(self.exception_callback_value, None)
def test_get_many(self):
"""
Read test data from the file and pull out multiple data particles at one time.
Assert that the results are those we expected.
"""
state = {StateKey.UNPROCESSED_DATA:[[0, 1000]],
StateKey.IN_PROCESS_DATA:[], StateKey.METADATA_SENT: False, StateKey.FILE_SIZE: 9400}
stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_shorter.dat'))
self.parser = DostadParser(self.config, state, stream_handle,
self.state_callback, self.pub_callback, self.exception_callback)
result = self.parser.get_records(4)
stream_handle.close()
self.assertEqual(result,
[self.particle_metadata, self.particle_a, self.particle_b, self.particle_c])
# 0-69 contains an incomplete block (end of a sample)
self.assert_state([],
[[0,69],[944,1000]])
self.assertEqual(self.publish_callback_value[0], self.particle_metadata)
self.assertEqual(self.publish_callback_value[1], self.particle_a)
self.assertEqual(self.publish_callback_value[2], self.particle_b)
self.assertEqual(self.publish_callback_value[3], self.particle_c)
self.assertEqual(self.exception_callback_value, None)
def test_long_stream(self):
stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_shorter.dat'))
self.parser = DostadParser(self.config, None, stream_handle,
self.state_callback, self.pub_callback, self.exception_callback)
result = self.parser.get_records(7)
stream_handle.close()
self.assertEqual(result[0], self.particle_metadata)
self.assertEqual(result[1], self.particle_a)
self.assertEqual(result[2], self.particle_b)
self.assertEqual(result[3], self.particle_c)
self.assertEqual(result[4], self.particle_d)
self.assertEqual(result[5], self.particle_e)
self.assertEqual(result[6], self.particle_f)
# 0-69 contains an incomplete block (end of a sample)
# 1329-1332 there are 3 extra \n's between sio blocks
# 2294-2363, and 4092-4161 contains an error text string in between two sio blocks
# 4351-4927 has a bad AD then CT message where the size from the header does not line up with
# the final \x03
self.assert_state([],
[[0, 69], [1329,1332], [2294,2363], [4092,4161], [4351, 4927], [9020, 9374]])
self.assertEqual(self.publish_callback_value[5], self.particle_e)
self.assertEqual(self.publish_callback_value[6], self.particle_f)
self.assertEqual(self.exception_callback_value, None)
def test_mid_state_start(self):
"""
test starting a parser with a state in the middle of processing
"""
new_state = {StateKey.IN_PROCESS_DATA:[],
StateKey.UNPROCESSED_DATA:[[0,69], [314,6300]],
StateKey.METADATA_SENT: True,
StateKey.FILE_SIZE: 9400}
stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_shorter.dat'))
self.parser = DostadParser(self.config, new_state, stream_handle,
self.state_callback, self.pub_callback, self.exception_callback)
result = self.parser.get_records(1)
# 0-69 contains an incomplete block (end of a sample)
# 1329-1332 there are 3 extra \n's between sio blocks
# 2294-2363, and 4092-4161 contains an error text string in between two sio blocks
# 4351-4927 has a bad AD then CT message where the size from the header does not line up with
# the final \x03
self.assert_result(result, [[637, 754, 1, 0],[6131, 6248, 1, 0]],
[[0,69],[637,754],[1329,1332], [2294,2363],
[4092,4161], [4351, 4927], [6131,6300]],
self.particle_b)
result = self.parser.get_records(2)
self.assertEqual(result[0], self.particle_c)
self.assertEqual(result[1], self.particle_d)
self.assert_state([], [[0,69],[1329,1332], [2294,2363],
[4092,4161], [4351, 4927], [6248,6300]])
stream_handle.close()
self.assertEqual(self.exception_callback_value, None)
def test_in_process_start(self):
"""
test starting a parser with a state in the middle of processing
"""
new_state = {StateKey.IN_PROCESS_DATA:[[390, 507, 1, 0], [637, 754, 1, 0], [6131, 6248, 1, 0]],
StateKey.UNPROCESSED_DATA:[ [390,6300]],
StateKey.METADATA_SENT: True,
StateKey.FILE_SIZE: 9400}
stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_shorter.dat'))
self.parser = DostadParser(self.config, new_state, stream_handle,
self.state_callback, self.pub_callback, self.exception_callback)
result = self.parser.get_records(1)
self.assert_result(result, [[637, 754, 1, 0], [6131, 6248, 1, 0]],
[[507,6300]],
self.particle_b)
result = self.parser.get_records(2)
self.assertEqual(result[0], self.particle_c)
self.assertEqual(result[1], self.particle_d)
self.assert_state([],
[[507,637], [754,6131], [6248, 6300]])
self.assertEqual(self.publish_callback_value[-1], self.particle_d)
self.assertEqual(self.exception_callback_value, None)
def test_set_state(self):
"""
test changing the state after initializing
"""
state = {StateKey.UNPROCESSED_DATA:[[0, 1000]], StateKey.IN_PROCESS_DATA:[],
StateKey.METADATA_SENT: False,
StateKey.FILE_SIZE: 9400}
new_state = {StateKey.UNPROCESSED_DATA:[[0,69],[944,6300]],
StateKey.IN_PROCESS_DATA:[], StateKey.METADATA_SENT: True,
StateKey.FILE_SIZE: 9400}
stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_shorter.dat'))
self.parser = DostadParser(self.config, state, stream_handle,
self.state_callback, self.pub_callback, self.exception_callback)
# there should only be 4 records, make sure we stop there
result = self.parser.get_records(4)
# 0-69 contains an incomplete block (end of a sample)
self.assert_state([],
[[0,69],[944,1000]])
result = self.parser.get_records(1)
self.assertEqual(result, [])
self.parser.set_state(new_state)
result = self.parser.get_records(1)
stream_handle.close()
# 0-69 contains an incomplete block (end of a sample)
# 1329-1332 there are 3 extra \n's between sio blocks
# 2294-2363, and 4092-4161 contains an error text string in between two sio blocks
# 4351-4927 has a bad AD then CT message where the size from the header does not line up with
# the final \x03
self.assert_result(result, [],
[[0,69],[1329,1332], [2294,2363], [4092,4161], [4351, 4927],
[6248,6300]],
self.particle_d)
self.assertEqual(self.exception_callback_value, None)
def test_update(self):
"""
Test a file which has had a section of data replaced by 0s, as if a block of data has not been received yet,
then using the returned state make a new parser with the test data that has the 0s filled in
"""
state = {StateKey.UNPROCESSED_DATA:[[0, 6300]],
StateKey.IN_PROCESS_DATA:[], StateKey.METADATA_SENT: False, StateKey.FILE_SIZE: 9400}
# this file has a block of DO data replaced by 0s
stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_replaced.dat'))
self.parser = DostadParser(self.config, state, stream_handle,
self.state_callback, self.pub_callback, self.exception_callback)
result = self.parser.get_records(1)
# 0-69 contains an incomplete block (end of a sample)
# 390-570 is the zeroed block
# 1329-1332 there are 3 extra \n's between sio blocks
# 2294-2363, and 4092-4161 contains an error text string in between two sio blocks
# 4351-4927 has a bad AD then CT message where the size from the header does not line up with
# the final \x03
self.assert_result(result, [[197,314,2,1], [637,754,1,0], [6131,6248,1,0]],
[[0,69], [197,314], [390,507], [637,754], [1329,1332], [2294,2363],
[4092,4161], [4351, 4927], [6131,6300]],
self.particle_metadata)
result = self.parser.get_records(1)
self.assert_result(result, [[637,754,1,0], [6131,6248,1,0]],
[[0,69], [390,507], [637,754], [1329,1332], [2294,2363],
[4092,4161], [4351, 4927], [6131,6300]],
self.particle_a)
result = self.parser.get_records(1)
self.assert_result(result, [[6131,6248,1,0]],
[[0,69], [390,507], [1329,1332], [2294,2363],
[4092,4161], [4351, 4927], [6131,6300]],
self.particle_c)
stream_handle.close()
next_state = self.parser._state
# this file has the block of data that was missing in the previous file
stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_shorter.dat'))
self.parser = DostadParser(self.config, next_state, stream_handle,
self.state_callback, self.pub_callback, self.exception_callback)
# first get the old 'in process' records from 6131-6248
# Once those are done, the un processed data will be checked
result = self.parser.get_records(1)
self.assert_result(result, [],
[[0,69], [390,507], [1329,1332], [2294,2363],
[4092,4161], [4351, 4927], [6248,6300]],
self.particle_d)
# this should be the first of the newly filled in particles from 390-507
result = self.parser.get_records(1)
self.assert_result(result, [],
[[0,69], [1329,1332], [2294,2363],
[4092,4161], [4351, 4927], [6248,6300]],
self.particle_b)
stream_handle.close()
self.assertEqual(self.exception_callback_value, None)
class DostadParserUnitTestCase(ParserUnitTestCase):
def state_callback(self, state):
""" Call back method to watch what comes in via the position callback """
self.state_callback_value = state
def pub_callback(self, pub):
""" Call back method to watch what comes in via the publish callback """
self.publish_callback_value = pub
def setUp(self):
ParserUnitTestCase.setUp(self)
self.config = {
DataSetDriverConfigKeys.PARTICLE_MODULE:
'mi.dataset.parser.dostad',
DataSetDriverConfigKeys.PARTICLE_CLASS: 'DostadParserDataParticle'
}
# first DO tag
self.timestamp1 = 3583610101.0
self.particle_a = DostadParserDataParticle('\xff\x11\x25\x11' \
'4831 128 302.636 98.918 16.355 30.771 ' \
'30.771 36.199 5.429 1309.9 1175.7 299.6\x0d\x0a',
internal_timestamp=self.timestamp1, new_sequence=True)
self.timestamp2 = 3583612801.0
self.particle_b = DostadParserDataParticle('\xff\x11\x25\x11' \
'4831 128 317.073 98.515 14.004 31.299 31.299' \
' 36.647 5.349 1338.5 1212.4 375.8\x0d\x0a',
internal_timestamp=self.timestamp2, new_sequence=False)
self.particle_b_new = DostadParserDataParticle('\xff\x11\x25\x11' \
'4831 128 317.073 98.515 14.004 31.299 31.299' \
' 36.647 5.349 1338.5 1212.4 375.8\x0d\x0a',
internal_timestamp=self.timestamp2, new_sequence=True)
self.timestamp3 = 3583634401.0
self.particle_c = DostadParserDataParticle('\xff\x11\x25\x11' \
'4831 128 326.544 96.726 11.873 31.965 31.965' \
' 37.134 5.169 1370.7 1245.9 444.4\x0d\x0a',
internal_timestamp=self.timestamp3, new_sequence=False)
self.particle_c_new = DostadParserDataParticle('\xff\x11\x25\x11' \
'4831 128 326.544 96.726 11.873 31.965 31.965' \
' 37.134 5.169 1370.7 1245.9 444.4\x0d\x0a',
internal_timestamp=self.timestamp3, new_sequence=True)
self.timestamp4 = 3583656002.0
self.particle_d = DostadParserDataParticle('\xff\x11\x25\x11' \
'4831 128 332.060 99.617 12.432 31.490 31.490' \
' 36.812 5.323 1026.0 1100.0 426.5\x0d\x0a',
internal_timestamp=self.timestamp4, new_sequence=True)
self.timestamp5 = 3583677602.0
self.particle_e = DostadParserDataParticle('\xff\x11\x25\x11' \
'4831 128 354.515 101.592 10.440 31.666 31.666' \
' 36.930 5.263 1020.1 1123.0 490.0\x0d\x0a',
internal_timestamp=self.timestamp5, new_sequence=False)
self.timestamp6 = 3583699202.0
self.particle_f = DostadParserDataParticle('\xff\x11\x25\x11' \
'4831 128 337.540 100.172 11.955 31.521 31.521' \
' 36.805 5.284 983.8 1092.6 441.7\x0d\x0a',
internal_timestamp=self.timestamp6, new_sequence=False)
self.state_callback_value = None
self.publish_callback_value = None
def assert_result(self, result, in_process_data, unprocessed_data,
timestamp, particle):
self.assertEqual(result, [particle])
self.assert_state(in_process_data, unprocessed_data, timestamp)
self.assert_(isinstance(self.publish_callback_value, list))
self.assertEqual(self.publish_callback_value[0], particle)
def assert_state(self, in_process_data, unprocessed_data, timestamp):
self.assertEqual(self.parser._state[StateKey.IN_PROCESS_DATA],
in_process_data)
self.assertEqual(self.parser._state[StateKey.UNPROCESSED_DATA],
unprocessed_data)
self.assertEqual(self.state_callback_value[StateKey.IN_PROCESS_DATA],
in_process_data)
self.assertEqual(self.state_callback_value[StateKey.UNPROCESSED_DATA],
unprocessed_data)
self.assertEqual(self.state_callback_value[StateKey.TIMESTAMP],
timestamp)
def test_simple(self):
"""
Read test data from the file and pull out data particles one at a time.
Assert that the results are those we expected.
"""
self.stream_handle = open(
os.path.join(RESOURCE_PATH, 'node59p1_shorter.dat'))
# NOTE: using the unprocessed data state of 0,6300 limits the file to reading
# just 6300 bytes, so even though the file is longer it only reads the first
# 6300
self.state = {
StateKey.UNPROCESSED_DATA: [[0, 6300]],
StateKey.IN_PROCESS_DATA: [],
StateKey.TIMESTAMP: 0.0
}
self.parser = DostadParser(self.config, self.state, self.stream_handle,
self.state_callback, self.pub_callback)
result = self.parser.get_records(1)
self.assert_result(
result,
[[390, 507, 1, 0, 0], [637, 754, 1, 0, 0], [6150, 6267, 1, 0, 1]],
[[0, 69], [390, 507], [637, 754], [944, 2370], [2560, 2947],
[3137, 4173], [4363, 5437], [5683, 6072], [6150, 6300]],
self.timestamp4, self.particle_a)
result = self.parser.get_records(1)
self.assert_result(
result, [[637, 754, 1, 0, 0], [6150, 6267, 1, 0, 1]],
[[0, 69], [637, 754], [944, 2370], [2560, 2947], [3137, 4173],
[4363, 5437], [5683, 6072], [6150, 6300]], self.timestamp4,
self.particle_b)
result = self.parser.get_records(1)
self.assert_result(result, [[6150, 6267, 1, 0, 1]],
[[0, 69], [944, 2370], [2560, 2947], [3137, 4173],
[4363, 5437], [5683, 6072], [6150, 6300]],
self.timestamp4, self.particle_c)
result = self.parser.get_records(1)
self.assert_result(result, [],
[[0, 69], [944, 2370], [2560, 2947], [3137, 4173],
[4363, 5437], [5683, 6072], [6267, 6300]],
self.timestamp4, self.particle_d)
self.stream_handle.close()
def test_get_many(self):
"""
Read test data from the file and pull out multiple data particles at one time.
Assert that the results are those we expected.
"""
self.state = {
StateKey.UNPROCESSED_DATA: [[0, 1000]],
StateKey.IN_PROCESS_DATA: [],
StateKey.TIMESTAMP: 0.0
}
self.stream_handle = open(
os.path.join(RESOURCE_PATH, 'node59p1_shorter.dat'))
self.parser = DostadParser(
self.config, self.state, self.stream_handle, self.state_callback,
self.pub_callback) # last one is the link to the data source
result = self.parser.get_records(3)
self.stream_handle.close()
self.assertEqual(result,
[self.particle_a, self.particle_b, self.particle_c])
self.assert_state([], [[0, 69], [944, 1000]], self.timestamp3)
self.assertEqual(self.publish_callback_value[0], self.particle_a)
self.assertEqual(self.publish_callback_value[1], self.particle_b)
self.assertEqual(self.publish_callback_value[2], self.particle_c)
def test_long_stream(self):
self.stream_handle = open(
os.path.join(RESOURCE_PATH, 'node59p1_shorter.dat'))
data = self.stream_handle.read()
data_len = len(data)
self.stream_handle.seek(0)
self.state = {
StateKey.UNPROCESSED_DATA: [[0, data_len]],
StateKey.IN_PROCESS_DATA: [],
StateKey.TIMESTAMP: 0.0
}
self.parser = DostadParser(
self.config, self.state, self.stream_handle, self.state_callback,
self.pub_callback) # last one is the link to the data source
result = self.parser.get_records(6)
self.stream_handle.close()
self.assertEqual(result[0], self.particle_a)
self.assertEqual(result[1], self.particle_b)
self.assertEqual(result[2], self.particle_c)
self.assertEqual(result[-3], self.particle_d)
self.assertEqual(result[-2], self.particle_e)
self.assertEqual(result[-1], self.particle_f)
self.assert_state([],
[[0, 69], [944, 2370], [2560, 2947], [3137, 4173],
[4363, 5437], [5683, 6072], [8273, 9400]],
self.timestamp6)
self.assertEqual(self.publish_callback_value[-2], self.particle_e)
self.assertEqual(self.publish_callback_value[-1], self.particle_f)
def test_mid_state_start(self):
"""
test starting a parser with a state in the middle of processing
"""
new_state = {
StateKey.IN_PROCESS_DATA: [],
StateKey.UNPROCESSED_DATA: [[0, 69], [314, 1000]],
StateKey.TIMESTAMP: self.timestamp1
}
self.stream_handle = open(
os.path.join(RESOURCE_PATH, 'node59p1_shorter.dat'))
self.parser = DostadParser(
self.config, new_state, self.stream_handle, self.state_callback,
self.pub_callback) # last one is the link to the data source
result = self.parser.get_records(1)
self.assert_result(result, [
[637, 754, 1, 0, 0],
], [[0, 69], [637, 754], [944, 1000]], self.timestamp3,
self.particle_b_new)
result = self.parser.get_records(1)
self.assert_result(result, [], [[0, 69], [944, 1000]], self.timestamp3,
self.particle_c)
self.stream_handle.close()
def test_in_process_start(self):
"""
test starting a parser with a state in the middle of processing
"""
new_state = {
StateKey.IN_PROCESS_DATA: [[390, 507, 1, 0, 0],
[637, 754, 1, 0, 0]],
StateKey.UNPROCESSED_DATA: [[0, 69], [390, 507], [637, 754],
[944, 6300]],
StateKey.TIMESTAMP:
self.timestamp3
}
self.stream_handle = open(
os.path.join(RESOURCE_PATH, 'node59p1_shorter.dat'))
self.parser = DostadParser(
self.config, new_state, self.stream_handle, self.state_callback,
self.pub_callback) # last one is the link to the data source
result = self.parser.get_records(1)
# even though the state says this particle is not a new sequence, since it is the
# first after setting the state it will be new
self.assert_result(result, [[637, 754, 1, 0, 0]],
[[0, 69], [637, 754], [944, 6300]], self.timestamp2,
self.particle_b_new)
result = self.parser.get_records(2)
self.assertEqual(result[0], self.particle_c)
self.assertEqual(result[1], self.particle_d)
self.assert_state([],
[[0, 69], [944, 2370], [2560, 2947], [3137, 4173],
[4363, 5437], [5683, 6072], [6267, 6300]],
self.timestamp4)
self.assertEqual(self.publish_callback_value[-1], self.particle_d)
def test_set_state(self):
"""
test changing the state after initializing
"""
self.state = {
StateKey.UNPROCESSED_DATA: [[0, 1000]],
StateKey.IN_PROCESS_DATA: [],
StateKey.TIMESTAMP: 0.0
}
new_state = {
StateKey.UNPROCESSED_DATA: [[0, 69], [944, 6300]],
StateKey.IN_PROCESS_DATA: [],
StateKey.TIMESTAMP: self.timestamp2
}
self.stream_handle = open(
os.path.join(RESOURCE_PATH, 'node59p1_shorter.dat'))
self.parser = DostadParser(
self.config, self.state, self.stream_handle, self.state_callback,
self.pub_callback) # last one is the link to the data source
# there should only be 3 records, make sure we stop there
result = self.parser.get_records(3)
self.assert_state([], [[0, 69], [944, 1000]], self.timestamp3)
result = self.parser.get_records(1)
self.assertEqual(result, [])
self.parser.set_state(new_state)
result = self.parser.get_records(1)
self.stream_handle.close()
self.assert_result(result, [],
[[0, 69], [944, 2370], [2560, 2947], [3137, 4173],
[4363, 5437], [5683, 6072], [6267, 6300]],
self.timestamp4, self.particle_d)
def test_update(self):
"""
Test a file which has had a section of data replaced by 0s, as if a block of data has not been received yet,
then using the returned state make a new parser with the test data that has the 0s filled in
"""
self.state = {
StateKey.UNPROCESSED_DATA: [[0, 6300]],
StateKey.IN_PROCESS_DATA: [],
StateKey.TIMESTAMP: 0.0
}
# this file has a block of FL data replaced by 0s
self.stream_handle = open(
os.path.join(RESOURCE_PATH, 'node59p1_replaced.dat'))
self.parser = DostadParser(
self.config, self.state, self.stream_handle, self.state_callback,
self.pub_callback) # last one is the link to the data source
result = self.parser.get_records(1)
self.assert_result(
result, [[637, 754, 1, 0, 1], [6150, 6267, 1, 0, 1]],
[[0, 69], [390, 507], [637, 754], [944, 2370], [2560, 2947],
[3137, 4173], [4363, 5437], [5683, 6072], [6150, 6300]],
self.timestamp4, self.particle_a)
result = self.parser.get_records(1)
self.assert_result(
result, [[6150, 6267, 1, 0, 1]],
[[0, 69], [390, 507], [944, 2370], [2560, 2947], [3137, 4173],
[4363, 5437], [5683, 6072], [6150, 6300]], self.timestamp4,
self.particle_c_new)
self.stream_handle.close()
next_state = self.parser._state
# this file has the block of data that was missing in the previous file
self.stream_handle = open(
os.path.join(RESOURCE_PATH, 'node59p1_shorter.dat'))
self.parser = DostadParser(
self.config, next_state, self.stream_handle, self.state_callback,
self.pub_callback) # last one is the link to the data source
# first get the old 'in process' records
# Once those are done, the un processed data will be checked
result = self.parser.get_records(1)
self.assert_result(
result, [],
[[0, 69], [390, 507], [944, 2370], [2560, 2947], [3137, 4173],
[4363, 5437], [5683, 6072], [6267, 6300]], self.timestamp4,
self.particle_d)
# this should be the first of the newly filled in particles from
result = self.parser.get_records(1)
self.assert_result(result, [],
[[0, 69], [944, 2370], [2560, 2947], [3137, 4173],
[4363, 5437], [5683, 6072], [6267, 6300]],
self.timestamp2, self.particle_b_new)
self.stream_handle.close()
class DostadParserUnitTestCase(ParserUnitTestCase):
def state_callback(self, state):
""" Call back method to watch what comes in via the position callback """
self.state_callback_value = state
def pub_callback(self, pub):
""" Call back method to watch what comes in via the publish callback """
self.publish_callback_value = pub
def setUp(self):
ParserUnitTestCase.setUp(self)
self.config = {
DataSetDriverConfigKeys.PARTICLE_MODULE: 'mi.dataset.parser.dostad',
DataSetDriverConfigKeys.PARTICLE_CLASS: 'DostadParserDataParticle'
}
# first DO tag
self.timestamp1 = 3583610101.0
self.particle_a = DostadParserDataParticle('\xff\x11\x25\x11' \
'4831 128 302.636 98.918 16.355 30.771 ' \
'30.771 36.199 5.429 1309.9 1175.7 299.6\x0d\x0a',
internal_timestamp=self.timestamp1, new_sequence=True)
self.timestamp2 = 3583612801.0
self.particle_b = DostadParserDataParticle('\xff\x11\x25\x11' \
'4831 128 317.073 98.515 14.004 31.299 31.299' \
' 36.647 5.349 1338.5 1212.4 375.8\x0d\x0a',
internal_timestamp=self.timestamp2, new_sequence=False)
self.particle_b_new = DostadParserDataParticle('\xff\x11\x25\x11' \
'4831 128 317.073 98.515 14.004 31.299 31.299' \
' 36.647 5.349 1338.5 1212.4 375.8\x0d\x0a',
internal_timestamp=self.timestamp2, new_sequence=True)
self.timestamp3 = 3583634401.0
self.particle_c = DostadParserDataParticle('\xff\x11\x25\x11' \
'4831 128 326.544 96.726 11.873 31.965 31.965' \
' 37.134 5.169 1370.7 1245.9 444.4\x0d\x0a',
internal_timestamp=self.timestamp3, new_sequence=False)
self.particle_c_new = DostadParserDataParticle('\xff\x11\x25\x11' \
'4831 128 326.544 96.726 11.873 31.965 31.965' \
' 37.134 5.169 1370.7 1245.9 444.4\x0d\x0a',
internal_timestamp=self.timestamp3, new_sequence=True)
self.timestamp4 = 3583656002.0
self.particle_d = DostadParserDataParticle('\xff\x11\x25\x11' \
'4831 128 332.060 99.617 12.432 31.490 31.490' \
' 36.812 5.323 1026.0 1100.0 426.5\x0d\x0a',
internal_timestamp=self.timestamp4, new_sequence=True)
self.timestamp5 = 3583677602.0
self.particle_e = DostadParserDataParticle('\xff\x11\x25\x11' \
'4831 128 354.515 101.592 10.440 31.666 31.666' \
' 36.930 5.263 1020.1 1123.0 490.0\x0d\x0a',
internal_timestamp=self.timestamp5, new_sequence=False)
self.timestamp6 = 3583699202.0
self.particle_f = DostadParserDataParticle('\xff\x11\x25\x11' \
'4831 128 337.540 100.172 11.955 31.521 31.521' \
' 36.805 5.284 983.8 1092.6 441.7\x0d\x0a',
internal_timestamp=self.timestamp6, new_sequence=False)
self.state_callback_value = None
self.publish_callback_value = None
def assert_result(self, result, in_process_data, unprocessed_data, timestamp, particle):
self.assertEqual(result, [particle])
self.assert_state(in_process_data, unprocessed_data, timestamp)
self.assert_(isinstance(self.publish_callback_value, list))
self.assertEqual(self.publish_callback_value[0], particle)
def assert_state(self, in_process_data, unprocessed_data, timestamp):
self.assertEqual(self.parser._state[StateKey.IN_PROCESS_DATA], in_process_data)
self.assertEqual(self.parser._state[StateKey.UNPROCESSED_DATA], unprocessed_data)
self.assertEqual(self.state_callback_value[StateKey.IN_PROCESS_DATA], in_process_data)
self.assertEqual(self.state_callback_value[StateKey.UNPROCESSED_DATA], unprocessed_data)
self.assertEqual(self.state_callback_value[StateKey.TIMESTAMP], timestamp)
def test_simple(self):
"""
Read test data from the file and pull out data particles one at a time.
Assert that the results are those we expected.
"""
self.stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_shorter.dat'))
# NOTE: using the unprocessed data state of 0,6300 limits the file to reading
# just 6300 bytes, so even though the file is longer it only reads the first
# 6300
self.state = {StateKey.UNPROCESSED_DATA:[[0, 6300]],
StateKey.IN_PROCESS_DATA:[], StateKey.TIMESTAMP:0.0}
self.parser = DostadParser(self.config, self.state, self.stream_handle,
self.state_callback, self.pub_callback)
result = self.parser.get_records(1)
self.assert_result(result,
[[390, 507, 1, 0, 0], [637, 754, 1, 0, 0], [6150, 6267, 1, 0, 1]],
[[0,69], [390,507], [637,754], [944, 2370], [2560, 2947],
[3137, 4173], [4363, 5437], [5683, 6072], [6150, 6300]],
self.timestamp4, self.particle_a)
result = self.parser.get_records(1)
self.assert_result(result, [[637, 754, 1, 0, 0], [6150, 6267, 1, 0, 1]],
[[0,69], [637,754], [944, 2370], [2560, 2947],
[3137, 4173], [4363, 5437], [5683, 6072], [6150, 6300]],
self.timestamp4, self.particle_b)
result = self.parser.get_records(1)
self.assert_result(result, [[6150, 6267, 1, 0, 1]],
[[0,69], [944, 2370], [2560, 2947], [3137, 4173],
[4363, 5437], [5683, 6072], [6150, 6300]],
self.timestamp4, self.particle_c)
result = self.parser.get_records(1)
self.assert_result(result, [],
[[0,69], [944, 2370], [2560, 2947], [3137, 4173],
[4363, 5437], [5683, 6072], [6267, 6300]],
self.timestamp4, self.particle_d)
self.stream_handle.close()
def test_get_many(self):
"""
Read test data from the file and pull out multiple data particles at one time.
Assert that the results are those we expected.
"""
self.state = {StateKey.UNPROCESSED_DATA:[[0, 1000]],
StateKey.IN_PROCESS_DATA:[], StateKey.TIMESTAMP:0.0}
self.stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_shorter.dat'))
self.parser = DostadParser(self.config, self.state, self.stream_handle,
self.state_callback, self.pub_callback) # last one is the link to the data source
result = self.parser.get_records(3)
self.stream_handle.close()
self.assertEqual(result,
[self.particle_a, self.particle_b, self.particle_c])
self.assert_state([],
[[0,69],[944,1000]],
self.timestamp3)
self.assertEqual(self.publish_callback_value[0], self.particle_a)
self.assertEqual(self.publish_callback_value[1], self.particle_b)
self.assertEqual(self.publish_callback_value[2], self.particle_c)
def test_long_stream(self):
self.stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_shorter.dat'))
data = self.stream_handle.read()
data_len = len(data)
self.stream_handle.seek(0)
self.state = {StateKey.UNPROCESSED_DATA:[[0, data_len]],
StateKey.IN_PROCESS_DATA:[], StateKey.TIMESTAMP:0.0}
self.parser = DostadParser(self.config, self.state, self.stream_handle,
self.state_callback, self.pub_callback) # last one is the link to the data source
result = self.parser.get_records(6)
self.stream_handle.close()
self.assertEqual(result[0], self.particle_a)
self.assertEqual(result[1], self.particle_b)
self.assertEqual(result[2], self.particle_c)
self.assertEqual(result[-3], self.particle_d)
self.assertEqual(result[-2], self.particle_e)
self.assertEqual(result[-1], self.particle_f)
self.assert_state([],
[[0, 69], [944, 2370], [2560, 2947], [3137, 4173], [4363, 5437], [5683, 6072], [8273, 9400]],
self.timestamp6)
self.assertEqual(self.publish_callback_value[-2], self.particle_e)
self.assertEqual(self.publish_callback_value[-1], self.particle_f)
def test_mid_state_start(self):
"""
test starting a parser with a state in the middle of processing
"""
new_state = {StateKey.IN_PROCESS_DATA:[],
StateKey.UNPROCESSED_DATA:[[0,69], [314,1000]],
StateKey.TIMESTAMP:self.timestamp1}
self.stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_shorter.dat'))
self.parser = DostadParser(self.config, new_state, self.stream_handle,
self.state_callback, self.pub_callback) # last one is the link to the data source
result = self.parser.get_records(1)
self.assert_result(result, [[637, 754, 1, 0, 0],],
[[0,69],[637,754],[944,1000]],
self.timestamp3, self.particle_b_new)
result = self.parser.get_records(1)
self.assert_result(result, [],
[[0,69],[944,1000]],
self.timestamp3, self.particle_c)
self.stream_handle.close()
def test_in_process_start(self):
"""
test starting a parser with a state in the middle of processing
"""
new_state = {StateKey.IN_PROCESS_DATA:[[390, 507, 1, 0, 0], [637, 754, 1, 0, 0]],
StateKey.UNPROCESSED_DATA:[[0,69], [390,507], [637,754], [944,6300]],
StateKey.TIMESTAMP:self.timestamp3}
self.stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_shorter.dat'))
self.parser = DostadParser(self.config, new_state, self.stream_handle,
self.state_callback, self.pub_callback) # last one is the link to the data source
result = self.parser.get_records(1)
# even though the state says this particle is not a new sequence, since it is the
# first after setting the state it will be new
self.assert_result(result, [[637, 754, 1, 0, 0]],
[[0,69],[637,754],[944,6300]],
self.timestamp2, self.particle_b_new)
result = self.parser.get_records(2)
self.assertEqual(result[0], self.particle_c)
self.assertEqual(result[1], self.particle_d)
self.assert_state([],
[[0,69],[944, 2370], [2560, 2947], [3137, 4173],
[4363, 5437], [5683, 6072], [6267, 6300]],
self.timestamp4)
self.assertEqual(self.publish_callback_value[-1], self.particle_d)
def test_set_state(self):
"""
test changing the state after initializing
"""
self.state = {StateKey.UNPROCESSED_DATA:[[0, 1000]], StateKey.IN_PROCESS_DATA:[],
StateKey.TIMESTAMP:0.0}
new_state = {StateKey.UNPROCESSED_DATA:[[0,69],[944,6300]],
StateKey.IN_PROCESS_DATA:[],
StateKey.TIMESTAMP:self.timestamp2}
self.stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_shorter.dat'))
self.parser = DostadParser(self.config, self.state, self.stream_handle,
self.state_callback, self.pub_callback) # last one is the link to the data source
# there should only be 3 records, make sure we stop there
result = self.parser.get_records(3)
self.assert_state([],
[[0,69],[944,1000]],
self.timestamp3)
result = self.parser.get_records(1)
self.assertEqual(result, [])
self.parser.set_state(new_state)
result = self.parser.get_records(1)
self.stream_handle.close()
self.assert_result(result, [],
[[0,69],[944, 2370], [2560, 2947], [3137, 4173],
[4363, 5437], [5683, 6072], [6267, 6300]],
self.timestamp4, self.particle_d)
def test_update(self):
"""
Test a file which has had a section of data replaced by 0s, as if a block of data has not been received yet,
then using the returned state make a new parser with the test data that has the 0s filled in
"""
self.state = {StateKey.UNPROCESSED_DATA:[[0, 6300]],
StateKey.IN_PROCESS_DATA:[], StateKey.TIMESTAMP:0.0}
# this file has a block of FL data replaced by 0s
self.stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_replaced.dat'))
self.parser = DostadParser(self.config, self.state, self.stream_handle,
self.state_callback, self.pub_callback) # last one is the link to the data source
result = self.parser.get_records(1)
self.assert_result(result, [[637,754,1,0,1], [6150,6267,1,0,1]],
[[0,69], [390,507], [637,754], [944, 2370], [2560, 2947],
[3137, 4173], [4363, 5437], [5683, 6072], [6150, 6300]],
self.timestamp4, self.particle_a)
result = self.parser.get_records(1)
self.assert_result(result, [[6150,6267,1,0,1]],
[[0,69], [390,507], [944, 2370], [2560, 2947],
[3137, 4173], [4363, 5437], [5683, 6072], [6150, 6300]],
self.timestamp4, self.particle_c_new)
self.stream_handle.close()
next_state = self.parser._state
# this file has the block of data that was missing in the previous file
self.stream_handle = open(os.path.join(RESOURCE_PATH,
'node59p1_shorter.dat'))
self.parser = DostadParser(self.config, next_state, self.stream_handle,
self.state_callback, self.pub_callback) # last one is the link to the data source
# first get the old 'in process' records
# Once those are done, the un processed data will be checked
result = self.parser.get_records(1)
self.assert_result(result, [],
[[0,69], [390,507], [944, 2370], [2560, 2947], [3137, 4173],
[4363, 5437], [5683, 6072], [6267, 6300]],
self.timestamp4, self.particle_d)
# this should be the first of the newly filled in particles from
result = self.parser.get_records(1)
self.assert_result(result, [],
[[0,69], [944, 2370], [2560, 2947], [3137, 4173], [4363, 5437],
[5683, 6072], [6267, 6300]],
self.timestamp2, self.particle_b_new)
self.stream_handle.close()
