def test_bad_diag_checksum_21_recs(self):
"""
The file used in this test has a power record with a missing timestamp.
This results in 9 particles being retrieved instead of 10, and also result in the exception
callback being called.
"""
log.debug(
'===== START TEST FOUND BAD DIAG HDR CHECKSUM AND TOO MANY RECS ====='
)
# Test the telemetered version
log.debug('----- TELEMETERED -----')
with open(
os.path.join(
RESOURCE_PATH,
'bad_diag_hdr_checksum_21_diag_20140813.velpt.log'),
'rb') as file_handle:
num_particles_to_request = num_expected_particles = 50
parser = VelptAbDclParser(self._telemetered_parser_config,
file_handle, self.exception_callback)
particles = parser.get_records(num_particles_to_request)
self.assertEquals(len(particles), num_expected_particles)
self.assert_particles(
particles, 'bad_diag_hdr_checksum_21_diag_20140813.velpt.yml',
RESOURCE_PATH)
# Test the recovered version
log.debug('------ RECOVERED ------')
with open(
os.path.join(
RESOURCE_PATH,
'bad_diag_hdr_checksum_21_diag_20140813.velpt.log'),
'rb') as file_handle:
num_particles_to_request = num_expected_particles = 50
parser = VelptAbDclParser(self._recovered_parser_config,
file_handle, self.exception_callback)
particles = parser.get_records(num_particles_to_request)
self.assertEquals(len(particles), num_expected_particles)
self.assert_particles(
particles,
'recovered_bad_diag_hdr_checksum_21_diag_20140813.velpt.yml',
RESOURCE_PATH)
log.debug(
'===== END TEST FOUND BAD DIAG HDR CHECKSUM AND TOO MANY RECS ====='
)
def test_diag_header_bad_checksum(self):
"""
The file used in this test has a record with a bad checksum.
This results in 491 particles being retrieved instead of 50.
The standard 20140813.velpt.log was used, the checksum of the
third velocity record was corrupted to make it fail.
"""
log.debug(
'===== START TEST FOUND BAD DIAGNOSTIC HEADER CHECKSUM =====')
# Test the telemetered version
log.debug('----- TELEMETERED -----')
with open(
os.path.join(RESOURCE_PATH,
'bad_diag_hdr_checksum_20140813.velpt.log'),
'rb') as file_handle:
num_particles_to_request = num_expected_particles = 49
parser = VelptAbDclParser(self._telemetered_parser_config,
file_handle, self.exception_callback)
particles = parser.get_records(num_particles_to_request)
self.assertEquals(len(particles), num_expected_particles)
self.assert_particles(particles,
'bad_diag_hdr_checksum_20140813.velpt.yml',
RESOURCE_PATH)
# Test the recovered version
log.debug('----- RECOVERED -----')
with open(
os.path.join(RESOURCE_PATH,
'bad_diag_hdr_checksum_20140813.velpt.log'),
'rb') as file_handle:
num_particles_to_request = num_expected_particles = 49
parser = VelptAbDclParser(self._recovered_parser_config,
file_handle, self.exception_callback)
particles = parser.get_records(num_particles_to_request)
self.assertEquals(len(particles), num_expected_particles)
self.assert_particles(
particles,
'recovered_bad_diag_hdr_checksum_20140813.velpt.yml',
RESOURCE_PATH)
log.debug('===== END TEST FOUND BAD DIAGNOSTIC HEADER CHECKSUM =====')
def test_invalid_sync_byte(self):
"""
The file used in this test has extra bytes between records which need to be skipped
in order to process the correct number of particles.
"""
log.debug('===== START TEST INVALID SYNC BYTE =====')
# Test the telemetered version
log.debug('----- TELEMETERED -----')
with open(
os.path.join(RESOURCE_PATH, 'extra_bytes_20140813.velpt.log'),
'rb') as file_handle:
num_particles_to_request = num_expected_particles = 50
parser = VelptAbDclParser(self._telemetered_parser_config,
file_handle, self.exception_callback)
particles = parser.get_records(num_particles_to_request)
self.assertEquals(len(particles), num_expected_particles)
self.assert_particles(particles, '20140813.velpt.yml',
RESOURCE_PATH)
# Test the recovered version
log.debug('------ RECOVERED ------')
with open(
os.path.join(RESOURCE_PATH, 'extra_bytes_20140813.velpt.log'),
'rb') as file_handle:
num_particles_to_request = num_expected_particles = 50
parser = VelptAbDclParser(self._recovered_parser_config,
file_handle, self.exception_callback)
particles = parser.get_records(num_particles_to_request)
self.assertEquals(len(particles), num_expected_particles)
self.assert_particles(particles, 'recovered_20140813.velpt.yml',
RESOURCE_PATH)
log.debug('===== END TEST INVALID SYNC BYTE =====')
def test_simple(self):
"""
Read files and verify that all expected particles can be read.
Verify that the contents of the particles are correct.
This is the happy path.
"""
log.debug('===== START TEST SIMPLE =====')
# Test the telemetered version
log.debug('----- TELEMETERED -----')
with open(os.path.join(RESOURCE_PATH, '20140813.velpt.log'),
'rb') as file_handle:
num_particles_to_request = num_expected_particles = 50
parser = VelptAbDclParser(self._telemetered_parser_config,
file_handle, self.exception_callback)
particles = parser.get_records(num_particles_to_request)
self.assertEquals(len(particles), num_expected_particles)
self.assert_particles(particles, '20140813.velpt.yml',
RESOURCE_PATH)
# Test the recovered version
log.debug('------ RECOVERED ------')
with open(os.path.join(RESOURCE_PATH, '20140813.velpt.log'),
'rb') as file_handle:
num_particles_to_request = num_expected_particles = 50
parser = VelptAbDclParser(self._recovered_parser_config,
file_handle, self.exception_callback)
particles = parser.get_records(num_particles_to_request)
self.assertEquals(len(particles), num_expected_particles)
self.assert_particles(particles, 'recovered_20140813.velpt.yml',
RESOURCE_PATH)
log.debug('===== END TEST SIMPLE =====')
def test_too_many_diagnostics_records(self):
"""
The file used in this test has 21 diagnostics records in the second set.
Twenty are expected.
"""
log.debug('===== START TEST TOO MANY DIAGNOSTICS RECORDS =====')
# Test the telemetered version
log.debug('----- TELEMETERED -----')
with open(os.path.join(RESOURCE_PATH, 'too_many_20140813.velpt.log'),
'rb') as file_handle:
num_particles_to_request = num_expected_particles = 51
parser = VelptAbDclParser(self._telemetered_parser_config,
file_handle, self.exception_callback)
particles = parser.get_records(num_particles_to_request)
self.assertEquals(len(particles), num_expected_particles)
self.assert_particles(particles, 'too_many_20140813.velpt.yml',
RESOURCE_PATH)
# Test the recovered version
log.debug('------ RECOVERED ------')
with open(os.path.join(RESOURCE_PATH, 'too_many_20140813.velpt.log'),
'rb') as file_handle:
num_particles_to_request = num_expected_particles = 51
parser = VelptAbDclParser(self._recovered_parser_config,
file_handle, self.exception_callback)
particles = parser.get_records(num_particles_to_request)
self.assertEquals(len(particles), num_expected_particles)
self.assert_particles(particles,
'recovered_too_many_20140813.velpt.yml',
RESOURCE_PATH)
log.debug('===== END TEST TOO MANY DIAGNOSTICS RECORDS =====')
def test_partial_configuration(self):
"""
Attempt to build a parser with a bad configuration.
"""
log.debug('===== START TEST PARTIAL CONFIGURATION =====')
with open(os.path.join(RESOURCE_PATH, '20140813.velpt.log'),
'rb') as file_handle:
with self.assertRaises(ConfigurationException):
parser = VelptAbDclParser(self._incomplete_parser_config,
file_handle, self.exception_callback)
log.debug('===== END TEST PARTIAL CONFIGURATION =====')
def _build_parser(self, stream_handle):
parser_config = {
DataSetDriverConfigKeys.PARTICLE_MODULE: 'mi.dataset.parser.velpt_ab_dcl_particles',
DataSetDriverConfigKeys.PARTICLE_CLASS: None,
DataSetDriverConfigKeys.PARTICLE_CLASSES_DICT: {
VelptAbDclParticleClassKey.METADATA_PARTICLE_CLASS: VelptAbDclDiagnosticsHeaderParticleRecovered,
VelptAbDclParticleClassKey.DIAGNOSTICS_PARTICLE_CLASS: VelptAbDclDiagnosticsDataParticleRecovered,
VelptAbDclParticleClassKey.INSTRUMENT_PARTICLE_CLASS: VelptAbDclInstrumentDataParticleRecovered
}
}
parser = VelptAbDclParser(parser_config,
stream_handle,
self._exception_callback)
return parser
def parse_live_logs(self):
"""
These tests were used to view the output of files associated with Bug_4341
"""
with open(os.path.join(RESOURCE_PATH, '20141110.velpt2.log'),
'rb') as file_handle:
parser = VelptAbDclParser(self._telemetered_parser_config,
file_handle, self.exception_callback)
particles = parser.get_records(100)
self.particle_to_yml(
particles, os.path.join(RESOURCE_PATH, '20141110.velpt2.yml'))
with open(os.path.join(RESOURCE_PATH, '20150613.velpt.log'),
'rb') as file_handle:
parser = VelptAbDclParser(self._telemetered_parser_config,
file_handle, self.exception_callback)
particles = parser.get_records(100)
self.particle_to_yml(
particles, os.path.join(RESOURCE_PATH, '20150613.velpt.yml'))
with open(os.path.join(RESOURCE_PATH, '20150518.velpt.log'),
'rb') as file_handle:
parser = VelptAbDclParser(self._telemetered_parser_config,
file_handle, self.exception_callback)
particles = parser.get_records(100)
self.particle_to_yml(
particles, os.path.join(RESOURCE_PATH, '20150518.velpt.yml'))
with open(os.path.join(RESOURCE_PATH, '20150409.velpt1.log'),
'rb') as file_handle:
parser = VelptAbDclParser(self._telemetered_parser_config,
file_handle, self.exception_callback)
particles = parser.get_records(100)
self.particle_to_yml(
particles, os.path.join(RESOURCE_PATH, '20150409.velpt1.yml'))
with open(os.path.join(RESOURCE_PATH, '20150428.velpt2.log'),
'rb') as file_handle:
parser = VelptAbDclParser(self._telemetered_parser_config,
file_handle, self.exception_callback)
particles = parser.get_records(100)
self.particle_to_yml(
particles, os.path.join(RESOURCE_PATH, '20150428.velpt2.yml'))
with open(os.path.join(RESOURCE_PATH, '20150824.velpt1.log'),
'rb') as file_handle:
parser = VelptAbDclParser(self._telemetered_parser_config,
file_handle, self.exception_callback)
particles = parser.get_records(100)
self.particle_to_yml(
particles, os.path.join(RESOURCE_PATH, '20150824.velpt1.yml'))
with open(os.path.join(RESOURCE_PATH, '20150829.velpt2.log'),
'rb') as file_handle:
parser = VelptAbDclParser(self._telemetered_parser_config,
file_handle, self.exception_callback)
particles = parser.get_records(100)
self.particle_to_yml(
particles, os.path.join(RESOURCE_PATH, '20150829.velpt2.yml'))