@@ -107,11 +107,7 @@ def setUp(self):
107107
108108 self .servers_mock .create .return_value = self .new_server
109109
110- self .image = fakes .FakeResource (
111- None ,
112- copy .deepcopy (image_fakes .IMAGE ),
113- loaded = True ,
114- )
110+ self .image = image_fakes .FakeImage .create_one_image ()
115111 self .cimages_mock .get .return_value = self .image
116112
117113 self .flavor = compute_fakes .FakeFlavor .create_one_flavor ()
@@ -537,13 +533,9 @@ def setUp(self):
537533 # This is the return value for utils.find_resource()
538534 self .servers_mock .get .return_value = self .server
539535
540- self .servers_mock .create_image .return_value = image_fakes .image_id
541-
542- self .images_mock .get .return_value = fakes .FakeResource (
543- None ,
544- copy .deepcopy (image_fakes .IMAGE ),
545- loaded = True ,
546- )
536+ self .image = image_fakes .FakeImage .create_one_image ()
537+ self .images_mock .get .return_value = self .image
538+ self .servers_mock .create_image .return_value = self .image .id
547539
548540 # Get the command object to test
549541 self .cmd = server .CreateServerImage (self .app , None )
@@ -569,12 +561,12 @@ def test_server_image_create_no_options(self):
569561 collist = ('id' , 'name' , 'owner' , 'protected' , 'tags' , 'visibility' )
570562 self .assertEqual (collist , columns )
571563 datalist = (
572- image_fakes . image_id ,
573- image_fakes . image_name ,
574- image_fakes . image_owner ,
575- image_fakes . image_protected ,
576- image_fakes . image_tags ,
577- image_fakes . image_visibility ,
564+ self . image . id ,
565+ self . image . name ,
566+ self . image . owner ,
567+ self . image . protected ,
568+ self . image . tags ,
569+ self . image . visibility ,
578570 )
579571 self .assertEqual (datalist , data )
580572
@@ -601,12 +593,12 @@ def test_server_image_create_name(self):
601593 collist = ('id' , 'name' , 'owner' , 'protected' , 'tags' , 'visibility' )
602594 self .assertEqual (collist , columns )
603595 datalist = (
604- image_fakes . image_id ,
605- image_fakes . image_name ,
606- image_fakes . image_owner ,
607- image_fakes . image_protected ,
608- image_fakes . image_tags ,
609- image_fakes . image_visibility ,
596+ self . image . id ,
597+ self . image . name ,
598+ self . image . owner ,
599+ self . image . protected ,
600+ self . image . tags ,
601+ self . image . visibility ,
610602 )
611603 self .assertEqual (datalist , data )
612604
0 commit comments