This fix is not necessary anymore in case "Fix Kimchi Model" is accepted.

On 14-10-2015 16:13, Aline Manera wrote: > Signed-off-by: Aline Manera <alinefm(a)linux.vnet.ibm.com&gt; > --- > src/wok/plugins/kimchi/mockmodel.py | 7 +++++-- > 1 file changed, 5 insertions(+), 2 deletions(-) > > diff --git a/src/wok/plugins/kimchi/mockmodel.py > b/src/wok/plugins/kimchi/mockmodel.py > index 0832b20..5b57bbd 100644 > --- a/src/wok/plugins/kimchi/mockmodel.py > +++ b/src/wok/plugins/kimchi/mockmodel.py > @@ -26,6 +26,7 @@ from lxml import objectify > from lxml.builder import E > > from wok.exception import NotFoundError, OperationFailed > +from wok.model.tasks import TaskModel > from wok.objectstore import ObjectStore > from wok.utils import add_task, get_next_clone_name, wok_log > from wok.xmlutils.utils import xml_item_update > @@ -99,6 +100,8 @@ class MockModel(Model): > super(MockModel, self).__init__('test:///default', > objstore_loc) > self.objstore_loc = objstore_loc > self.objstore = ObjectStore(objstore_loc) > + kargs = {'objstore': self.objstore} > + self.task = TaskModel(**kargs) > > # The MockModel methods are instantiated on runtime > according to Model > # and BaseModel > @@ -371,7 +374,7 @@ class MockModel(Model): > def _mock_host_swupdate(self, args=None): > task_id = add_task('/plugins/kimchi/host/swupdate', > self._mock_swupdate.doUpdate, > self.objstore) > - return self.task_lookup(task_id) > + return self.task.lookup(task_id) > > def _mock_repositories_get_list(self): > return self._mock_repositories.repos.keys() > @@ -424,7 +427,7 @@ class MockModel(Model): > taskid = add_task(u'/plugins/kimchi/vms/%s/snapshots/%s' % > (vm_name, name), > self._vmsnapshots_create_task, > self.objstore, params) > - return self.task_lookup(taskid) > + return self.task.lookup(taskid) > > def _vmsnapshots_create_task(self, cb, params): > vm_name = params['vm_name'] >