302 lines
11 KiB
Python
302 lines
11 KiB
Python
# -*- coding: utf-8 -*-
|
|
# 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
|
|
from StringIO import StringIO
|
|
|
|
import pytest
|
|
import yaml
|
|
|
|
from solar.core.resource import composer as cr
|
|
from solar.events.controls import Dep
|
|
from solar.events.controls import React
|
|
|
|
from types import NoneType
|
|
|
|
|
|
@pytest.fixture
|
|
def good_events():
|
|
events = '''
|
|
- type: depends_on
|
|
parent_action: 'service1.run'
|
|
state: 'success'
|
|
child_action: 'config1.run'
|
|
- type: react_on
|
|
parent_action: 'config1.run'
|
|
state: 'success'
|
|
child_action: 'service1.apply_config'
|
|
'''
|
|
return yaml.load(StringIO(events))
|
|
|
|
|
|
@pytest.fixture
|
|
def bad_event_type():
|
|
events = '''
|
|
- type: skip
|
|
parent_action: 'service1.run'
|
|
state: 'success'
|
|
child_action: 'config1.run'
|
|
'''
|
|
return yaml.load(StringIO(events))
|
|
|
|
|
|
def test_create_path_does_not_exists():
|
|
with pytest.raises(Exception) as excinfo:
|
|
cr.create('node1', '/path/does/not/exists')
|
|
assert excinfo.filename == '/path/does/not/exists'
|
|
|
|
|
|
def test_create_resource():
|
|
node_path = os.path.join(
|
|
os.path.dirname(os.path.realpath(__file__)),
|
|
'resource_fixtures', 'node')
|
|
resources = cr.create('node1', node_path)
|
|
assert len(resources) == 1
|
|
assert resources[0].name == 'node1'
|
|
|
|
|
|
def test_create_from_composer_file(tmpdir):
|
|
base_path = os.path.join(
|
|
os.path.dirname(os.path.realpath(__file__)),
|
|
'resource_fixtures')
|
|
vr_tmpl_path = os.path.join(base_path, 'nodes.yaml.tmpl')
|
|
node_resource_path = os.path.join(base_path, 'node')
|
|
with open(vr_tmpl_path) as f:
|
|
vr_data = f.read().format(resource_path=node_resource_path)
|
|
vr_file = tmpdir.join('nodes.yaml')
|
|
vr_file.write(vr_data)
|
|
resources = cr.create('nodes', str(vr_file))
|
|
assert len(resources) == 2
|
|
|
|
|
|
def test_create_from_composer_file_with_list(tmpdir):
|
|
base_path = os.path.join(
|
|
os.path.dirname(os.path.realpath(__file__)),
|
|
'resource_fixtures')
|
|
vr_tmpl_path = os.path.join(base_path, 'resource_with_list.yaml.tmpl')
|
|
base_resource_path = os.path.join(base_path, 'base_service')
|
|
with open(vr_tmpl_path) as f:
|
|
vr_data = f.read().format(resource_path=base_resource_path)
|
|
vr_file = tmpdir.join('base.yaml')
|
|
vr_file.write(vr_data)
|
|
resources = cr.create('base', str(vr_file))
|
|
assert len(resources) == 1
|
|
res = resources[0]
|
|
assert res.args['servers'] == [1, 2]
|
|
|
|
|
|
def test_create_from_composer_file_with_dict(tmpdir):
|
|
base_path = os.path.join(
|
|
os.path.dirname(os.path.realpath(__file__)),
|
|
'resource_fixtures')
|
|
vr_tmpl_path = os.path.join(base_path, 'resource_with_dict.yaml.tmpl')
|
|
base_resource_path = os.path.join(base_path, 'base_service')
|
|
with open(vr_tmpl_path) as f:
|
|
vr_data = f.read().format(resource_path=base_resource_path)
|
|
vr_file = tmpdir.join('base.yaml')
|
|
vr_file.write(vr_data)
|
|
resources = cr.create('base', str(vr_file))
|
|
assert len(resources) == 1
|
|
res = resources[0]
|
|
assert res.args['servers'] == {'a': 1, 'b': 2}
|
|
|
|
|
|
def test_update(tmpdir):
|
|
# XXX: make helper for it
|
|
base_path = os.path.join(
|
|
os.path.dirname(os.path.realpath(__file__)),
|
|
'resource_fixtures')
|
|
vr_node_tmpl_path = os.path.join(base_path, 'nodes.yaml.tmpl')
|
|
vr_update_tmpl_path = os.path.join(base_path, 'update.yaml.tmpl')
|
|
update_path = os.path.join(base_path, 'update')
|
|
node_resource_path = os.path.join(base_path, 'node')
|
|
with open(vr_node_tmpl_path) as f:
|
|
vr_data = f.read().format(resource_path=node_resource_path)
|
|
with open(vr_update_tmpl_path) as f:
|
|
update_data = f.read().format(resource_path=update_path)
|
|
vr_file = tmpdir.join('nodes.yaml')
|
|
vr_file.write(vr_data)
|
|
update_file = tmpdir.join('update.yaml')
|
|
update_file.write(update_data)
|
|
resources = cr.create('nodes', str(vr_file))
|
|
cr.create('updates', str(update_file))
|
|
assert resources[0].args['ip'] == '10.0.0.4'
|
|
|
|
|
|
def test_parse_events(good_events):
|
|
events = [Dep(parent='service1', parent_action='run',
|
|
child='config1', child_action='run',
|
|
state='success'),
|
|
React(parent='config1', parent_action='run',
|
|
child='service1', child_action='apply_config',
|
|
state='success')]
|
|
parsed = cr.parse_events(good_events)
|
|
assert events == parsed
|
|
|
|
|
|
def test_parse_bad_event(bad_event_type):
|
|
with pytest.raises(Exception) as execinfo:
|
|
cr.parse_events(bad_event_type)
|
|
error = 'Invalid event type: skip'
|
|
assert error == str(execinfo.value)
|
|
|
|
|
|
def test_add_connections(mocker, resources):
|
|
mocked_signals = mocker.patch(
|
|
'solar.core.resource.resource.Resource.connect_with_events')
|
|
args = {'ip': 'node1::ip',
|
|
'servers': ['node1::ip', 'node2::ip'],
|
|
'alias': 'ser1'
|
|
}
|
|
cr.update_inputs('service1', args)
|
|
assert mocked_signals.call_count == 2
|
|
|
|
|
|
def test_add_list_values(mocker, resources):
|
|
mocked_signals = mocker.patch(
|
|
'solar.core.resource.resource.Resource.connect_with_events')
|
|
args = {'ip': 'node1::ip',
|
|
'servers': ['server1', 'server2'],
|
|
'alias': 'ser1'
|
|
}
|
|
cr.update_inputs('service1', args)
|
|
assert mocked_signals.call_count == 1
|
|
|
|
|
|
def test_double_parent_input(mocker, resources):
|
|
mocked_signals = mocker.patch(
|
|
'solar.core.resource.resource.Resource.connect_with_events')
|
|
args = {'ip': 'node1::ip',
|
|
'servers': ['server1', 'server2'],
|
|
'alias': 'ser1',
|
|
'private_ip': 'node1::ip'
|
|
}
|
|
cr.update_inputs('service1', args)
|
|
call_args = mocked_signals.call_args[0][1]
|
|
assert 'ip' in call_args['ip']
|
|
assert 'private_ip' in call_args['ip']
|
|
|
|
|
|
def test_parse_connection():
|
|
correct_connection = {'child_input': 'ip',
|
|
'parent': 'node1',
|
|
'parent_input': 'ip',
|
|
'events': None
|
|
}
|
|
connection = cr.parse_connection('ip', 'node1::ip')
|
|
assert correct_connection == connection
|
|
|
|
|
|
def test_parse_dict_input():
|
|
correct_connections = [{'child_input': 'env:host',
|
|
'events': None,
|
|
'parent': 'node1',
|
|
'parent_input': 'ip'}]
|
|
connections, _ = cr.parse_dict_input('env', {'host': 'node1::ip'})
|
|
assert correct_connections == connections
|
|
|
|
|
|
def test_parse_list_of_dicts1():
|
|
_, assigments, _ = cr.parse_inputs({'map': [{'a': 1, 'b': 3},
|
|
{'a': 2, 'b': 4}]})
|
|
assert assigments == {'map': [{'a': 1, 'b': 3}, {'a': 2, 'b': 4}]}
|
|
|
|
|
|
def test_parse_simple_list():
|
|
_, assigments, _ = cr.parse_inputs({'map': [1, 2, 3]})
|
|
assert assigments == {'map': [1, 2, 3]}
|
|
|
|
|
|
def test_parse_computable_input():
|
|
comp, conn = cr.parse_computable_input('cmd', {'computable': {
|
|
'func': 'echo 1',
|
|
'type': 'full',
|
|
'lang': 'jinja',
|
|
'connections': ['N::ip']}})
|
|
assert comp == {'lang': 'jinja',
|
|
'type': 'full',
|
|
'name': 'cmd',
|
|
'func': 'echo 1'}
|
|
assert conn == [{'child_input': 'cmd',
|
|
'parent_input': 'ip',
|
|
'parent': 'N',
|
|
'events': None}]
|
|
|
|
|
|
def test_parse_connection_disable_events():
|
|
correct_connection = {'child_input': 'ip',
|
|
'parent': 'node1',
|
|
'parent_input': 'ip',
|
|
'events': False
|
|
}
|
|
connection = cr.parse_connection('ip', 'node1::ip::NO_EVENTS')
|
|
assert correct_connection == connection
|
|
|
|
|
|
def test_parse_list_of_connected_dicts():
|
|
inputs = {'list': [
|
|
{'key': 'emitter1::key'},
|
|
{'key': 'emitter2::key'}]}
|
|
connections, assignments, _ = cr.parse_inputs(inputs)
|
|
assert assignments == {}
|
|
assert connections == [
|
|
{'child_input': 'list:key', 'parent_input': 'key',
|
|
'parent': 'emitter1', 'events': None},
|
|
{'child_input': 'list:key', 'parent_input': 'key',
|
|
'parent': 'emitter2', 'events': None}]
|
|
|
|
|
|
def test_setting_location(tmpdir):
|
|
# XXX: make helper for it
|
|
base_path = os.path.join(
|
|
os.path.dirname(os.path.realpath(__file__)),
|
|
'resource_fixtures')
|
|
vr_node_tmpl_path = os.path.join(base_path, 'nodes.yaml.tmpl')
|
|
vr_location_tmpl_path = os.path.join(base_path, 'with_location.yaml.tmpl')
|
|
base_service_path = os.path.join(base_path, 'base_service')
|
|
node_resource_path = os.path.join(base_path, 'node')
|
|
with open(vr_node_tmpl_path) as f:
|
|
vr_data = f.read().format(resource_path=node_resource_path)
|
|
with open(vr_location_tmpl_path) as f:
|
|
location_data = f.read().format(resource_path=base_service_path)
|
|
vr_file = tmpdir.join('nodes.yaml')
|
|
vr_file.write(vr_data)
|
|
location_file = tmpdir.join('with_location.yaml')
|
|
location_file.write(location_data)
|
|
cr.create('nodes', str(vr_file))
|
|
resources = cr.create('updates', str(location_file))
|
|
assert 'location=node1' in resources[0].tags
|
|
|
|
|
|
def test_correct_types(tmpdir):
|
|
base_path = os.path.join(
|
|
os.path.dirname(os.path.realpath(__file__)),
|
|
'resource_fixtures')
|
|
vr_tmpl_path = os.path.join(base_path, 'types.yaml.tmpl')
|
|
sub_resource_path = os.path.join(base_path, 'type.yaml.tmpl')
|
|
target_resource_path = os.path.join(base_path, 'types_test')
|
|
with open(sub_resource_path) as f:
|
|
vr_sub_data = f.read()
|
|
vr_sub_data = vr_sub_data.replace('{type_path}', target_resource_path)
|
|
vr_sub_file = tmpdir.join('type.yaml')
|
|
vr_sub_file.write(vr_sub_data)
|
|
with open(vr_tmpl_path) as f:
|
|
vr_data = f.read().format(sub_path=str(vr_sub_file))
|
|
vr_file = tmpdir.join('types.yaml')
|
|
vr_file.write(vr_data)
|
|
resource = cr.create('types', str(vr_file))[0]
|
|
exps = (('as_int', int), ('as_string', str), ('as_null', NoneType))
|
|
for name, exp in exps:
|
|
assert isinstance(resource.args[name], exp)
|