text: # Copyright 2013 OpenStack Foundation
  # All Rights Reserved.
  #
  #    Licensed under the Apache License, Version 2.0 (the "License"); you may
  #    not use this file except in compliance with the License. You may obtain
  #    a copy of the License at
  #
  #         http://www.apache.org/licenses/LICENSE-2.0
  #
  #    Unless required by applicable law or agreed to in writing, software
  #    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
  #    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
  #    License for the specific language governing permissions and limitations
  #    under the License.
  
  import six
  from tempest.lib.common.utils import data_utils
  from tempest.lib import exceptions as lib_exc
  from tempest import test
  
  from neutron.tests.tempest.api import base
  
  
  class QuotasTestBase(base.BaseAdminNetworkTest):
  
      @classmethod
      @test.requires_ext(extension="quotas", service="network")
      def resource_setup(cls):
          super(QuotasTestBase, cls).resource_setup()
  
      def _create_tenant(self):
          # Add a tenant to conduct the test
          test_tenant = data_utils.rand_name('test_tenant_')
          test_description = data_utils.rand_name('desc_')
          tenant = self.identity_admin_client.create_tenant(
              name=test_tenant,
              description=test_description)['tenant']
          self.addCleanup(
              self.identity_admin_client.create_tenant, tenant['id'])
          return tenant
  
      def _setup_quotas(self, project_id, **new_quotas):
          # Change quotas for tenant
          quota_set = self.admin_client.update_quotas(project_id,
                                                      **new_quotas)
          self.addCleanup(self._cleanup_quotas, project_id)
          return quota_set
  
      def _cleanup_quotas(self, project_id):
          # Try to clean up the resources. If it fails, then
          # assume that everything was already deleted, so
          # it is OK to continue.
          try:
              self.admin_client.reset_quotas(project_id)
          except lib_exc.NotFound:
              pass
  
  
  class QuotasTest(QuotasTestBase):
      """Test the Neutron API of Quotas.
  
      Tests the following operations in the Neutron API using the REST client for
      Neutron:
  
          list quotas for tenants who have non-default quota values
          show quotas for a specified tenant
          update quotas for a specified tenant
          reset quotas to default values for a specified tenant
  
      v2.0 of the API is assumed.
      It is also assumed that the per-tenant quota extension API is configured
      in /etc/neutron/neutron.conf as follows:
  
          quota_driver = neutron.db.driver.DbQuotaDriver
      """
  
      @test.attr(type='gate')
      @test.idempotent_id('2390f766-836d-40ef-9aeb-e810d78207fb')
      def test_quotas(self):
          tenant_id = self._create_tenant()['id']
          new_quotas = {'network': 0, 'security_group': 0}
  
          # Change quotas for tenant
          quota_set = self._setup_quotas(tenant_id, **new_quotas)
          for key, value in six.iteritems(new_quotas):
              self.assertEqual(value, quota_set[key])
  
          # Confirm our tenant is listed among tenants with non default quotas
          non_default_quotas = self.admin_client.list_quotas()
          found = False
          for qs in non_default_quotas['quotas']:
              if qs['tenant_id'] == tenant_id:
                  found = True
          self.assertTrue(found)
  
          # Confirm from API quotas were changed as requested for tenant
          quota_set = self.admin_client.show_quotas(tenant_id)
          quota_set = quota_set['quota']
          for key, value in six.iteritems(new_quotas):
              self.assertEqual(value, quota_set[key])
  
          # Reset quotas to default and confirm
          self.admin_client.reset_quotas(tenant_id)
          non_default_quotas = self.admin_client.list_quotas()
          for q in non_default_quotas['quotas']:
              self.assertNotEqual(tenant_id, q['tenant_id'])
  