297 lines
9.9 KiB
Python
297 lines
9.9 KiB
Python
#
|
|
# 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 copy
|
|
import json
|
|
import os
|
|
|
|
import fixtures
|
|
|
|
from tests import common
|
|
|
|
|
|
class HookPuppetTest(common.RunScriptTest):
|
|
|
|
data = {
|
|
'id': '1234',
|
|
'creation_time': '2015-07-16T11:40:20',
|
|
'name': 'fake_resource_name',
|
|
'group': 'puppet',
|
|
'options': {
|
|
'enable_hiera': True,
|
|
'enable_facter': True,
|
|
'enable_debug': True,
|
|
'enable_verbose': True,
|
|
},
|
|
'inputs': [
|
|
{'name': 'foo', 'value': 'bar'},
|
|
{'name': 'another', 'value': 'input'}
|
|
],
|
|
'outputs': [
|
|
{'name': 'first_output'},
|
|
{'name': 'second_output'}
|
|
],
|
|
'config': 'the puppet script'
|
|
}
|
|
|
|
def setUp(self):
|
|
super(HookPuppetTest, self).setUp()
|
|
self.hook_path = self.relative_path(
|
|
__file__,
|
|
'..',
|
|
'heat-config-puppet/install.d/hook-puppet.py')
|
|
|
|
self.fake_tool_path = self.relative_path(
|
|
__file__,
|
|
'config-tool-fake.py')
|
|
|
|
self.working_dir = self.useFixture(fixtures.TempDir())
|
|
self.outputs_dir = self.useFixture(fixtures.TempDir())
|
|
self.log_dir = self.useFixture(fixtures.TempDir())
|
|
self.hiera_datadir = self.useFixture(fixtures.TempDir())
|
|
self.test_state_path = self.outputs_dir.join('test_state.json')
|
|
|
|
self.env = os.environ.copy()
|
|
self.env.update({
|
|
'HEAT_PUPPET_WORKING': self.working_dir.join(),
|
|
'HEAT_PUPPET_OUTPUTS': self.outputs_dir.join(),
|
|
'HEAT_PUPPET_LOGDIR': self.log_dir.join(),
|
|
'HEAT_PUPPET_HIERA_DATADIR': self.hiera_datadir.join(),
|
|
'HEAT_PUPPET_CMD': self.fake_tool_path,
|
|
'TEST_STATE_PATH': self.test_state_path,
|
|
})
|
|
|
|
def test_hook(self):
|
|
|
|
self.env.update({
|
|
'TEST_RESPONSE': json.dumps({
|
|
'stdout': 'puppet success',
|
|
'stderr': 'thing happened',
|
|
'files': {
|
|
self.outputs_dir.join('1234.first_output'): 'output 1',
|
|
self.outputs_dir.join('1234.second_output'): 'output 2',
|
|
}
|
|
}),
|
|
})
|
|
returncode, stdout, stderr = self.run_cmd(
|
|
[self.hook_path], self.env, json.dumps(self.data))
|
|
|
|
self.assertEqual(0, returncode, stderr)
|
|
self.assertEqual({
|
|
'deploy_stdout': 'puppet success',
|
|
'deploy_stderr': 'thing happened',
|
|
'deploy_status_code': 0,
|
|
'first_output': 'output 1',
|
|
'second_output': 'output 2',
|
|
}, json.loads(stdout.decode('utf-8')))
|
|
|
|
state = self.json_from_file(self.test_state_path)
|
|
puppet_script = self.working_dir.join('1234.pp')
|
|
self.assertEqual(
|
|
[
|
|
self.fake_tool_path,
|
|
'apply',
|
|
'--color',
|
|
'false',
|
|
'--detailed-exitcodes',
|
|
'--logdest',
|
|
'console',
|
|
'--debug',
|
|
'--logdest',
|
|
'/var/log/puppet/heat-debug.log',
|
|
'--verbose',
|
|
'--logdest',
|
|
'/var/log/puppet/heat-verbose.log',
|
|
puppet_script
|
|
],
|
|
state['args'])
|
|
|
|
self.assertEqual('bar', state['env']['FACTER_foo'])
|
|
self.assertEqual('input', state['env']['FACTER_another'])
|
|
self.assertEqual(self.outputs_dir.join('1234'),
|
|
state['env']['FACTER_heat_outputs_path'])
|
|
with open(puppet_script) as f:
|
|
self.assertEqual('the puppet script', f.read())
|
|
|
|
def test_hook_no_debug(self):
|
|
self.data['options']['enable_debug'] = False
|
|
self.env.update({
|
|
'TEST_RESPONSE': json.dumps({
|
|
'stdout': 'success',
|
|
'stderr': '',
|
|
}),
|
|
})
|
|
returncode, stdout, stderr = self.run_cmd(
|
|
[self.hook_path], self.env, json.dumps(self.data))
|
|
|
|
state = self.json_from_file(self.test_state_path)
|
|
puppet_script = self.working_dir.join('1234.pp')
|
|
self.assertEqual(
|
|
[
|
|
self.fake_tool_path,
|
|
'apply',
|
|
'--color',
|
|
'false',
|
|
'--detailed-exitcodes',
|
|
'--logdest',
|
|
'console',
|
|
'--verbose',
|
|
'--logdest',
|
|
'/var/log/puppet/heat-verbose.log',
|
|
puppet_script
|
|
],
|
|
state['args'])
|
|
self.data['options']['enable_debug'] = True
|
|
|
|
def test_hook_no_verbose(self):
|
|
self.data['options']['enable_verbose'] = False
|
|
self.env.update({
|
|
'TEST_RESPONSE': json.dumps({
|
|
'stdout': 'success',
|
|
'stderr': '',
|
|
}),
|
|
})
|
|
returncode, stdout, stderr = self.run_cmd(
|
|
[self.hook_path], self.env, json.dumps(self.data))
|
|
|
|
state = self.json_from_file(self.test_state_path)
|
|
puppet_script = self.working_dir.join('1234.pp')
|
|
self.assertEqual(
|
|
[
|
|
self.fake_tool_path,
|
|
'apply',
|
|
'--color',
|
|
'false',
|
|
'--detailed-exitcodes',
|
|
'--logdest',
|
|
'console',
|
|
'--debug',
|
|
'--logdest',
|
|
'/var/log/puppet/heat-debug.log',
|
|
puppet_script
|
|
],
|
|
state['args'])
|
|
self.data['options']['enable_verbose'] = True
|
|
|
|
def test_hook_puppet_failed(self):
|
|
|
|
self.env.update({
|
|
'TEST_RESPONSE': json.dumps({
|
|
'stdout': 'puppet failed',
|
|
'stderr': 'bad thing happened',
|
|
'returncode': 4
|
|
}),
|
|
})
|
|
returncode, stdout, stderr = self.run_cmd(
|
|
[self.hook_path], self.env, json.dumps(self.data))
|
|
|
|
self.assertEqual(0, returncode, stderr)
|
|
self.assertEqual({
|
|
'deploy_stdout': 'puppet failed',
|
|
'deploy_stderr': 'bad thing happened',
|
|
'deploy_status_code': 4,
|
|
}, json.loads(stdout.decode('utf-8')))
|
|
|
|
state = self.json_from_file(self.test_state_path)
|
|
puppet_script = self.working_dir.join('1234.pp')
|
|
self.assertEqual(
|
|
[
|
|
self.fake_tool_path,
|
|
'apply',
|
|
'--color',
|
|
'false',
|
|
'--detailed-exitcodes',
|
|
'--logdest',
|
|
'console',
|
|
'--debug',
|
|
'--logdest',
|
|
'/var/log/puppet/heat-debug.log',
|
|
'--verbose',
|
|
'--logdest',
|
|
'/var/log/puppet/heat-verbose.log',
|
|
puppet_script
|
|
],
|
|
state['args'])
|
|
|
|
self.assertEqual('bar', state['env']['FACTER_foo'])
|
|
self.assertEqual('input', state['env']['FACTER_another'])
|
|
self.assertEqual(self.outputs_dir.join('1234'),
|
|
state['env']['FACTER_heat_outputs_path'])
|
|
with open(puppet_script) as f:
|
|
self.assertEqual('the puppet script', f.read())
|
|
|
|
def test_hook_hiera(self):
|
|
|
|
self.env.update({
|
|
'TEST_RESPONSE': json.dumps({
|
|
'stdout': 'puppet success',
|
|
'stderr': 'thing happened',
|
|
'files': {
|
|
self.outputs_dir.join('1234.first_output'): 'output 1',
|
|
self.outputs_dir.join('1234.second_output'): 'output 2',
|
|
}
|
|
}),
|
|
})
|
|
modulepath = self.working_dir.join()
|
|
data = copy.deepcopy(self.data)
|
|
data['options']['modulepath'] = modulepath
|
|
data['options']['tags'] = 'package,file'
|
|
returncode, stdout, stderr = self.run_cmd(
|
|
[self.hook_path], self.env, json.dumps(data))
|
|
|
|
self.assertEqual(0, returncode, stderr)
|
|
self.assertEqual({
|
|
'deploy_stdout': 'puppet success',
|
|
'deploy_stderr': 'thing happened',
|
|
'deploy_status_code': 0,
|
|
'first_output': 'output 1',
|
|
'second_output': 'output 2',
|
|
}, json.loads(stdout.decode('utf-8')))
|
|
|
|
state = self.json_from_file(self.test_state_path)
|
|
puppet_script = self.working_dir.join('1234.pp')
|
|
hiera_datafile = self.hiera_datadir.join('heat_config_%s.json'
|
|
% self.data['name'])
|
|
self.assertEqual(
|
|
[
|
|
self.fake_tool_path,
|
|
'apply',
|
|
'--color',
|
|
'false',
|
|
'--detailed-exitcodes',
|
|
'--logdest',
|
|
'console',
|
|
'--modulepath',
|
|
modulepath,
|
|
'--tags',
|
|
'package,file',
|
|
'--debug',
|
|
'--logdest',
|
|
'/var/log/puppet/heat-debug.log',
|
|
'--verbose',
|
|
'--logdest',
|
|
'/var/log/puppet/heat-verbose.log',
|
|
puppet_script
|
|
],
|
|
state['args'])
|
|
|
|
self.assertEqual(self.outputs_dir.join('1234'),
|
|
state['env']['FACTER_heat_outputs_path'])
|
|
with open(puppet_script) as f:
|
|
self.assertEqual('the puppet script', f.read())
|
|
with open(hiera_datafile) as f:
|
|
self.assertEqual({
|
|
'foo': 'bar',
|
|
'another': 'input',
|
|
}, json.loads(f.read()))
|