summaryrefslogtreecommitdiffstats
path: root/azure/aria/aria-extension-cloudify/src/aria/tests/utils/test_plugin.py
blob: c91d0c9385e51a199bab05b066420779a0cb1d2d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
# 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.

import os

import pytest

from aria.orchestrator import exceptions
from aria.utils.plugin import create as create_plugin

from ..fixtures import (  # pylint: disable=unused-import
    plugins_dir,
    plugin_manager,
    inmemory_model as model
)


PACKAGE_NAME = 'mock-plugin'
PACKAGE_VERSION = '100'


class TestPluginManager(object):

    def test_install(self, plugin_manager, mock_plugin, model, plugins_dir):
        plugin = plugin_manager.install(mock_plugin)
        assert plugin.package_name == PACKAGE_NAME
        assert plugin.package_version == PACKAGE_VERSION
        assert plugin == model.plugin.get(plugin.id)
        plugin_dir = os.path.join(plugins_dir, '{0}-{1}'.format(PACKAGE_NAME, PACKAGE_VERSION))
        assert os.path.isdir(plugin_dir)
        assert plugin_dir == plugin_manager.get_plugin_dir(plugin)

    def test_install_already_exits(self, plugin_manager, mock_plugin):
        plugin_manager.install(mock_plugin)
        with pytest.raises(exceptions.PluginAlreadyExistsError):
            plugin_manager.install(mock_plugin)


@pytest.fixture
def mock_plugin(tmpdir):
    source_dir = tmpdir.join('mock_plugin')
    source_dir.mkdir()
    setup_py = source_dir.join('setup.py')
    setup_py.write('from setuptools import setup; setup(name="{0}", version="{1}")'
                   .format(PACKAGE_NAME, PACKAGE_VERSION))
    return create_plugin(source=str(source_dir), destination_dir=str(tmpdir))