def setUp(self):
events.unsub_all()
parser = AstromParser()
directory_manager = LocalDirectoryWorkingContext(
self._get_working_dir())
self.context = directory_manager
progress_manager = LocalProgressManager(directory_manager)
workunit_provider = WorkUnitProvider(
tasks.get_suffix(self._get_task()), directory_manager,
progress_manager,
self._get_workunit_builder(parser, progress_manager))
self.workunit_provider = workunit_provider
self.progress_manager = progress_manager
self.singlet_download_manager = Mock(spec=AsynchronousDownloadManager)
self.triplet_download_manager = Mock(spec=AsynchronousDownloadManager)
self.image_manager = ImageManager(self.singlet_download_manager,
self.triplet_download_manager)
self.model = ValidationModel(self.workunit_provider,
self.image_manager, None)
self.custom_setup()
self.model.start_work()
def setUp(self):
self.progress_manager = InMemoryProgressManager(
Mock(spec=LocalDirectoryWorkingContext))
self.workunit_provider = Mock(spec=WorkUnitProvider)
self.workunit1 = MagicMock(spec=WorkUnit)
self.file1 = "file1"
self.workunit1.get_filename.return_value = self.file1
self.workunit2 = MagicMock(spec=WorkUnit)
self.file2 = "file2"
self.workunit2.get_filename.return_value = self.file2
workunits = [self.workunit1, self.workunit2]
def get_workunit(index):
workunit = workunits[index]
self.progress_manager.lock(workunit.get_filename())
return workunit
self.workunit_provider.get_workunit.side_effect = (
get_workunit(index) for index in range(2))
image_manager = Mock(spec=ImageManager)
self.undertest = ValidationModel(self.workunit_provider, image_manager,
None)
self.undertest.start_work()
def setUp(self):
events.unsub_all()
self.workunit_provider = Mock(spec=PreFetchingWorkUnitProvider)
self.image_manager = Mock(spec=ImageManager)
self.synchronization_manager = Mock(spec=SynchronizationManager)
self.model = ValidationModel(self.workunit_provider,
self.image_manager,
self.synchronization_manager)
def setUp(self):
WxWidgetTestCase.setUp(self)
parser = AstromParser()
context = LocalDirectoryWorkingContext(
self.get_abs_path("data/controller_testdir"))
progress_manager = LocalProgressManager(context)
workunit_provider = WorkUnitProvider(tasks.get_suffix(tasks.REALS_TASK),
context, progress_manager,
RealsWorkUnitBuilder(
parser,
context,
context,
progress_manager))
image_manager = Mock(spec=ImageManager)
self.model = ValidationModel(workunit_provider, image_manager, None)
self.model.start_work()
self.model.get_writer = Mock(return_value=Mock(spec=mpc.MPCWriter))
# We don't actually have any images loaded, so mock this out
source_cutout = Mock(spec=SourceCutout)
source_cutout.pixel_x = 11
source_cutout.pixel_y = 50
self.model.get_current_cutout = Mock(return_value=source_cutout)
x_cen = 10
y_cen = 50
mag = 24.0
self.model.get_current_source_observed_magnitude = Mock(return_value=(x_cen, y_cen, mag))
self.model.is_current_source_adjusted = Mock(return_value=False)
self.model.get_current_fits_header = Mock()
self.name_generator = Mock(spec=ProvisionalNameGenerator)
self.name_generator.generate_name.return_value = TEST_PROVISIONAL_NAME
class TestFactory(object):
def __init__(self, model, name_generator):
self.model = model
self.name_generator = name_generator
def create_controller(self, view):
return ProcessRealsController(self.model, view,
self.name_generator)
self.view = ApplicationView(
TestFactory(self.model, self.name_generator))
self.controller = self.view.controller
self.controller.display_current_image = Mock()
