27
27
from oslo_utils import uuidutils
28
28
import six
29
29
30
- from nova .compute import flavors
31
30
from nova .compute import manager
32
31
from nova .compute import power_state
33
32
from nova .compute import task_states
60
59
61
60
def create_instance (context , user_id = 'fake' , project_id = 'fake' , params = None ):
62
61
"""Create a test instance."""
63
- flavor = flavors . get_flavor_by_name ( 'm1.tiny' )
62
+ flavor = objects . Flavor . get_by_name ( context , 'm1.tiny' )
64
63
net_info = model .NetworkInfo ([])
65
64
info_cache = objects .InstanceInfoCache (network_info = net_info )
66
65
inst = objects .Instance (context = context ,
@@ -410,6 +409,7 @@ def fake_get_nw_info(cls, ctxt, instance):
410
409
self .user_id = 'fake'
411
410
self .project_id = 'fake'
412
411
self .context = context .RequestContext (self .user_id , self .project_id )
412
+ self .flavor = objects .Flavor .get_by_name (self .context , 'm1.tiny' )
413
413
414
414
def fake_show (meh , context , id , ** kwargs ):
415
415
return {'id' : 1 , 'properties' : {'kernel_id' : 1 , 'ramdisk_id' : 1 }}
@@ -440,9 +440,9 @@ def test_notify_usage_exists(self):
440
440
self .assertEqual (payload ['user_id' ], self .user_id )
441
441
self .assertEqual (payload ['instance_id' ], instance ['uuid' ])
442
442
self .assertEqual (payload ['instance_type' ], 'm1.tiny' )
443
- type_id = flavors . get_flavor_by_name ( 'm1.tiny' )[ 'id' ]
443
+ type_id = self . flavor . id
444
444
self .assertEqual (str (payload ['instance_type_id' ]), str (type_id ))
445
- flavor_id = flavors . get_flavor_by_name ( 'm1.tiny' )[ ' flavorid' ]
445
+ flavor_id = self . flavor . flavorid
446
446
self .assertEqual (str (payload ['instance_flavor_id' ]), str (flavor_id ))
447
447
for attr in ('display_name' , 'created_at' , 'launched_at' ,
448
448
'state' , 'state_description' ,
@@ -473,7 +473,7 @@ def test_notify_usage_exists_emits_versioned(self):
473
473
self .assertEqual (payload ['uuid' ], instance ['uuid' ])
474
474
flavor = payload ['flavor' ]['nova_object.data' ]
475
475
self .assertEqual (flavor ['name' ], 'm1.tiny' )
476
- flavorid = flavors . get_flavor_by_name ( 'm1.tiny' )[ ' flavorid' ]
476
+ flavorid = self . flavor . flavorid
477
477
self .assertEqual (str (flavor ['flavorid' ]), str (flavorid ))
478
478
479
479
for attr in ('display_name' , 'created_at' , 'launched_at' ,
@@ -504,9 +504,9 @@ def test_notify_usage_exists_deleted_instance(self):
504
504
self .assertEqual (payload ['user_id' ], self .user_id )
505
505
self .assertEqual (payload ['instance_id' ], instance ['uuid' ])
506
506
self .assertEqual (payload ['instance_type' ], 'm1.tiny' )
507
- type_id = flavors . get_flavor_by_name ( 'm1.tiny' )[ 'id' ]
507
+ type_id = self . flavor . id
508
508
self .assertEqual (str (payload ['instance_type_id' ]), str (type_id ))
509
- flavor_id = flavors . get_flavor_by_name ( 'm1.tiny' )[ ' flavorid' ]
509
+ flavor_id = self . flavor . flavorid
510
510
self .assertEqual (str (payload ['instance_flavor_id' ]), str (flavor_id ))
511
511
for attr in ('display_name' , 'created_at' , 'launched_at' ,
512
512
'state' , 'state_description' ,
@@ -552,9 +552,9 @@ def test_notify_about_instance_action(self):
552
552
self .assertEqual (payload ['user_id' ], self .user_id )
553
553
self .assertEqual (payload ['uuid' ], instance ['uuid' ])
554
554
555
- flavorid = flavors . get_flavor_by_name ( 'm1.tiny' )[ 'flavorid' ]
556
- flavor = payload [ ' flavor' ][ 'nova_object.data' ]
557
- self . assertEqual ( str (flavor [ ' flavorid' ]), flavorid )
555
+ self . assertEqual (
556
+ self . flavor . flavorid ,
557
+ str (payload [ ' flavor' ][ 'nova_object.data' ][ ' flavorid' ]))
558
558
559
559
for attr in ('display_name' , 'created_at' , 'launched_at' ,
560
560
'state' , 'task_state' , 'display_description' , 'locked' ,
@@ -598,7 +598,7 @@ def test_notify_about_instance_create(self):
598
598
self .assertEqual ('fake' , payload ['user_id' ])
599
599
self .assertEqual (instance ['uuid' ], payload ['uuid' ])
600
600
601
- flavorid = flavors . get_flavor_by_name ( 'm1.tiny' )[ ' flavorid' ]
601
+ flavorid = self . flavor . flavorid
602
602
flavor = payload ['flavor' ]['nova_object.data' ]
603
603
self .assertEqual (flavorid , str (flavor ['flavorid' ]))
604
604
@@ -641,9 +641,9 @@ def test_notify_about_instance_create_without_keypair(self):
641
641
self .assertEqual ('fake' , payload ['user_id' ])
642
642
self .assertEqual (instance ['uuid' ], payload ['uuid' ])
643
643
644
- flavorid = flavors . get_flavor_by_name ( 'm1.tiny' )[ 'flavorid' ]
645
- flavor = payload [ ' flavor' ][ 'nova_object.data' ]
646
- self . assertEqual ( flavorid , str (flavor ['flavorid' ]))
644
+ self . assertEqual (
645
+ self . flavor . flavorid ,
646
+ str (payload [ ' flavor' ][ 'nova_object.data' ] ['flavorid' ]))
647
647
648
648
self .assertEqual (0 , len (payload ['keypairs' ]))
649
649
for attr in ('display_name' , 'created_at' , 'launched_at' ,
@@ -680,9 +680,9 @@ def test_notify_about_instance_create_with_tags(self):
680
680
self .assertEqual ('fake' , payload ['user_id' ])
681
681
self .assertEqual (instance .uuid , payload ['uuid' ])
682
682
683
- flavorid = flavors . get_flavor_by_name ( 'm1.tiny' )[ 'flavorid' ]
684
- flavor = payload [ ' flavor' ][ 'nova_object.data' ]
685
- self . assertEqual ( flavorid , str (flavor ['flavorid' ]))
683
+ self . assertEqual (
684
+ self . flavor . flavorid ,
685
+ str (payload [ ' flavor' ][ 'nova_object.data' ] ['flavorid' ]))
686
686
687
687
self .assertEqual (0 , len (payload ['keypairs' ]))
688
688
for attr in ('display_name' , 'created_at' , 'launched_at' ,
@@ -718,9 +718,9 @@ def test_notify_about_volume_swap(self):
718
718
self .assertEqual (self .user_id , payload ['user_id' ])
719
719
self .assertEqual (instance ['uuid' ], payload ['uuid' ])
720
720
721
- flavorid = flavors . get_flavor_by_name ( 'm1.tiny' )[ 'flavorid' ]
722
- flavor = payload [ ' flavor' ][ 'nova_object.data' ]
723
- self . assertEqual ( flavorid , str (flavor ['flavorid' ]))
721
+ self . assertEqual (
722
+ self . flavor . flavorid ,
723
+ str (payload [ ' flavor' ][ 'nova_object.data' ] ['flavorid' ]))
724
724
725
725
for attr in ('display_name' , 'created_at' , 'launched_at' ,
726
726
'state' , 'task_state' ):
@@ -760,9 +760,9 @@ def test_notify_about_volume_swap_with_error(self):
760
760
self .assertEqual (self .user_id , payload ['user_id' ])
761
761
self .assertEqual (instance ['uuid' ], payload ['uuid' ])
762
762
763
- flavorid = flavors . get_flavor_by_name ( 'm1.tiny' )[ 'flavorid' ]
764
- flavor = payload [ ' flavor' ][ 'nova_object.data' ]
765
- self . assertEqual ( flavorid , str (flavor ['flavorid' ]))
763
+ self . assertEqual (
764
+ self . flavor . flavorid ,
765
+ str (payload [ ' flavor' ][ 'nova_object.data' ] ['flavorid' ]))
766
766
767
767
for attr in ('display_name' , 'created_at' , 'launched_at' ,
768
768
'state' , 'task_state' ):
@@ -808,9 +808,9 @@ def test_notify_about_instance_rescue_action(self):
808
808
self .assertEqual (payload ['user_id' ], self .user_id )
809
809
self .assertEqual (payload ['uuid' ], instance ['uuid' ])
810
810
811
- flavorid = flavors . get_flavor_by_name ( 'm1.tiny' )[ 'flavorid' ]
812
- flavor = payload [ ' flavor' ][ 'nova_object.data' ]
813
- self . assertEqual ( str (flavor [ ' flavorid' ]), flavorid )
811
+ self . assertEqual (
812
+ self . flavor . flavorid ,
813
+ str (payload [ ' flavor' ][ 'nova_object.data' ][ ' flavorid' ]))
814
814
815
815
for attr in ('display_name' , 'created_at' , 'launched_at' ,
816
816
'state' , 'task_state' , 'display_description' , 'locked' ,
@@ -823,7 +823,7 @@ def test_notify_about_instance_rescue_action(self):
823
823
def test_notify_about_resize_prep_instance (self ):
824
824
instance = create_instance (self .context )
825
825
826
- new_flavor = flavors . get_flavor_by_name ( 'm1.small' )
826
+ new_flavor = objects . Flavor . get_by_name ( self . context , 'm1.small' )
827
827
828
828
compute_utils .notify_about_resize_prep_instance (
829
829
self .context , instance , 'fake-compute' , 'start' , new_flavor )
@@ -842,9 +842,9 @@ def test_notify_about_resize_prep_instance(self):
842
842
self .assertEqual (payload ['user_id' ], self .user_id )
843
843
self .assertEqual (payload ['uuid' ], instance ['uuid' ])
844
844
845
- flavorid = flavors . get_flavor_by_name ( 'm1.tiny' )[ 'flavorid' ]
846
- flavor = payload [ ' flavor' ][ 'nova_object.data' ]
847
- self . assertEqual ( str (flavor [ ' flavorid' ]), flavorid )
845
+ self . assertEqual (
846
+ self . flavor . flavorid ,
847
+ str (payload [ ' flavor' ][ 'nova_object.data' ][ ' flavorid' ]))
848
848
849
849
for attr in ('display_name' , 'created_at' , 'launched_at' ,
850
850
'state' , 'task_state' , 'display_description' , 'locked' ,
@@ -869,10 +869,10 @@ def test_notify_usage_exists_instance_not_found(self):
869
869
self .assertEqual (payload ['user_id' ], self .user_id )
870
870
self .assertEqual (payload ['instance_id' ], instance ['uuid' ])
871
871
self .assertEqual (payload ['instance_type' ], 'm1.tiny' )
872
- type_id = flavors . get_flavor_by_name ( 'm1.tiny' )[ 'id' ]
873
- self . assertEqual ( str (payload ['instance_type_id' ]), str ( type_id ))
874
- flavor_id = flavors . get_flavor_by_name ( 'm1.tiny' )[ ' flavorid' ]
875
- self . assertEqual ( str (payload ['instance_flavor_id' ]), str ( flavor_id ))
872
+ self . assertEqual ( str ( self . flavor . id ),
873
+ str (payload ['instance_type_id' ]))
874
+ self . assertEqual ( str ( self . flavor . flavorid ),
875
+ str (payload ['instance_flavor_id' ]))
876
876
for attr in ('display_name' , 'created_at' , 'launched_at' ,
877
877
'state' , 'state_description' ,
878
878
'bandwidth' , 'audit_period_beginning' ,
@@ -943,10 +943,9 @@ def test_notify_about_instance_usage(self):
943
943
self .assertEqual (payload ['user_id' ], self .user_id )
944
944
self .assertEqual (payload ['instance_id' ], instance ['uuid' ])
945
945
self .assertEqual (payload ['instance_type' ], 'm1.tiny' )
946
- type_id = flavors .get_flavor_by_name ('m1.tiny' )['id' ]
947
- self .assertEqual (str (payload ['instance_type_id' ]), str (type_id ))
948
- flavor_id = flavors .get_flavor_by_name ('m1.tiny' )['flavorid' ]
949
- self .assertEqual (str (payload ['instance_flavor_id' ]), str (flavor_id ))
946
+ self .assertEqual (str (self .flavor .id ), str (payload ['instance_type_id' ]))
947
+ self .assertEqual (str (self .flavor .flavorid ),
948
+ str (payload ['instance_flavor_id' ]))
950
949
for attr in ('display_name' , 'created_at' , 'launched_at' ,
951
950
'state' , 'state_description' , 'image_meta' ):
952
951
self .assertIn (attr , payload , "Key %s not in payload" % attr )
@@ -1299,8 +1298,8 @@ def setUp(self):
1299
1298
self .context = context .RequestContext ('fake' , 'fake' )
1300
1299
1301
1300
def test_upsize_quota_delta (self ):
1302
- old_flavor = flavors . get_flavor_by_name ( 'm1.tiny' )
1303
- new_flavor = flavors . get_flavor_by_name ( 'm1.medium' )
1301
+ old_flavor = objects . Flavor . get_by_name ( self . context , 'm1.tiny' )
1302
+ new_flavor = objects . Flavor . get_by_name ( self . context , 'm1.medium' )
1304
1303
1305
1304
expected_deltas = {
1306
1305
'cores' : new_flavor ['vcpus' ] - old_flavor ['vcpus' ],
0 commit comments