Add the billing/quote request
Change-Id: I9dcffe181108aa9f854fa696862b0b47b0f907f6
This commit is contained in:
0
cloudkittyclient/tests/v1/billing/__init__.py
Normal file
0
cloudkittyclient/tests/v1/billing/__init__.py
Normal file
60
cloudkittyclient/tests/v1/billing/test_quote.py
Normal file
60
cloudkittyclient/tests/v1/billing/test_quote.py
Normal file
@@ -0,0 +1,60 @@
|
|||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
# Copyright 2014 Objectif Libre
|
||||||
|
#
|
||||||
|
# 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.
|
||||||
|
#
|
||||||
|
# @author: François Magimel (linkid)
|
||||||
|
|
||||||
|
"""
|
||||||
|
Tests for the manager billing.quote `cloudkittyclient.v1.billing.quote`.
|
||||||
|
"""
|
||||||
|
|
||||||
|
from cloudkittyclient.openstack.common.apiclient import fake_client
|
||||||
|
from cloudkittyclient.tests import base
|
||||||
|
from cloudkittyclient.v1.billing import quote
|
||||||
|
from cloudkittyclient.v1 import client
|
||||||
|
|
||||||
|
|
||||||
|
compute = {
|
||||||
|
'desc': {
|
||||||
|
'image_id': "a41fba37-2429-4f15-aa00-b5bc4bf557bf",
|
||||||
|
},
|
||||||
|
'service': "compute",
|
||||||
|
'volume': 1
|
||||||
|
}
|
||||||
|
|
||||||
|
fixtures = {
|
||||||
|
'/v1/billing/quote': {
|
||||||
|
'POST': (
|
||||||
|
{},
|
||||||
|
'4.2'
|
||||||
|
),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
class QuoteManagerTest(base.TestCase):
|
||||||
|
def setUp(self):
|
||||||
|
super(QuoteManagerTest, self).setUp()
|
||||||
|
fake_http_client = fake_client.FakeHTTPClient(fixtures=fixtures)
|
||||||
|
api_client = client.Client(fake_http_client)
|
||||||
|
self.mgr = quote.QuoteManager(api_client)
|
||||||
|
|
||||||
|
def test_post(self):
|
||||||
|
_quote = self.mgr.post(json=compute)
|
||||||
|
self.assertIn('Quote', repr(_quote))
|
||||||
|
self.assertEqual(4.2, _quote.price)
|
||||||
|
|
||||||
|
def test_post_raw(self):
|
||||||
|
_quote = self.mgr.post(json=compute, return_raw=True)
|
||||||
|
self.assertEqual(4.2, _quote)
|
0
cloudkittyclient/v1/billing/__init__.py
Normal file
0
cloudkittyclient/v1/billing/__init__.py
Normal file
64
cloudkittyclient/v1/billing/quote.py
Normal file
64
cloudkittyclient/v1/billing/quote.py
Normal file
@@ -0,0 +1,64 @@
|
|||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
# Copyright 2014 Objectif Libre
|
||||||
|
#
|
||||||
|
# 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.
|
||||||
|
#
|
||||||
|
# @author: François Magimel (linkid)
|
||||||
|
|
||||||
|
"""
|
||||||
|
Quote resource and manager.
|
||||||
|
"""
|
||||||
|
|
||||||
|
from cloudkittyclient.openstack.common.apiclient import base
|
||||||
|
|
||||||
|
|
||||||
|
class Quote(base.Resource):
|
||||||
|
"""A resource represents a particular instance of an object (tenant, user,
|
||||||
|
etc). This is pretty much just a bag for attributes.
|
||||||
|
|
||||||
|
:param manager: Manager object
|
||||||
|
:param info: dictionary representing resource attributes
|
||||||
|
:param loaded: prevent lazy-loading if set to True
|
||||||
|
"""
|
||||||
|
|
||||||
|
def _add_details(self, info):
|
||||||
|
try:
|
||||||
|
setattr(self, 'price', info)
|
||||||
|
except AttributeError:
|
||||||
|
# In this case we already defined the attribute on the class
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
class QuoteManager(base.CrudManager):
|
||||||
|
"""Managers interact with a particular type of API and provide CRUD
|
||||||
|
operations for them.
|
||||||
|
"""
|
||||||
|
|
||||||
|
resource_class = Quote
|
||||||
|
collection_key = 'billing/quote'
|
||||||
|
key = 'quote'
|
||||||
|
|
||||||
|
def _post(self, url, json, response_key=None, return_raw=False):
|
||||||
|
"""Create an object."""
|
||||||
|
body = self.client.post(url, json=json).json()
|
||||||
|
if return_raw:
|
||||||
|
return body
|
||||||
|
return self.resource_class(self, body)
|
||||||
|
|
||||||
|
def post(self, **kwargs):
|
||||||
|
"""Get the price corresponding to resources attributes."""
|
||||||
|
kwargs = self._filter_kwargs(kwargs)
|
||||||
|
return self._post(
|
||||||
|
url=self.build_url(base_url='/v1', **kwargs),
|
||||||
|
json=kwargs.get('json'),
|
||||||
|
return_raw=kwargs.get('return_raw'))
|
@@ -20,6 +20,7 @@ OpenStack Client interface. Handles the REST calls and responses.
|
|||||||
"""
|
"""
|
||||||
|
|
||||||
from cloudkittyclient.openstack.common.apiclient import client
|
from cloudkittyclient.openstack.common.apiclient import client
|
||||||
|
from cloudkittyclient.v1.billing import quote
|
||||||
from cloudkittyclient.v1 import report
|
from cloudkittyclient.v1 import report
|
||||||
|
|
||||||
|
|
||||||
@@ -30,4 +31,10 @@ class Client(client.BaseClient):
|
|||||||
"""Initialize a new client for the Cloudkitty v1 API."""
|
"""Initialize a new client for the Cloudkitty v1 API."""
|
||||||
super(Client, self).__init__(http_client, extensions)
|
super(Client, self).__init__(http_client, extensions)
|
||||||
|
|
||||||
|
self.billing = Billing(self)
|
||||||
self.report = report.ReportManager(self)
|
self.report = report.ReportManager(self)
|
||||||
|
|
||||||
|
|
||||||
|
class Billing(object):
|
||||||
|
def __init__(self, http_client):
|
||||||
|
self.quote = quote.QuoteManager(http_client)
|
||||||
|
Reference in New Issue
Block a user