azure/aria/aria-extension-cloudify/src/aria/tests/fixtures.py

   1 # Licensed to the Apache Software Foundation (ASF) under one or more
   2 # contributor license agreements.  See the NOTICE file distributed with
   3 # this work for additional information regarding copyright ownership.
   4 # The ASF licenses this file to You under the Apache License, Version 2.0
   5 # (the "License"); you may not use this file except in compliance with
   6 # the License.  You may obtain a copy of the License at
   7 #
   8 #     http://www.apache.org/licenses/LICENSE-2.0
   9 #
  10 # Unless required by applicable law or agreed to in writing, software
  11 # distributed under the License is distributed on an "AS IS" BASIS,
  12 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13 # See the License for the specific language governing permissions and
  14 # limitations under the License.
  15
  16 import shutil
  17
  18 import pytest
  19
  20 from aria import (
  21     application_model_storage,
  22     application_resource_storage
  23 )
  24 from aria.orchestrator import plugin
  25 from aria.storage import (
  26     sql_mapi,
  27     filesystem_rapi
  28 )
  29
  30 from . import storage
  31
  32
  33 @pytest.fixture
  34 def inmemory_model():
  35     model = application_model_storage(sql_mapi.SQLAlchemyModelAPI,
  36                                       initiator=storage.init_inmemory_model_storage)
  37     yield model
  38     storage.release_sqlite_storage(model)
  39
  40
  41 @pytest.fixture
  42 def fs_model(tmpdir):
  43     result = application_model_storage(sql_mapi.SQLAlchemyModelAPI,
  44                                        initiator_kwargs=dict(base_dir=str(tmpdir)),
  45                                        initiator=sql_mapi.init_storage)
  46     yield result
  47     storage.release_sqlite_storage(result)
  48
  49
  50 @pytest.fixture
  51 def resource_storage(tmpdir):
  52     result = tmpdir.join('resources')
  53     result.mkdir()
  54     resource_storage = application_resource_storage(
  55         filesystem_rapi.FileSystemResourceAPI,
  56         api_kwargs=dict(directory=str(result)))
  57     yield resource_storage
  58     shutil.rmtree(str(result))
  59
  60
  61 @pytest.fixture
  62 def plugins_dir(tmpdir):
  63     result = tmpdir.join('plugins')
  64     result.mkdir()
  65     return str(result)
  66
  67
  68 @pytest.fixture
  69 def plugin_manager(model, plugins_dir):
  70     return plugin.PluginManager(model=model, plugins_dir=plugins_dir)