def setUp(self):
super(TestUnitRemoveCommand, self).setUp()
self.command = UnitRemoveCommand(self.context)
self.mock_poll = mock.MagicMock().poll
self.command.poll = self.mock_poll
self.mock_remove_binding = mock.MagicMock().remove
self.mock_remove_binding.return_value = task_simulator.create_fake_task_response()
self.bindings.repo_unit.remove = self.mock_remove_binding