@@ -131,17 +131,21 @@ def fixture_aws_identity():
131
131
class TestAWSAPI :
132
132
def test_login (self , fixture_aws_mock_api , fixture_aws_identity ):
133
133
api = fixture_aws_mock_api
134
- with patch (
135
- "JciHitachi.aws_connection.GetUser.get_data"
136
- ) as mock_get_data_get_user , patch (
137
- "JciHitachi.aws_connection.GetUser.login"
138
- ) as mock_login_get_user , patch (
139
- "JciHitachi.aws_connection.ListSubUser.get_data"
140
- ) as mock_get_data_list_subuser , patch (
141
- "JciHitachi.aws_connection.GetAllDevice.get_data"
142
- ) as mock_get_data_get_all_device , patch (
143
- "JciHitachi.aws_connection.JciHitachiAWSMqttConnection.connect"
144
- ) as mock_connect_mqtt :
134
+ with (
135
+ patch (
136
+ "JciHitachi.aws_connection.GetUser.get_data"
137
+ ) as mock_get_data_get_user ,
138
+ patch ("JciHitachi.aws_connection.GetUser.login" ) as mock_login_get_user ,
139
+ patch (
140
+ "JciHitachi.aws_connection.ListSubUser.get_data"
141
+ ) as mock_get_data_list_subuser ,
142
+ patch (
143
+ "JciHitachi.aws_connection.GetAllDevice.get_data"
144
+ ) as mock_get_data_get_all_device ,
145
+ patch (
146
+ "JciHitachi.aws_connection.JciHitachiAWSMqttConnection.connect"
147
+ ) as mock_connect_mqtt ,
148
+ ):
145
149
# initialization
146
150
aws_tokens = api ._aws_tokens
147
151
api ._aws_tokens = None
@@ -182,11 +186,14 @@ def test_login(self, fixture_aws_mock_api, fixture_aws_identity):
182
186
183
187
def test_change_password (self , fixture_aws_mock_api ):
184
188
api = fixture_aws_mock_api
185
- with patch (
186
- "JciHitachi.aws_connection.ChangePassword.get_data"
187
- ) as mock_get_data_1 , patch (
188
- "JciHitachi.connection.UpdateUserCredential.get_data"
189
- ) as mock_get_data_2 :
189
+ with (
190
+ patch (
191
+ "JciHitachi.aws_connection.ChangePassword.get_data"
192
+ ) as mock_get_data_1 ,
193
+ patch (
194
+ "JciHitachi.connection.UpdateUserCredential.get_data"
195
+ ) as mock_get_data_2 ,
196
+ ):
190
197
mock_get_data_1 .return_value = ("OK" , "" )
191
198
mock_get_data_2 .return_value = ("OK" , "" )
192
199
api .change_password ("new_password" )
0 commit comments