fup: Move _wait_for_volume_attach into InstanceHelperMixin

As discussed in I332d4f33ea6b9506cc24ac12e5c0994f208a3107 this is useful
enough to be copied into InstanceHelperMixin.

Most of this change is migrating the original functional test that
introduced _wait_for_volume_attach over to use _IntegratedTestBase.

Change-Id: I345dce93c6e2593c1ff1b863425f9a854c49ab9d

2 files changed, 8 insertions, 104 deletions

diff --git a/nova/tests/functional/regressions/test_bug_1675570.py b/nova/tests/functional/regressions/test_bug_1675570.py
index 62244a1f6c..e7c435fecc 100644
--- a/nova/tests/functional/regressions/test_bug_1675570.py
+++ b/nova/tests/functional/regressions/test_bug_1675570.py
@@ -12,19 +12,15 @@
 # License for the specific language governing permissions and limitations
 # under the License.
 
-import time
-
 from oslo_log import log as logging
 
-from nova import test
-from nova.tests import fixtures as nova_fixtures
 from nova.tests.functional.api import client
-from nova.tests.functional import fixtures as func_fixtures
+from nova.tests.functional import integrated_helpers
 
 LOG = logging.getLogger(__name__)
 
 
-class TestLocalDeleteAttachedVolumes(test.TestCase):
+class TestLocalDeleteAttachedVolumes(integrated_helpers._IntegratedTestBase):
     """Test local delete in the API of a server with a volume attached.
 
     This test creates a server, then shelve-offloads it, attaches a
@@ -38,89 +34,18 @@ class TestLocalDeleteAttachedVolumes(test.TestCase):
     and destroy the related BlockDeviceMappings.
     """
 
-    def setUp(self):
-        super(TestLocalDeleteAttachedVolumes, self).setUp()
-        self.useFixture(nova_fixtures.RealPolicyFixture())
-        # We need the CinderFixture to stub out the volume API.
-        self.cinder = self.useFixture(
-            nova_fixtures.CinderFixture(self))
-        # The NeutronFixture is needed to stub out validate_networks in API.
-        self.useFixture(nova_fixtures.NeutronFixture(self))
-        # Use the PlacementFixture to avoid annoying warnings in the logs.
-        self.useFixture(func_fixtures.PlacementFixture())
-        api_fixture = self.useFixture(nova_fixtures.OSAPIFixture(
-            api_version='v2.1'))
-        self.api = api_fixture.api
-        # We want to use 2.37 for passing networks='none' on server create.
-        # We also need this since you can only attach a volume to a
-        # shelved-offloaded server in microversion 2.20+.
-        self.api.microversion = 'latest'
-
-        # the image fake backend needed for image discovery
-        self.useFixture(nova_fixtures.GlanceFixture(self))
-
-        self.start_service('conductor')
-        self.start_service('scheduler')
-        self.start_service('compute')
-
-        self.useFixture(nova_fixtures.CastAsCallFixture(self))
+    microversion = 'latest'
 
+    def setUp(self):
+        super().setUp()
         self.flavor_id = self.api.get_flavors()[0]['id']
 
-    def _wait_for_instance_status(self, server_id, status):
-        timeout = 0.0
-        server = self.api.get_server(server_id)
-        while server['status'] != status and timeout < 10.0:
-            time.sleep(.1)
-            timeout += .1
-            server = self.api.get_server(server_id)
-        if server['status'] != status:
-            self.fail('Timed out waiting for server %s to have status: %s. '
-                      'Current status: %s' %
-                      (server_id, status, server['status']))
-        return server
-
-    def _wait_for_instance_delete(self, server_id):
-        timeout = 0.0
-        while timeout < 10.0:
-            try:
-                server = self.api.get_server(server_id)
-            except client.OpenStackApiNotFoundException:
-                # the instance is gone so we're happy
-                return
-            else:
-                time.sleep(.1)
-                timeout += .1
-
-        self.fail('Timed out waiting for server %s to be deleted. '
-                  'Current vm_state: %s. Current task_state: %s' %
-                  (server_id, server['OS-EXT-STS:vm_state'],
-                   server['OS-EXT-STS:task_state']))
-
     def _delete_server(self, server):
         try:
             self.api.delete_server(server['id'])
         except client.OpenStackApiNotFoundException:
             pass
 
