diff options
Diffstat (limited to 'troveclient/tests/osc/v1/test_database_instances.py')
| -rw-r--r-- | troveclient/tests/osc/v1/test_database_instances.py | 11 |
1 files changed, 6 insertions, 5 deletions
diff --git a/troveclient/tests/osc/v1/test_database_instances.py b/troveclient/tests/osc/v1/test_database_instances.py index 62c9a25..38893bd 100644 --- a/troveclient/tests/osc/v1/test_database_instances.py +++ b/troveclient/tests/osc/v1/test_database_instances.py @@ -410,14 +410,15 @@ class TestDatabaseInstanceResizeFlavor(TestInstances): @mock.patch.object(utils, 'find_resource') def test_instance_resize_flavor(self, mock_find): args = ['instance1', 'flavor_id'] - mock_find.return_value = 'fake_instance_id' - + mock_find.side_effect = [ + mock.MagicMock(id='fake_instance_id'), + mock.MagicMock(id='fake_flavor_id') + ] parsed_args = self.check_parser(self.cmd, args, []) - result = self.cmd.take_action(parsed_args) + self.cmd.take_action(parsed_args) self.instance_client.resize_instance.assert_called_with( - 'fake_instance_id', 'flavor_id') - self.assertIsNone(result) + 'fake_instance_id', 'fake_flavor_id') class TestDatabaseInstanceUpgrade(TestInstances): |
