def setUp(self):
self.process = MagicMock()
self.child = MagicMock()
def getitem(name):
return name
self.child.__getitem__.side_effect = getitem
self.params = MagicMock()
self.params.readoutTime = 0.002
self.process.get_block.return_value = self.child
self.o = PandABoxDriverPart(self.process, self.params)
list(self.o.create_attributes())
def setUp(self):
self.process = MagicMock()
self.child = MagicMock()
def getitem(name):
return name
self.child.__getitem__.side_effect = getitem
self.params = MagicMock()
self.params.readoutTime = 0.002
self.process.get_block.return_value = self.child
self.o = PandABoxDriverPart(self.process, self.params)
list(self.o.create_attributes())
class TestPandaABoxDriverPart(unittest.TestCase):
def setUp(self):
self.process = MagicMock()
self.child = MagicMock()
def getitem(name):
return name
self.child.__getitem__.side_effect = getitem
self.params = MagicMock()
self.params.readoutTime = 0.002
self.process.get_block.return_value = self.child
self.o = PandABoxDriverPart(self.process, self.params)
list(self.o.create_attributes())
def test_configure(self):
task = MagicMock()
completed_steps = 0
steps_to_do = 6
part_info = ANY
self.o.configure(task, completed_steps, steps_to_do, part_info)
task.put_many.assert_called_once_with(self.child, dict(
imageMode="Multiple",
numImages=steps_to_do,
arrayCounter=completed_steps,
arrayCallbacks=True))
task.post_async.assert_called_once_with(self.child["start"])
def test_run(self):
task = MagicMock()
update_completed_steps = MagicMock()
self.o.start_future = MagicMock()
self.o.run(task, update_completed_steps)
task.subscribe.assert_called_once_with(
self.child["arrayCounter"], update_completed_steps, self.o)
task.wait_all.assert_called_once_with(self.o.start_future)
def test_abort(self):
task = MagicMock()
self.o.abort(task)
task.post.assert_called_once_with(self.child["stop"])
class TestPandaABoxDriverPart(unittest.TestCase):
def setUp(self):
self.process = MagicMock()
self.child = MagicMock()
def getitem(name):
return name
self.child.__getitem__.side_effect = getitem
self.params = MagicMock()
self.params.readoutTime = 0.002
self.process.get_block.return_value = self.child
self.o = PandABoxDriverPart(self.process, self.params)
list(self.o.create_attributes())
def test_configure(self):
task = MagicMock()
completed_steps = 0
steps_to_do = 6
part_info = ANY
self.o.configure(task, completed_steps, steps_to_do, part_info)
task.put_many.assert_called_once_with(
self.child,
dict(imageMode="Multiple",
numImages=steps_to_do,
arrayCounter=completed_steps,
arrayCallbacks=True))
task.post_async.assert_called_once_with(self.child["start"])
def test_run(self):
task = MagicMock()
update_completed_steps = MagicMock()
self.o.start_future = MagicMock()
self.o.run(task, update_completed_steps)
task.subscribe.assert_called_once_with(self.child["arrayCounter"],
update_completed_steps, self.o)
task.wait_all.assert_called_once_with(self.o.start_future)
def test_abort(self):
task = MagicMock()
self.o.abort(task)
task.post.assert_called_once_with(self.child["stop"])
