def __init__(self, filename, parsed_data, progress_manager, output_context, dry_run=False): self.filename = filename self.data = parsed_data self.progress_manager = progress_manager self.output_context = output_context self.dry_run = dry_run self.sources = StatefulCollection(parsed_data.get_sources()) self.readings_by_source = {} for source in self.sources: self.readings_by_source[source] = StatefulCollection( source.get_readings()) self.processed_items = set() self._mark_previously_processed_items() if self.is_current_item_processed(): self.next_item() self.finished_callbacks = [] self._unlocked = False
def __init__(self, workunit_provider, image_manager, synchronization_manager): self.workunit_provider = workunit_provider self.image_manager = image_manager self.synchronization_manager = synchronization_manager self.work_units = StatefulCollection() self.num_processed = 0 self.sources_discovered = set() self.image_state = SingletState(self)
def test_start_empty(self): undertest = StatefulCollection() assert_that(undertest, has_length(0)) assert_that(undertest.get_current_item(), is_(none())) assert_that(undertest.get_index(), equal_to(-1)) item1 = 1 undertest.append(item1) assert_that(undertest.get_current_item(), equal_to(item1)) assert_that(undertest.get_index(), equal_to(0)) item2 = 2 undertest.append(item2) next(undertest) assert_that(undertest.get_index(), equal_to(1)) assert_that(undertest.get_current_item(), equal_to(item2))
def test_basics(self): items = [1, 2, 3] undertest = StatefulCollection(items) assert_that(undertest[0], equal_to(items[0])) assert_that(undertest[1], equal_to(items[1])) assert_that(undertest[2], equal_to(items[2])) assert_that(undertest, has_length(3)) assert_that(undertest.get_index(), equal_to(0)) assert_that(undertest.get_current_item(), equal_to(items[0])) next(undertest) assert_that(undertest.get_index(), equal_to(1)) assert_that(undertest.get_current_item(), equal_to(items[1])) undertest.previous() assert_that(undertest.get_index(), equal_to(0)) assert_that(undertest.get_current_item(), equal_to(items[0]))