def test_set_state(self): """ Test changing to a new state after initializing the parser and reading data, as if new data has been found and the state has changed """ file_path = os.path.join(RESOURCE_PATH, '11079364_WC_HMR.txt') stream_handle = open(file_path, 'r') # 11079419_PPB_OCR_20.yml has the metadata and the first 19 # engineering particles in it expected_results = self.get_dict_from_yml('11079364_WC_HMR_recov.yml') parser = WcHmrCsppParser( self.config.get(WcHmrDataTypeKey.WC_HMR_CSPP_RECOVERED), None, stream_handle, self.state_callback, self.pub_callback, self.exception_callback) particles = parser.get_records(2) log.debug("Num particles: %s", len(particles)) self.assertTrue(len(particles) == 2) for i in range(len(particles)): self.assert_result(expected_results['data'][i], particles[i]) # position 945 is the byte at the start of the 18th data record new_state = {StateKey.POSITION: 945, StateKey.METADATA_EXTRACTED: True} parser.set_state(new_state) particles = parser.get_records(2) self.assertTrue(len(particles) == 2) # offset in the expected results offset = 18 for i in range(len(particles)): self.assert_result(expected_results['data'][i + offset], particles[i]) stream_handle.close()
def test_set_state(self): """ Test changing to a new state after initializing the parser and reading data, as if new data has been found and the state has changed """ file_path = os.path.join(RESOURCE_PATH, '11079364_WC_HMR.txt') stream_handle = open(file_path, 'r') # 11079419_PPB_OCR_20.yml has the metadata and the first 19 # engineering particles in it expected_results = self.get_dict_from_yml('11079364_WC_HMR_recov.yml') parser = WcHmrCsppParser(self.config.get(WcHmrDataTypeKey.WC_HMR_CSPP_RECOVERED), None, stream_handle, self.state_callback, self.pub_callback, self.exception_callback) particles = parser.get_records(2) log.debug("Num particles: %s", len(particles)) self.assertTrue(len(particles) == 2) for i in range(len(particles)): self.assert_result(expected_results['data'][i], particles[i]) # position 945 is the byte at the start of the 18th data record new_state = {StateKey.POSITION: 945, StateKey.METADATA_EXTRACTED: True} parser.set_state(new_state) particles = parser.get_records(2) self.assertTrue(len(particles) == 2) # offset in the expected results offset = 18 for i in range(len(particles)): self.assert_result(expected_results['data'][i + offset], particles[i]) stream_handle.close()