X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;f=rwlaunchpad%2Fplugins%2Frwimagemgr%2Ftest%2Futest_dts_handlers.py;h=108da8650fc80d9939d3392ffa35dab570141b88;hb=ba63fbf8a4f63b9525041a1f93fd2b6458e37933;hp=88205bdad42c6d3d63e31d4de9d18f8e3a935bf3;hpb=9ad945aab0b5a992e1df860bede8ecc9b143470e;p=osm%2FSO.git diff --git a/rwlaunchpad/plugins/rwimagemgr/test/utest_dts_handlers.py b/rwlaunchpad/plugins/rwimagemgr/test/utest_dts_handlers.py index 88205bda..108da865 100755 --- a/rwlaunchpad/plugins/rwimagemgr/test/utest_dts_handlers.py +++ b/rwlaunchpad/plugins/rwimagemgr/test/utest_dts_handlers.py @@ -45,6 +45,7 @@ import rift.test.dts from rift.tasklets.rwimagemgr import tasklet from rift.tasklets.rwimagemgr import upload +from rift.mano.utils.project import ManoProject, DEFAULT_PROJECT from rift.test.dts import async_test @@ -76,15 +77,16 @@ class RwImageRPCTestCase(rift.test.dts.AbstractDTSTest): self.log.debug("STARTING - %s", self.id()) self.tinfo = self.new_tinfo(self.id()) self.dts = rift.tasklets.DTS(self.tinfo, self.schema, self.loop) + self.project = ManoProject(self.log, name=DEFAULT_PROJECT) self.task_creator_mock = create_upload_task_creator_mock() self.job_controller_mock = create_job_controller_mock() self.rpc_handler = tasklet.ImageDTSRPCHandler( self.log, self.loop, self.dts, {'mock', None}, object(), self.task_creator_mock, - self.job_controller_mock + self.job_controller_mock, self.project, ) self.show_handler = tasklet.ImageDTSShowHandler( - self.log, self.loop, self.dts, self.job_controller_mock + self.log, self.loop, self.dts, self.job_controller_mock, self.project, ) self.tinfo_c = self.new_tinfo(self.id() + "_client") @@ -119,7 +121,8 @@ class RwImageRPCTestCase(rift.test.dts.AbstractDTSTest): "onboarded_image": { "image_name": upload_task.image_name, "image_checksum": upload_task.image_checksum, - } + }, + "project_name": self.project.name, }) query_iter = yield from self.dts_c.query_rpc( @@ -138,7 +141,7 @@ class RwImageRPCTestCase(rift.test.dts.AbstractDTSTest): ) query_iter = yield from self.dts_c.query_read( - "D,/rw-project:project/rw-image-mgmt:upload-jobs", + self.project.add_project("D,/rw-image-mgmt:upload-jobs"), ) for fut_resp in query_iter: