def use_cands_workunit(self):
self.workunit = CandidatesWorkUnit("file", self.data,
self.progress_manager,
self.output_context)
self.workunit.get_writer = Mock()
self.model.add_workunit(self.workunit)
assert_that(self.model.get_current_reading(),
equal_to(self.sources[0].get_reading(0)))
def setUp(self):
super(CandidatesWorkUnitTest, self).setUp()
self.workunit = CandidatesWorkUnit(self.testfile, self.data,
self.progress_manager,
self.output_context)
self.writer = Mock(spec=StreamingAstromWriter)
self.workunit.get_writer = Mock(return_value=self.writer)
