def setUp(self): super(DbTestCase, self).setUp() self.dbapi = dbapi.get_instance() global _DB_CACHE if not _DB_CACHE: _DB_CACHE = Database(sqla_api, migration, sql_connection=CONF.database.connection) self.useFixture(_DB_CACHE)
class Handler(object): dbapi = dbapi.get_instance() def __init__(self): super(Handler, self).__init__() def create_account(cls, context, **kwargs): LOG.debug('create account: Received message from RPC.') account = db_models.Account(**kwargs) return cls.dbapi.create_account(context, account) def get_account(cls, context, **kwargs): LOG.debug('get account: Received message from RPC.') account = cls.dbapi.get_account(context, **kwargs) return account def get_accounts(cls, context, **kwargs): LOG.debug('get accounts: Received message from RPC.') accounts = cls.dbapi.get_accounts(context, **kwargs) return accounts def get_accounts_count(cls, context, **kwargs): LOG.debug('get accounts count: Received message from RPC.') accounts = cls.dbapi.get_accounts_count(context, **kwargs) return accounts def delete_account(cls, context, **kwargs): LOG.debug('delete account: Received message from RPC.') cls.dbapi.delete_account(context, **kwargs) def change_account_level(cls, context, **kwargs): LOG.debug('change account level: Received message from RPC.') return cls.dbapi.change_account_level(context, **kwargs) def charge_account(cls, context, **kwargs): LOG.debug('update account: Received message from RPC.') return cls.dbapi.charge_account(context, kwargs.pop('user_id'), **kwargs) def update_account(cls, context, **kwargs): LOG.debug('update account: Received message from RPC.') return cls.dbapi.update_account(context, kwargs.pop('user_id'), **kwargs) def get_charges(cls, context, **kwargs): LOG.debug('get charges: Received message from RPC.') return cls.dbapi.get_charges(context, **kwargs) def get_charges_price_and_count(cls, context, **kwargs): LOG.debug('get_charges_price_and_count: Received message from RPC.') return cls.dbapi.get_charges_price_and_count(context, **kwargs)
class Handler(object): dbapi = dbapi.get_instance() def get_billing_owner(cls, context, **kwargs): LOG.debug('Conductor Function: get_billing_owner.') billing_owner = cls.dbapi.get_billing_owner(context, kwargs['project_id']) return billing_owner def change_billing_owner(cls, context, **kwargs): LOG.debug('Conductor Function: change_billing_owner.') cls.dbapi.change_billing_owner(context, kwargs['project_id'], kwargs['user_id']) def create_project(cls, context, **kwargs): LOG.debug('Conductor Function: create_project.') project = db_models.Project(**kwargs) return cls.dbapi.create_project(context, project) def get_project(cls, context, **kwargs): LOG.debug('Conductor Function: get_project.') project = cls.dbapi.get_project(context, kwargs['project_id']) return project def delete_project(cls, context, **kwargs): LOG.debug('Conductor Function: delete_project.') cls.dbapi.delete_project(context, kwargs['project_id']) def get_relation(cls, context, **kwargs): LOG.debug('Conductor Function: get_user_projects.') return cls.dbapi.get_relation(context, **kwargs) def get_projects(cls, context, **kwargs): LOG.debug('Conductor Function: get_projects.') return cls.dbapi.get_projects(context, **kwargs)
# 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. from shadowfiend.tests.unit.db import utils as db_utils from shadowfiend.db import models as db_models from shadowfiend.db import api as dbapi dbapi = dbapi.get_instance() def _create_temp_account(**kwargs): db_account = db_utils.get_test_account(**kwargs) account = db_models.Account(**db_account) return account def _create_temp_project(**kwargs): db_project = db_utils.get_test_project(**kwargs) project = db_models.Project(**db_project) return project def _create_temp_relation(**kwargs):
def create_test_account(context, **kwargs): account = get_test_account(**kwargs) dbapi = db_api.get_instance() return dbapi.create_account(context, db_models.Account(**account))