Cluster upgrade extension for Fuel
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

test_extension.py 1.1KB

1234567891011121314151617181920212223242526272829
  1. # -*- coding: utf-8 -*-
  2. # Copyright 2015 Mirantis, Inc.
  3. #
  4. # Licensed under the Apache License, Version 2.0 (the "License"); you may
  5. # not use this file except in compliance with the License. You may obtain
  6. # a copy of the License at
  7. #
  8. # http://www.apache.org/licenses/LICENSE-2.0
  9. #
  10. # Unless required by applicable law or agreed to in writing, software
  11. # distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
  12. # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
  13. # License for the specific language governing permissions and limitations
  14. # under the License.
  15. import mock
  16. from nailgun.test.base import BaseTestCase
  17. from .. import extension
  18. from ..objects import relations
  19. class TestExtension(BaseTestCase):
  20. @mock.patch.object(relations.UpgradeRelationObject, "delete_relation")
  21. def test_on_cluster_delete(self, mock_on_cluster_delete):
  22. cluster = mock.Mock(id=42)
  23. extension.ClusterUpgradeExtension.on_cluster_delete(cluster)
  24. mock_on_cluster_delete.assert_called_once_with(42)