def test_mid_state_start(self):
"""
Test starting a parser with a state in the middle of processing.
"""
log.debug('===== START TEST MID-STATE START RECOVERED =====')
in_file = self.open_file(FILE3)
# Start at the beginning of the record 10 (of 16 total).
initial_state = {SpkirStateKey.POSITION: 1376}
parser = self.create_rec_parser(in_file, new_state=initial_state)
# Generate a list of expected result particles.
expected_particle = []
for expected in EXPECTED_FILE3[-7:]:
particle = SpkirAbjDclRecoveredInstrumentDataParticle(expected)
expected_particle.append(particle)
# In a single read, get all particles for this file.
result = parser.get_records(len(expected_particle))
self.assertEqual(result, expected_particle)
self.assertEqual(self.rec_exception_callback_value, None)
in_file.close()
log.debug('===== START TEST MID-STATE START TELEMETERED =====')
in_file = self.open_file(FILE2)
# Start at the beginning of the record 6 (of 13 total).
initial_state = {SpkirStateKey.POSITION: 731}
parser = self.create_tel_parser(in_file, new_state=initial_state)
# Generate a list of expected result particles.
expected_particle = []
for expected in EXPECTED_FILE2[-8:]:
particle = SpkirAbjDclTelemeteredInstrumentDataParticle(expected)
expected_particle.append(particle)
# In a single read, get all particles for this file.
result = parser.get_records(len(expected_particle))
self.assertEqual(result, expected_particle)
self.assertEqual(self.tel_exception_callback_value, None)
in_file.close()
log.debug('===== END TEST MID-STATE START =====')
def test_simple(self):
"""
Read data from a file and pull out data particles
one at a time. Verify that the results are those we expected.
"""
log.debug('===== START TEST SIMPLE RECOVERED =====')
in_file = self.open_file(FILE2)
parser = self.create_rec_parser(in_file)
for expected in EXPECTED_FILE2:
# Generate expected particle
expected_particle = SpkirAbjDclRecoveredInstrumentDataParticle(
expected)
# Get record and verify.
result = parser.get_records(1)
self.assertEqual(result, [expected_particle])
self.assertEqual(self.rec_exception_callback_value, None)
in_file.close()
log.debug('===== START TEST SIMPLE TELEMETERED =====')
in_file = self.open_file(FILE3)
parser = self.create_tel_parser(in_file)
for expected in EXPECTED_FILE3:
# Generate expected particle
expected_particle = SpkirAbjDclTelemeteredInstrumentDataParticle(
expected)
# Get record and verify.
result = parser.get_records(1)
self.assertEqual(result, [expected_particle])
self.assertEqual(self.tel_exception_callback_value, None)
in_file.close()
log.debug('===== END TEST SIMPLE =====')
def test_get_many(self):
"""
Read a file and pull out multiple data particles at one time.
Verify that the results are those we expected.
"""
log.debug('===== START TEST GET MANY RECOVERED =====')
in_file = self.open_file(FILE5)
parser = self.create_rec_parser(in_file)
# Generate a list of expected result particles.
expected_particle = []
for expected in EXPECTED_FILE5:
particle = SpkirAbjDclRecoveredInstrumentDataParticle(expected)
expected_particle.append(particle)
# In a single read, get all particles for this file.
result = parser.get_records(len(expected_particle))
self.assertEqual(result, expected_particle)
self.assertEqual(self.rec_exception_callback_value, None)
in_file.close()
log.debug('===== START TEST GET MANY TELEMETERED =====')
in_file = self.open_file(FILE4)
parser = self.create_tel_parser(in_file)
# Generate a list of expected result particles.
expected_particle = []
for expected in EXPECTED_FILE4:
particle = SpkirAbjDclTelemeteredInstrumentDataParticle(expected)
expected_particle.append(particle)
# In a single read, get all particles for this file.
result = parser.get_records(len(expected_particle))
self.assertEqual(result, expected_particle)
self.assertEqual(self.tel_exception_callback_value, None)
in_file.close()
log.debug('===== END TEST GET MANY =====')
def test_set_state(self):
"""
This test verifies that the state can be changed after starting.
Some particles are read and then the parser state is modified to
skip ahead or back.
"""
log.debug('===== START TEST SET STATE RECOVERED =====')
in_file = self.open_file(FILE4)
parser = self.create_rec_parser(in_file)
# Read and verify 5 particles (of the 15).
for expected in EXPECTED_FILE4[:5]:
# Generate expected particle
expected_particle = SpkirAbjDclRecoveredInstrumentDataParticle(
expected)
# Get record and verify.
result = parser.get_records(1)
self.assertEqual(result, [expected_particle])
# Skip ahead in the file so that we get the last 4 particles.
new_state = {SpkirStateKey.POSITION: 1854}
# Set the state.
parser.set_state(new_state)
# Read and verify the last 4 particles.
for expected in EXPECTED_FILE4[-4:]:
# Generate expected particle
expected_particle = SpkirAbjDclRecoveredInstrumentDataParticle(
expected)
# Get record and verify.
result = parser.get_records(1)
self.assertEqual(result, [expected_particle])
log.debug('===== START TEST SET STATE TELEMETERED =====')
in_file = self.open_file(FILE5)
parser = self.create_tel_parser(in_file)
# Read and verify 8 particles (of the 12).
for expected in EXPECTED_FILE5[:8]:
# Generate expected particle
expected_particle = SpkirAbjDclTelemeteredInstrumentDataParticle(
expected)
# Get record and verify.
result = parser.get_records(1)
self.assertEqual(result, [expected_particle])
# Skip back in the file so that we get the last 8 particles.
new_state = {
SpkirStateKey.POSITION: 956,
}
# Set the state.
parser.set_state(new_state)
# Read and verify the last 8 particles.
for expected in EXPECTED_FILE5[-8:]:
# Generate expected particle
expected_particle = SpkirAbjDclTelemeteredInstrumentDataParticle(
expected)
# Get record and verify.
result = parser.get_records(1)
self.assertEqual(result, [expected_particle])
log.debug('===== END TEST SET STATE =====')
def stream_config(cls):
return [
SpkirAbjDclRecoveredInstrumentDataParticle.type(),
SpkirAbjDclTelemeteredInstrumentDataParticle.type()
]