-    def _wait_for_volume_attach(self, server_id, volume_id):
-        timeout = 0.0
-        server = self.api.get_server(server_id)
-        attached_vols = [vol['id'] for vol in
-                         server['os-extended-volumes:volumes_attached']]
-
-        while volume_id not in attached_vols and timeout < 10.0:
-            time.sleep(.1)
-            timeout += .1
-            server = self.api.get_server(server_id)
-            attached_vols = [vol['id'] for vol in
-                             server['os-extended-volumes:volumes_attached']]
-
-        if volume_id not in attached_vols:
-            self.fail('Timed out waiting for volume %s to be attached to '
-                      'server %s. Currently attached volumes: %s' %
-                      (volume_id, server_id, attached_vols))
-
     def test_local_delete_with_volume_attached(self, mock_version_get=None):
         LOG.info('Creating server and waiting for it to be ACTIVE.')
         server = dict(
@@ -133,12 +58,12 @@ class TestLocalDeleteAttachedVolumes(test.TestCase):
         server = self.api.post_server({'server': server})
         server_id = server['id']
         self.addCleanup(self._delete_server, server)
-        self._wait_for_instance_status(server_id, 'ACTIVE')
+        self._wait_for_state_change(server, 'ACTIVE')
 
         LOG.info('Shelve-offloading server %s', server_id)
         self.api.post_server_action(server_id, {'shelve': None})
         # Wait for the server to be offloaded.
-        self._wait_for_instance_status(server_id, 'SHELVED_OFFLOADED')
+        self._wait_for_state_change(server, 'SHELVED_OFFLOADED')
 
         volume_id = '9a695496-44aa-4404-b2cc-ccab2501f87e'
         LOG.info('Attaching volume %s to server %s', volume_id, server_id)
@@ -159,7 +84,7 @@ class TestLocalDeleteAttachedVolumes(test.TestCase):
         LOG.info('Deleting shelved-offloaded server %s.', server_id)
         self._delete_server(server)
         # Now wait for the server to be gone.
-        self._wait_for_instance_delete(server_id)
+        self._wait_until_deleted(server)
 
         LOG.info('Validating that volume %s was detached from server %s.',
                  volume_id, server_id)
diff --git a/nova/tests/functional/regressions/test_bug_1937375.py b/nova/tests/functional/regressions/test_bug_1937375.py
index 88dfea0ad7..860d80acd7 100644
--- a/nova/tests/functional/regressions/test_bug_1937375.py
+++ b/nova/tests/functional/regressions/test_bug_1937375.py
@@ -13,7 +13,6 @@
 # under the License.
 
 import mock
-import time
 
 from nova import context
 from nova import exception
@@ -34,26 +33,6 @@ class TestDuplicateVolAttachRace(integrated_helpers._IntegratedTestBase):
 
     microversion = 'latest'
 
-    # TODO(lyarwood): Copied from test_bug_1675570.py, move both into
-    # _IntegratedTestBase.
-    def _wait_for_volume_attach(self, server_id, volume_id):
-        timeout = 0.0
-        server = self.api.get_server(server_id)
-        attached_vols = [vol['id'] for vol in
-                         server['os-extended-volumes:volumes_attached']]
-
-        while volume_id not in attached_vols and timeout < 10.0:
-            time.sleep(.1)
-            timeout += .1
-            server = self.api.get_server(server_id)
-            attached_vols = [vol['id'] for vol in
-                             server['os-extended-volumes:volumes_attached']]
-
-        if volume_id not in attached_vols:
-            self.fail('Timed out waiting for volume %s to be attached to '
-                      'server %s. Currently attached volumes: %s' %
-                      (volume_id, server_id, attached_vols))
-
     def test_duplicate_volume_attach_race(self):
 
         ctxt = context.get_admin_context()