71 lines
2.0 KiB
Python
71 lines
2.0 KiB
Python
# Copyright 2015 Mirantis, Inc.
|
|
#
|
|
# 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 os
|
|
import time
|
|
|
|
from solar.core.resource import Resource
|
|
from solar.dblayer.model import Model, ModelMeta, get_bucket
|
|
|
|
import pytest
|
|
|
|
|
|
def patched_get_bucket_name(cls):
|
|
return cls.__name__ + str(time.time())
|
|
|
|
@pytest.fixture
|
|
def resources():
|
|
base_path = os.path.join(
|
|
os.path.dirname(os.path.realpath(__file__)),
|
|
'resource_fixtures')
|
|
|
|
node_path = os.path.join(base_path, 'node')
|
|
node1 = Resource('node1', node_path, args={'ip':'10.0.0.1'})
|
|
node2 = Resource('node2', node_path, args={'ip':'10.0.0.2'})
|
|
|
|
base_service_path = os.path.join(base_path, 'base_service')
|
|
service1 = Resource('service1', base_service_path)
|
|
return {'node1' : node1,
|
|
'node2' : node2,
|
|
'service1': service1
|
|
}
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def setup(request):
|
|
|
|
for model in ModelMeta._defined_models:
|
|
model.bucket = get_bucket(None, model, ModelMeta)
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def setup(request):
|
|
|
|
for model in ModelMeta._defined_models:
|
|
model.bucket = get_bucket(None, model, ModelMeta)
|
|
|
|
def pytest_runtest_teardown(item, nextitem):
|
|
ModelMeta.session_end(result=True)
|
|
return nextitem
|
|
|
|
# It will run before all fixtures
|
|
def pytest_runtest_setup(item):
|
|
ModelMeta.session_start()
|
|
|
|
# it will run after fixtures but before test
|
|
def pytest_runtest_call(item):
|
|
ModelMeta.session_end()
|
|
ModelMeta.session_start()
|
|
|
|
|
|
Model.get_bucket_name = classmethod(patched_get_bucket_name)
|