mirror of
https://github.com/apache/cloudstack.git
synced 2025-10-26 08:42:29 +01:00
182899d0b5
* refactor: remove trailing whitespace from Python files * Add the GitHub Super-Linter Add Python flake8 linting for W291 trailing whitespace * Add licenses
265 lines
9.2 KiB
Python
265 lines
9.2 KiB
Python
# Licensed to the Apache Software Foundation (ASF) under one
|
|
# or more contributor license agreements. See the NOTICE file
|
|
# distributed with this work for additional information
|
|
# regarding copyright ownership. The ASF licenses this file
|
|
# to you 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.
|
|
""" Test cases for checking quota API
|
|
"""
|
|
|
|
#Import Local Modules
|
|
import marvin
|
|
from marvin.cloudstackTestCase import *
|
|
from marvin.cloudstackAPI import *
|
|
from marvin.lib.utils import *
|
|
from marvin.lib.base import *
|
|
from marvin.lib.common import *
|
|
from marvin.lib.utils import (random_gen)
|
|
from nose.plugins.attrib import attr
|
|
|
|
#Import System modules
|
|
import time
|
|
|
|
class TestQuota(cloudstackTestCase):
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
# Create Account
|
|
testClient = super(TestQuota, cls).getClsTestClient()
|
|
cls.apiclient = testClient.getApiClient()
|
|
cls.services = testClient.getParsedTestDataConfig()
|
|
|
|
# Get Zone, Domain
|
|
cls.domain = get_domain(cls.apiclient)
|
|
cls.zone = get_zone(cls.apiclient, cls.testClient.getZoneForTests())
|
|
|
|
# Create Account
|
|
cls.account = Account.create(
|
|
cls.apiclient,
|
|
cls.services["account"],
|
|
domainid=cls.domain.id
|
|
)
|
|
cls._cleanup = [
|
|
cls.account,
|
|
]
|
|
return
|
|
|
|
@classmethod
|
|
def tearDownClass(cls):
|
|
super(TestQuota, cls).tearDownClass()
|
|
|
|
def setUp(self):
|
|
self.apiclient = self.testClient.getApiClient()
|
|
self.hypervisor = self.testClient.getHypervisorInfo()
|
|
self.dbclient = self.testClient.getDbConnection()
|
|
self.services = self.testClient.getParsedTestDataConfig()
|
|
self.zone = get_zone(self.apiclient, self.testClient.getZoneForTests())
|
|
self.pod = get_pod(self.apiclient, self.zone.id)
|
|
self.cleanup = []
|
|
return
|
|
|
|
def tearDown(self):
|
|
super(TestQuota, self).tearDown()
|
|
|
|
#Check quotaTariffList API returning 22 items
|
|
@attr(tags=["smoke", "advanced"], required_hardware="false")
|
|
def test_01_quota(self):
|
|
if not is_config_suitable(
|
|
apiclient=self.apiclient,
|
|
name='quota.enable.service',
|
|
value='true'):
|
|
self.skipTest('quota.enable.service should be true. skipping')
|
|
|
|
cmd = quotaTariffList.quotaTariffListCmd()
|
|
response = self.apiclient.quotaTariffList(cmd)
|
|
|
|
self.debug("Number of quota usage types: %s" % len(response))
|
|
self.assertEqual(
|
|
len(response), 22
|
|
)
|
|
for quota in response:
|
|
self.debug("Usage Name: %s" % quota.usageName)
|
|
self.assertEqual(
|
|
hasattr(quota, 'usageName'),
|
|
True,
|
|
"Check whether usgaeName field is there"
|
|
)
|
|
|
|
return
|
|
|
|
#Check quota tariff on a particualr day
|
|
@attr(tags=["smoke", "advanced"], required_hardware="false")
|
|
def test_02_quota(self):
|
|
if not is_config_suitable(
|
|
apiclient=self.apiclient,
|
|
name='quota.enable.service',
|
|
value='true'):
|
|
self.skipTest('quota.enable.service should be true. skipping')
|
|
|
|
cmd = quotaTariffList.quotaTariffListCmd()
|
|
cmd.startdate='2015-07-06'
|
|
response = self.apiclient.quotaTariffList(cmd)
|
|
|
|
self.debug("Number of quota usage types: %s" % len(response))
|
|
self.assertEqual(
|
|
len(response), 22
|
|
)
|
|
|
|
return
|
|
|
|
#check quota tariff of a particular item
|
|
@attr(tags=["smoke", "advanced"], required_hardware="false")
|
|
def test_03_quota(self):
|
|
if not is_config_suitable(
|
|
apiclient=self.apiclient,
|
|
name='quota.enable.service',
|
|
value='true'):
|
|
self.skipTest('quota.enable.service should be true. skipping')
|
|
|
|
cmd = quotaTariffList.quotaTariffListCmd()
|
|
cmd.startdate='2015-07-06'
|
|
cmd.usagetype='10'
|
|
response = self.apiclient.quotaTariffList(cmd)
|
|
|
|
self.debug("Number of quota usage types: %s" % len(response))
|
|
self.assertEqual(
|
|
len(response), 1
|
|
)
|
|
return
|
|
|
|
|
|
#check quota tariff
|
|
#Change it
|
|
#Check on affective date the new tariff should be applicable
|
|
#check the old tariff it should be same
|
|
@attr(tags=["smoke", "advanced"], required_hardware="false")
|
|
def test_04_quota(self):
|
|
if not is_config_suitable(
|
|
apiclient=self.apiclient,
|
|
name='quota.enable.service',
|
|
value='true'):
|
|
self.skipTest('quota.enable.service should be true. skipping')
|
|
|
|
cmd = quotaTariffList.quotaTariffListCmd()
|
|
cmd.startdate='2015-07-06'
|
|
cmd.usagetype='10'
|
|
response = self.apiclient.quotaTariffList(cmd)
|
|
|
|
self.debug("Number of quota usage types: %s" % len(response))
|
|
self.assertEqual(
|
|
len(response), 1
|
|
)
|
|
quota = response[0]
|
|
self.debug("Tariff Value for 10: %s" % quota.tariffValue)
|
|
|
|
cmd = quotaTariffUpdate.quotaTariffUpdateCmd()
|
|
tomorrow = datetime.date.today() + datetime.timedelta(days=1)
|
|
cmd.startdate=tomorrow
|
|
cmd.usagetype='10'
|
|
cmd.value='2.9'
|
|
response = self.apiclient.quotaTariffUpdate(cmd)
|
|
|
|
cmd = quotaTariffList.quotaTariffListCmd()
|
|
cmd.startdate=tomorrow
|
|
cmd.usagetype='10'
|
|
response = self.apiclient.quotaTariffList(cmd)
|
|
self.assertEqual(
|
|
len(response), 1
|
|
)
|
|
quota = response[0]
|
|
self.debug("Tariff Value for 10: %s" % quota.tariffValue)
|
|
|
|
self.assertEqual( quota.tariffValue, 2.9)
|
|
|
|
|
|
cmd = quotaTariffList.quotaTariffListCmd()
|
|
cmd.startdate='2015-07-07'
|
|
cmd.usagetype='10'
|
|
response = self.apiclient.quotaTariffList(cmd)
|
|
self.assertEqual(
|
|
len(response), 1
|
|
)
|
|
quota = response[0]
|
|
self.debug("Tariff Value for 10: %s" % quota.tariffValue)
|
|
|
|
self.assertEqual( quota.tariffValue, 0)
|
|
|
|
return
|
|
|
|
|
|
#Make credit deposit
|
|
@attr(tags=["smoke", "advanced"], required_hardware="false")
|
|
def test_05_quota(self):
|
|
if not is_config_suitable(
|
|
apiclient=self.apiclient,
|
|
name='quota.enable.service',
|
|
value='true'):
|
|
self.skipTest('quota.enable.service should be true. skipping')
|
|
|
|
cmd = quotaCredits.quotaCreditsCmd()
|
|
cmd.domainid = self.account.domainid
|
|
cmd.account = self.account.name
|
|
cmd.value = '10'
|
|
cmd.quota_enforce = '1'
|
|
cmd.min_balance = '9'
|
|
response = self.apiclient.quotaCredits(cmd)
|
|
|
|
self.debug("Credit response update on: %s" % response.updated_on)
|
|
|
|
return
|
|
|
|
|
|
#Make credit deposit and check today balance
|
|
@attr(tags=["smoke", "advanced"], required_hardware="false")
|
|
def test_06_quota(self):
|
|
if not is_config_suitable(
|
|
apiclient=self.apiclient,
|
|
name='quota.enable.service',
|
|
value='true'):
|
|
self.skipTest('quota.enable.service should be true. skipping')
|
|
|
|
cmd = quotaBalance.quotaBalanceCmd()
|
|
today = datetime.date.today()
|
|
cmd.domainid = self.account.domainid
|
|
cmd.account = self.account.name
|
|
cmd.startdate = today
|
|
response = self.apiclient.quotaBalance(cmd)
|
|
|
|
self.debug(f"Quota Balance: {response.balance}")
|
|
|
|
self.assertEqual( response.startquota, 0, f"startQuota is supposed to be 0 but was {response.balance.startquota}")
|
|
return
|
|
|
|
#make credit deposit and check start and end date balances
|
|
@attr(tags=["smoke", "advanced"], required_hardware="false")
|
|
def test_07_quota(self):
|
|
if not is_config_suitable(
|
|
apiclient=self.apiclient,
|
|
name='quota.enable.service',
|
|
value='true'):
|
|
self.skipTest('quota.enable.service should be true. skipping')
|
|
|
|
cmd = quotaBalance.quotaBalanceCmd()
|
|
today = datetime.date.today()
|
|
cmd.domainid = self.account.domainid
|
|
cmd.account = self.account.name
|
|
cmd.startdate = today - datetime.timedelta(days=2)
|
|
cmd.enddate = today
|
|
response = self.apiclient.quotaBalance(cmd)
|
|
|
|
self.debug("Quota Balance: {response.balance}")
|
|
|
|
self.assertEqual( response.balance.startquota, 0, f"startquota was supposed to be 0 but was {response.balance.startquota}")
|
|
self.assertEqual( response.balance.endquota, 10, f"endquota was supposed to be 10 but was {response.balance.endquota}")
|
|
return
|