@@ -943,17 +943,17 @@ def test_create_cluster_with_keypair(self):
943943 response = self .post_json ('/clusters' , bdict )
944944 self .assertEqual ('application/json' , response .content_type )
945945 self .assertEqual (202 , response .status_int )
946- cluster , timeout = self .mock_cluster_create .call_args
947- self .assertEqual ('keypair2' , cluster [ 0 ] .keypair )
946+ cluster = self .mock_cluster_create .call_args . args [ 0 ]
947+ self .assertEqual ('keypair2' , cluster .keypair )
948948
949949 def test_create_cluster_without_keypair (self ):
950950 bdict = apiutils .cluster_post_data ()
951951 response = self .post_json ('/clusters' , bdict )
952952 self .assertEqual ('application/json' , response .content_type )
953953 self .assertEqual (202 , response .status_int )
954- cluster , timeout = self .mock_cluster_create .call_args
954+ cluster = self .mock_cluster_create .call_args . args [ 0 ]
955955 # Verify keypair from ClusterTemplate is used
956- self .assertEqual ('keypair1' , cluster [ 0 ] .keypair )
956+ self .assertEqual ('keypair1' , cluster .keypair )
957957
958958 def test_create_cluster_with_multi_keypair_same_name (self ):
959959 bdict = apiutils .cluster_post_data ()
@@ -969,17 +969,17 @@ def test_create_cluster_with_docker_volume_size(self):
969969 response = self .post_json ('/clusters' , bdict )
970970 self .assertEqual ('application/json' , response .content_type )
971971 self .assertEqual (202 , response .status_int )
972- cluster , timeout = self .mock_cluster_create .call_args
973- self .assertEqual (3 , cluster [ 0 ] .docker_volume_size )
972+ cluster = self .mock_cluster_create .call_args . args [ 0 ]
973+ self .assertEqual (3 , cluster .docker_volume_size )
974974
975975 def test_create_cluster_with_labels (self ):
976976 bdict = apiutils .cluster_post_data ()
977977 bdict ['labels' ] = {'key' : 'value' }
978978 response = self .post_json ('/clusters' , bdict )
979979 self .assertEqual ('application/json' , response .content_type )
980980 self .assertEqual (202 , response .status_int )
981- cluster , timeout = self .mock_cluster_create .call_args
982- self .assertEqual ({'key' : 'value' }, cluster [ 0 ] .labels )
981+ cluster = self .mock_cluster_create .call_args . args [ 0 ]
982+ self .assertEqual ({'key' : 'value' }, cluster .labels )
983983
984984 def test_create_cluster_without_docker_volume_size (self ):
985985 bdict = apiutils .cluster_post_data ()
@@ -988,19 +988,19 @@ def test_create_cluster_without_docker_volume_size(self):
988988 response = self .post_json ('/clusters' , bdict )
989989 self .assertEqual ('application/json' , response .content_type )
990990 self .assertEqual (202 , response .status_int )
991- cluster , timeout = self .mock_cluster_create .call_args
991+ cluster = self .mock_cluster_create .call_args . args [ 0 ]
992992 # Verify docker_volume_size from ClusterTemplate is used
993- self .assertEqual (20 , cluster [ 0 ] .docker_volume_size )
993+ self .assertEqual (20 , cluster .docker_volume_size )
994994
995995 def test_create_cluster_without_labels (self ):
996996 bdict = apiutils .cluster_post_data ()
997997 bdict .pop ('labels' )
998998 response = self .post_json ('/clusters' , bdict )
999999 self .assertEqual ('application/json' , response .content_type )
10001000 self .assertEqual (202 , response .status_int )
1001- cluster , timeout = self .mock_cluster_create .call_args
1001+ cluster = self .mock_cluster_create .call_args . args [ 0 ]
10021002 # Verify labels from ClusterTemplate is used
1003- self .assertEqual ({'key1' : u'val1' , 'key2' : u'val2' }, cluster [ 0 ] .labels )
1003+ self .assertEqual ({'key1' : u'val1' , 'key2' : u'val2' }, cluster .labels )
10041004
10051005 def test_create_cluster_with_invalid_docker_volume_size (self ):
10061006 invalid_values = [(- 1 , None ), ('notanint' , None ),
@@ -1026,35 +1026,35 @@ def test_create_cluster_with_master_flavor_id(self):
10261026 response = self .post_json ('/clusters' , bdict )
10271027 self .assertEqual ('application/json' , response .content_type )
10281028 self .assertEqual (202 , response .status_int )
1029- cluster , timeout = self .mock_cluster_create .call_args
1030- self .assertEqual ('m2.small' , cluster [ 0 ] .master_flavor_id )
1029+ cluster = self .mock_cluster_create .call_args . args [ 0 ]
1030+ self .assertEqual ('m2.small' , cluster .master_flavor_id )
10311031
10321032 def test_create_cluster_without_master_flavor_id (self ):
10331033 bdict = apiutils .cluster_post_data ()
10341034 response = self .post_json ('/clusters' , bdict )
10351035 self .assertEqual ('application/json' , response .content_type )
10361036 self .assertEqual (202 , response .status_int )
1037- cluster , timeout = self .mock_cluster_create .call_args
1037+ cluster = self .mock_cluster_create .call_args . args [ 0 ]
10381038 # Verify master_flavor_id from ClusterTemplate is used
1039- self .assertEqual ('m1.small' , cluster [ 0 ] .master_flavor_id )
1039+ self .assertEqual ('m1.small' , cluster .master_flavor_id )
10401040
10411041 def test_create_cluster_with_flavor_id (self ):
10421042 bdict = apiutils .cluster_post_data ()
10431043 bdict ['flavor_id' ] = 'm2.small'
10441044 response = self .post_json ('/clusters' , bdict )
10451045 self .assertEqual ('application/json' , response .content_type )
10461046 self .assertEqual (202 , response .status_int )
1047- cluster , timeout = self .mock_cluster_create .call_args
1048- self .assertEqual ('m2.small' , cluster [ 0 ] .flavor_id )
1047+ cluster = self .mock_cluster_create .call_args . args [ 0 ]
1048+ self .assertEqual ('m2.small' , cluster .flavor_id )
10491049
10501050 def test_create_cluster_without_flavor_id (self ):
10511051 bdict = apiutils .cluster_post_data ()
10521052 response = self .post_json ('/clusters' , bdict )
10531053 self .assertEqual ('application/json' , response .content_type )
10541054 self .assertEqual (202 , response .status_int )
1055- cluster , timeout = self .mock_cluster_create .call_args
1055+ cluster = self .mock_cluster_create .call_args . args [ 0 ]
10561056 # Verify flavor_id from ClusterTemplate is used
1057- self .assertEqual ('m1.small' , cluster [ 0 ] .flavor_id )
1057+ self .assertEqual ('m1.small' , cluster .flavor_id )
10581058
10591059 def test_create_cluster_with_cinder_csi_disabled (self ):
10601060 self .cluster_template .volume_driver = 'cinder'
@@ -1075,8 +1075,8 @@ def test_create_cluster_without_merge_labels(self):
10751075 response = self .post_json ('/clusters' , bdict )
10761076 self .assertEqual ('application/json' , response .content_type )
10771077 self .assertEqual (202 , response .status_int )
1078- cluster , timeout = self .mock_cluster_create .call_args
1079- self .assertEqual (cluster_labels , cluster [ 0 ] .labels )
1078+ cluster = self .mock_cluster_create .call_args . args [ 0 ]
1079+ self .assertEqual (cluster_labels , cluster .labels )
10801080
10811081 def test_create_cluster_with_merge_labels (self ):
10821082 self .cluster_template .labels = {'label1' : 'value1' , 'label2' : 'value2' }
@@ -1087,10 +1087,10 @@ def test_create_cluster_with_merge_labels(self):
10871087 response = self .post_json ('/clusters' , bdict )
10881088 self .assertEqual ('application/json' , response .content_type )
10891089 self .assertEqual (202 , response .status_int )
1090- cluster , timeout = self .mock_cluster_create .call_args
1090+ cluster = self .mock_cluster_create .call_args . args [ 0 ]
10911091 expected = self .cluster_template .labels
10921092 expected .update (cluster_labels )
1093- self .assertEqual (expected , cluster [ 0 ] .labels )
1093+ self .assertEqual (expected , cluster .labels )
10941094
10951095 def test_create_cluster_with_merge_labels_no_labels (self ):
10961096 self .cluster_template .labels = {'label1' : 'value1' , 'label2' : 'value2' }
@@ -1100,8 +1100,8 @@ def test_create_cluster_with_merge_labels_no_labels(self):
11001100 response = self .post_json ('/clusters' , bdict )
11011101 self .assertEqual ('application/json' , response .content_type )
11021102 self .assertEqual (202 , response .status_int )
1103- cluster , timeout = self .mock_cluster_create .call_args
1104- self .assertEqual (self .cluster_template .labels , cluster [ 0 ] .labels )
1103+ cluster = self .mock_cluster_create .call_args . args [ 0 ]
1104+ self .assertEqual (self .cluster_template .labels , cluster .labels )
11051105
11061106
11071107class TestDelete (api_base .FunctionalTest ):
0 commit comments