def test_no_op(self):
bootstrap = Bootstrap(ztps_default_config=True)
version = random_string()
bootstrap.eapi.version = version
bootstrap.ztps.set_definition_response(
actions=[{
'action': 'test_action',
'attributes': {
'url': random_string(),
'version': version
}
}, {
'action': 'startup_config_action'
}])
bootstrap.ztps.set_action_response('test_action',
get_action('install_image'))
bootstrap.ztps.set_action_response('startup_config_action',
startup_config_action())
bootstrap.start_test()
try:
self.failUnless(bootstrap.success())
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
bootstrap.end_test()
def test(self):
open(RC_EOS, 'w').write(random_string())
open(BOOT_EXTENSIONS, 'w').write(random_string())
os.makedirs(BOOT_EXTENSIONS_FOLDER)
open('%s/%s' % (BOOT_EXTENSIONS_FOLDER, random_string()),
'w').write(random_string())
bootstrap = Bootstrap()
bootstrap.ztps.set_config_response()
bootstrap.ztps.set_node_check_response()
bootstrap.ztps.set_definition_response()
bootstrap.start_test()
try:
self.failUnless(bootstrap.eapi_node_information_collected())
self.failUnless(bootstrap.missing_startup_config_failure())
self.failIf(os.path.exists(RC_EOS))
self.failIf(os.path.exists(BOOT_EXTENSIONS))
self.failIf(os.path.exists(BOOT_EXTENSIONS_FOLDER))
self.failIf(bootstrap.error)
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
bootstrap.end_test()
def test_no_downgrade(self):
bootstrap = Bootstrap(ztps_default_config=True)
version1 = '4.18.1F'
version2 = '4.17.2F'
bootstrap.eapi.version = version1
bootstrap.ztps.set_definition_response(
actions=[{
'action': 'test_action',
'attributes': {
'downgrade': False,
'url': random_string(),
'version': version2
}
}, {
'action': 'startup_config_action'
}])
bootstrap.ztps.set_action_response('test_action',
get_action('install_image'))
bootstrap.ztps.set_action_response('startup_config_action',
startup_config_action())
bootstrap.start_test()
image_file = '%s/EOS-%s.swi' % (bootstrap.flash, version2)
try:
self.failUnless(bootstrap.success())
self.failUnless('install_image: nothing to do: downgrade disabled'
in bootstrap.output)
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
remove_file(image_file)
bootstrap.end_test()
def test_variables_loose(self):
bootstrap = Bootstrap(ztps_default_config=True)
config = random_string()
url = config
var_dict = {'a': 'A', 'b': 'A', 'xxx': '999', 'dummy': 'DUMMY'}
bootstrap.ztps.set_definition_response(actions=[{
'action': 'test_action',
'attributes': {
'url': url,
'substitution_mode': 'loose',
'variables': var_dict
}
}])
bootstrap.ztps.set_action_response('test_action',
get_action('add_config'))
contents = '$a 1234 $b 4 321 $xxx$a'
expected_contents = Template(contents).safe_substitute(var_dict)
bootstrap.ztps.set_file_response(config, contents)
bootstrap.start_test()
try:
self.failUnless(os.path.isfile(bootstrap.startup_config))
self.failUnless(
[expected_contents] == file_log(bootstrap.startup_config))
self.failUnless(bootstrap.success())
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
bootstrap.end_test()
def test(self):
bootstrap = Bootstrap()
open(bootstrap.rc_eos, 'w').write(random_string())
open(bootstrap.boot_extensions, 'w').write(random_string())
os.makedirs(bootstrap.boot_extensions_folder)
open('%s/%s' % (bootstrap.boot_extensions_folder,
random_string()),
'w').write(random_string())
bootstrap.ztps.set_config_response()
bootstrap.ztps.set_node_check_response()
bootstrap.ztps.set_definition_response()
bootstrap.start_test()
try:
self.failUnless(bootstrap.eapi_node_information_collected())
self.failUnless(bootstrap.missing_startup_config_failure())
self.failIf(os.path.exists(bootstrap.rc_eos))
self.failIf(os.path.exists(bootstrap.boot_extensions))
self.failIf(os.path.exists(bootstrap.boot_extensions_folder))
self.failIf(bootstrap.error)
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
bootstrap.end_test()
def test_attribute(self):
bootstrap = Bootstrap()
bootstrap.ztps.set_config_response()
bootstrap.ztps.set_node_check_response()
text = random_string()
bootstrap.ztps.set_definition_response(
actions=[{
'action': 'startup_config_action'
}, {
'action': 'print_action',
'attributes': {
'print_action-attr': text
}
}])
bootstrap.ztps.set_action_response('startup_config_action',
startup_config_action())
bootstrap.ztps.set_action_response('print_action',
print_action(use_attribute=True))
bootstrap.start_test()
try:
self.failUnless(bootstrap.eapi_node_information_collected())
self.failUnless(bootstrap.success())
self.failUnless(text in bootstrap.output)
self.failIf(bootstrap.error)
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
bootstrap.end_test()
def test_success(self):
bootstrap = Bootstrap(ztps_default_config=True)
version = random_string()
image = random_string()
url = 'http://%s/%s' % (bootstrap.server, image)
bootstrap.ztps.set_definition_response(
actions=[{'action' : 'test_action',
'attributes' : {
'url' : url,
'version' : version}},
{'action' :'startup_config_action'}])
action = get_action('install_image')
bootstrap.ztps.set_action_response('test_action',
action)
bootstrap.ztps.set_action_response('startup_config_action',
startup_config_action())
bootstrap.ztps.set_file_response(image, print_action())
bootstrap.start_test()
image_file = '%s/EOS-%s.swi' % (FLASH, version)
try:
self.failUnless(os.path.isfile(image_file))
self.failUnless(bootstrap.success())
self.failUnless(eapi_log()[-1] ==
'install source flash:EOS-%s.swi' % version)
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
remove_file(image_file)
bootstrap.end_test()
def test_failure(self):
bootstrap = Bootstrap(ztps_default_config=True)
config = random_string()
url = config
bootstrap.ztps.set_definition_response(actions=[{
'action': 'test_action',
'attributes': {
'url': url
}
}])
bootstrap.ztps.set_action_response('test_action',
get_action('run_cli_commands'))
contents = random_string()
bootstrap.ztps.set_file_response(config, contents)
bootstrap.eapi.add_failing_command(contents)
bootstrap.start_test()
try:
self.failUnless(eapi_log()[-1] == contents)
self.failUnless(bootstrap.action_failure())
self.failUnless('Running CLI commands [\'%s\'] failed' %
contents in bootstrap.output)
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
bootstrap.end_test()
def test_success(self):
bootstrap = Bootstrap(ztps_default_config=True)
smarthost = "%s:2525" % str(bootstrap.server).split(':')[0]
bootstrap.ztps.set_definition_response(
actions=[{
'action': 'startup_config_action'
}, {
'action': 'test_action',
'attributes': {
'smarthost': smarthost,
'sender': 'ztps@localhost',
'receivers': ['ztps@localhost']
}
}])
bootstrap.ztps.set_action_response('startup_config_action',
startup_config_action())
action = get_action('send_email')
bootstrap.ztps.set_action_response('test_action', action)
bootstrap.start_test()
try:
self.failUnless(bootstrap.success())
except Exception as exception: #pylint: disable=W0703
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(exception)
finally:
bootstrap.end_test()
def test_variables_loose(self):
bootstrap = Bootstrap(ztps_default_config=True)
config = random_string()
url = config
var_dict = { 'a' : 'A',
'b' : 'A',
'xxx' : '999',
'dummy': 'DUMMY'}
bootstrap.ztps.set_definition_response(
actions=[{'action' : 'test_action',
'attributes': {'url' : url,
'substitution_mode': 'loose',
'variables': var_dict}}])
bootstrap.ztps.set_action_response('test_action',
get_action('add_config'))
contents = '$a 1234 $b 4 321 $xxx$a'
expected_contents = Template(contents).safe_substitute(var_dict)
bootstrap.ztps.set_file_response(config, contents)
bootstrap.start_test()
try:
self.failUnless(os.path.isfile(bootstrap.startup_config))
self.failUnless([expected_contents] ==
file_log(bootstrap.startup_config))
self.failUnless(bootstrap.success())
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
bootstrap.end_test()
def test_multiple_actions(self):
bootstrap = Bootstrap()
bootstrap.ztps.set_config_response()
bootstrap.ztps.set_node_check_response()
bootstrap.ztps.set_definition_response(
actions=[{'action' : 'startup_config_action'},
{'action' : 'print_action_1'},
{'action' : 'print_action_2'}])
bootstrap.ztps.set_action_response('startup_config_action',
startup_config_action())
text_1 = random_string()
bootstrap.ztps.set_action_response('print_action_1',
print_action(text_1))
text_2 = random_string()
bootstrap.ztps.set_action_response('print_action_2',
print_action(text_2))
bootstrap.start_test()
try:
self.failUnless(bootstrap.eapi_node_information_collected())
self.failUnless(bootstrap.success())
self.failUnless(text_1 in bootstrap.output)
self.failUnless(text_2 in bootstrap.output)
self.failIf(bootstrap.error)
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise assertion
finally:
bootstrap.end_test()
def test_keep_backup(self):
bootstrap = Bootstrap(ztps_default_config=True)
source = random_string()
destination = '/tmp/%s' % random_string()
url = 'http://%s/%s' % (bootstrap.server, source)
bootstrap.ztps.set_definition_response(
actions=[{
'action': 'startup_config_action'
}, {
'action': 'test_action',
'attributes': {
'src_url': url,
'dst_url': destination,
'overwrite': 'backup'
}
}])
bootstrap.ztps.set_action_response('startup_config_action',
startup_config_action())
action = get_action('copy_file')
# Make the destinaton persistent
action = action.replace('PERSISTENT_STORAGE = [',
'PERSISTENT_STORAGE = [\'%s\', ' % destination)
bootstrap.ztps.set_action_response('test_action', action)
contents = random_string()
bootstrap.ztps.set_file_response(source, contents)
destination_path = '%s/%s' % (destination, source)
backup_contents = random_string()
os.makedirs(destination)
file_descriptor = open(destination_path, 'w')
file_descriptor.write(backup_contents)
file_descriptor.close()
bootstrap.start_test()
backup_path = '%s.backup' % destination_path
try:
self.failUnless(os.path.isfile(destination_path))
self.failUnless([contents] == file_log(destination_path))
self.failUnless(os.path.isfile(backup_path))
self.failUnless([backup_contents] == file_log(backup_path))
self.failIf(os.path.isfile(bootstrap.rc_eos))
self.failUnless(bootstrap.success())
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
remove_file(destination_path)
remove_file(backup_path)
shutil.rmtree(destination)
bootstrap.end_test()
def test_replace(self):
bootstrap = Bootstrap(ztps_default_config=True)
source = random_string()
destination = random_string()
url = 'http://%s/%s' % (bootstrap.server, source)
attributes = {'src_url': url, 'dst_url': destination}
# 'replace' is the default
if bool(random.getrandbits(1)):
attributes['overwrite'] = 'replace'
mode = None
if True or bool(random.getrandbits(1)):
mode = random_permissions()
attributes['mode'] = mode
bootstrap.ztps.set_definition_response(
actions=[{
'action': 'startup_config_action'
}, {
'action': 'test_action',
'attributes': attributes
}])
bootstrap.ztps.set_action_response('startup_config_action',
startup_config_action())
persistent_dir = '/tmp'
action = get_action('copy_file')
action = action.replace('/mnt/flash/.ztp-files', persistent_dir)
bootstrap.ztps.set_action_response('test_action', action)
contents = random_string()
bootstrap.ztps.set_file_response(source, contents)
destination_path = '%s/%s' % (persistent_dir, source)
bootstrap.start_test()
try:
self.failUnless(os.path.isfile(destination_path))
self.failUnless([contents] == file_log(destination_path))
self.failUnless(os.path.isfile(bootstrap.rc_eos))
log = file_log(bootstrap.rc_eos)
self.failUnless('#!/bin/bash' in log)
self.failUnless('sudo cp %s %s' %
(destination_path, destination) in log)
if mode:
self.failUnless('sudo chmod %s %s' %
(mode, destination) in log)
self.failUnless(bootstrap.success())
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
remove_file(destination_path)
bootstrap.end_test()
def test_success(self):
bootstrap = Bootstrap(ztps_default_config=True)
config = random_string()
url = config
bootstrap.ztps.set_definition_response(
actions=[{
'action': 'startup_config_action'
}, {
'action': 'test_action',
'attributes': {
'url': url
}
}])
bootstrap.ztps.set_action_response('startup_config_action',
startup_config_action())
bootstrap.ztps.set_action_response('test_action',
get_action('run_bash_script'))
print_string = random_string()
contents = '''#!/usr/bin/env python
print "%s"''' % print_string
bootstrap.ztps.set_file_response(config, contents)
bootstrap.start_test()
try:
self.failUnless(os.path.isfile(bootstrap.startup_config))
self.failUnless(print_string in bootstrap.output)
self.failUnless(bootstrap.success())
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
bootstrap.end_test()
def test_append(self):
bootstrap = Bootstrap(ztps_default_config=True)
config = random_string()
url = 'http://%s/%s' % (bootstrap.server, config)
bootstrap.ztps.set_definition_response(
actions=[{'action' : 'startup_config_action'},
{'action' : 'test_action',
'attributes': {'url' : url}}])
bootstrap.ztps.set_action_response('test_action',
get_action('add_config'))
startup_config_text = random_string()
bootstrap.ztps.set_action_response(
'startup_config_action',
startup_config_action(lines=[startup_config_text]))
contents = random_string()
bootstrap.ztps.set_file_response(config, contents)
bootstrap.start_test()
try:
self.failUnless(os.path.isfile(STARTUP_CONFIG))
log = file_log(STARTUP_CONFIG)
self.failUnless(contents in log)
self.failUnless(startup_config_text in log)
self.failUnless(bootstrap.success())
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise assertion
finally:
bootstrap.end_test()
def test_success(self):
bootstrap = Bootstrap(ztps_default_config=True)
version = random_string()
image = random_string()
url = 'http://%s/%s' % (bootstrap.server, image)
bootstrap.ztps.set_definition_response(
actions=[{'action' : 'test_action',
'attributes' : {
'url' : url,
'version' : version}}])
boot_file = '/tmp/boot-config'
action = get_action('install_image')
action = action.replace('/mnt/flash/boot-config',
boot_file)
bootstrap.ztps.set_action_response('test_action',
action)
bootstrap.ztps.set_file_response(image, print_action())
bootstrap.start_test()
image_file = '%s/%s.swi' % (FLASH, version)
try:
self.failUnless('! boot system flash:/%s.swi' % version
in file_log(STARTUP_CONFIG))
self.failUnless(os.path.isfile(image_file))
self.failUnless(['SWI=flash:/%s.swi' % version] ==
file_log(boot_file))
self.failUnless(bootstrap.success())
except AssertionError:
raise
finally:
remove_file(image_file)
remove_file(boot_file)
bootstrap.end_test()
def test_attribute(self):
bootstrap = Bootstrap()
bootstrap.ztps.set_config_response()
bootstrap.ztps.set_node_check_response()
text = random_string()
bootstrap.ztps.set_definition_response(
actions=[{'action' : 'startup_config_action'},
{'action' : 'print_action',
'attributes' : {'print_action-attr':text}}])
bootstrap.ztps.set_action_response('startup_config_action',
startup_config_action())
bootstrap.ztps.set_action_response('print_action',
print_action(use_attribute=True))
bootstrap.start_test()
try:
self.failUnless(bootstrap.eapi_node_information_collected())
self.failUnless(bootstrap.success())
self.failUnless(text in bootstrap.output)
self.failIf(bootstrap.error)
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise assertion
finally:
bootstrap.end_test()
def test_failure(self):
bootstrap = Bootstrap(ztps_default_config=True)
config = random_string()
url = config
bootstrap.ztps.set_definition_response(
actions=[{'action' : 'test_action',
'attributes': {'url' : url}}])
bootstrap.ztps.set_action_response('test_action',
get_action('run_cli_commands'))
contents = random_string()
bootstrap.ztps.set_file_response(config, contents)
bootstrap.eapi.add_failing_command(contents)
bootstrap.start_test()
try:
self.failUnless(eapi_log()[-1] == contents)
self.failUnless(bootstrap.action_failure())
self.failUnless('Running CLI commands [\'%s\'] failed' %
contents in bootstrap.output)
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
bootstrap.end_test()
def test_duplicate_actions(self):
bootstrap = Bootstrap()
bootstrap.ztps.set_config_response()
bootstrap.ztps.set_node_check_response()
bootstrap.ztps.set_definition_response(
actions=[{'action' : 'startup_config_action'},
{'action' : 'print_action'},
{'action' : 'print_action'}])
bootstrap.ztps.set_action_response('startup_config_action',
startup_config_action())
text = random_string()
bootstrap.ztps.set_action_response('print_action',
print_action(text))
bootstrap.start_test()
try:
self.failUnless(bootstrap.eapi_node_information_collected())
self.failUnless(bootstrap.success())
self.failUnless(bootstrap.output.count('Downloading action '
'print_action') == 1)
self.failUnless(bootstrap.output.count('Executing action '
'print_action') == 2)
self.failUnless(bootstrap.output.count(text) == 2)
self.failIf(bootstrap.error)
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise assertion
finally:
bootstrap.end_test()
def test_success(self):
bootstrap = Bootstrap(ztps_default_config=True)
config = random_string()
url = config
bootstrap.ztps.set_definition_response(
actions=[{'action' : 'startup_config_action'},
{'action' : 'test_action',
'attributes': {'url' : url}}])
bootstrap.ztps.set_action_response(
'startup_config_action', startup_config_action())
bootstrap.ztps.set_action_response('test_action',
get_action('run_cli_commands'))
contents = random_string()
bootstrap.ztps.set_file_response(config, contents)
bootstrap.start_test()
try:
self.failUnless(os.path.isfile(bootstrap.startup_config))
self.failUnless(eapi_log()[-1] == contents)
self.failUnless(bootstrap.success())
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
bootstrap.end_test()
def test_multi_lines(self):
bootstrap = Bootstrap(ztps_default_config=True)
config = random_string()
url = 'http://%s/%s' % (bootstrap.server, config)
bootstrap.ztps.set_definition_response(
actions=[{'action' : 'startup_config_action'},
{'action' : 'test_action',
'attributes': {'url' : url}}])
bootstrap.ztps.set_action_response('test_action',
get_action('add_config'))
startup_config_lines = [random_string(), random_string(),
random_string(), random_string()]
bootstrap.ztps.set_action_response(
'startup_config_action',
startup_config_action(lines=startup_config_lines))
contents = '\n'.join([random_string(), random_string(),
random_string(), random_string(),
random_string(), random_string()])
bootstrap.ztps.set_file_response(config, contents)
bootstrap.start_test()
try:
self.failUnless(os.path.isfile(bootstrap.startup_config))
log = file_log(bootstrap.startup_config)
all_lines = startup_config_lines + contents.split()
for line in all_lines:
self.failUnless(line in log)
self.failUnless(bootstrap.success())
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
bootstrap.end_test()
def test_no_downgrade(self):
bootstrap = Bootstrap(ztps_default_config=True)
version1 = '4.18.1F'
version2 = '4.17.2F'
bootstrap.eapi.version = version1
bootstrap.ztps.set_definition_response(
actions=[{'action' : 'test_action',
'attributes': {
'downgrade' : False,
'url' : random_string(),
'version' : version2}},
{'action' :'startup_config_action'}])
bootstrap.ztps.set_action_response('test_action',
get_action('install_image'))
bootstrap.ztps.set_action_response('startup_config_action',
startup_config_action())
bootstrap.start_test()
image_file = '%s/EOS-%s.swi' % (bootstrap.flash, version2)
try:
self.failUnless(bootstrap.success())
self.failUnless('install_image: nothing to do: downgrade disabled'
in bootstrap.output)
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
remove_file(image_file)
bootstrap.end_test()
def test(self):
open(RC_EOS, 'w').write(random_string())
open(BOOT_EXTENSIONS, 'w').write(random_string())
os.makedirs(BOOT_EXTENSIONS_FOLDER)
open('%s/%s' % (BOOT_EXTENSIONS_FOLDER, random_string()),
'w').write(random_string())
bootstrap = Bootstrap()
bootstrap.ztps.set_config_response()
bootstrap.ztps.set_node_check_response()
bootstrap.ztps.set_definition_response()
bootstrap.start_test()
try:
self.failUnless(bootstrap.eapi_node_information_collected())
self.failUnless(bootstrap.missing_startup_config_failure())
self.failIf(os.path.exists(RC_EOS))
self.failIf(os.path.exists(BOOT_EXTENSIONS))
self.failIf(os.path.exists(BOOT_EXTENSIONS_FOLDER))
self.failIf(bootstrap.error)
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise assertion
finally:
bootstrap.end_test()
def test_success(self):
bootstrap = Bootstrap(ztps_default_config=True)
config = random_string()
url = config
bootstrap.ztps.set_definition_response(
actions=[{'action' : 'startup_config_action'},
{'action' : 'test_action',
'attributes': {'url' : url}}])
bootstrap.ztps.set_action_response(
'startup_config_action', startup_config_action())
bootstrap.ztps.set_action_response('test_action',
get_action('run_bash_script'))
print_string = random_string()
contents = '''#!/usr/bin/env python
print "%s"''' % print_string
bootstrap.ztps.set_file_response(config, contents)
bootstrap.start_test()
try:
self.failUnless(os.path.isfile(bootstrap.startup_config))
self.failUnless(print_string in bootstrap.output)
self.failUnless(bootstrap.success())
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
bootstrap.end_test()
def test_multiple_actions(self):
bootstrap = Bootstrap()
bootstrap.ztps.set_config_response()
bootstrap.ztps.set_node_check_response()
bootstrap.ztps.set_definition_response(
actions=[{
'action': 'startup_config_action'
}, {
'action': 'print_action_1'
}, {
'action': 'print_action_2'
}])
bootstrap.ztps.set_action_response('startup_config_action',
startup_config_action())
text_1 = random_string()
bootstrap.ztps.set_action_response('print_action_1',
print_action(text_1))
text_2 = random_string()
bootstrap.ztps.set_action_response('print_action_2',
print_action(text_2))
bootstrap.start_test()
try:
self.failUnless(bootstrap.eapi_node_information_collected())
self.failUnless(bootstrap.success())
self.failUnless(text_1 in bootstrap.output)
self.failUnless(text_2 in bootstrap.output)
self.failIf(bootstrap.error)
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
bootstrap.end_test()
def test(self):
bootstrap = Bootstrap()
open(bootstrap.rc_eos, 'w').write(random_string())
open(bootstrap.boot_extensions, 'w').write(random_string())
os.makedirs(bootstrap.boot_extensions_folder)
open('%s/%s' % (bootstrap.boot_extensions_folder, random_string()),
'w').write(random_string())
bootstrap.ztps.set_config_response()
bootstrap.ztps.set_node_check_response()
bootstrap.ztps.set_definition_response()
bootstrap.start_test()
try:
self.failUnless(bootstrap.eapi_node_information_collected())
self.failUnless(bootstrap.missing_startup_config_failure())
self.failIf(os.path.exists(bootstrap.rc_eos))
self.failIf(os.path.exists(bootstrap.boot_extensions))
self.failIf(os.path.exists(bootstrap.boot_extensions_folder))
self.failIf(bootstrap.error)
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
bootstrap.end_test()
def test_success(self):
bootstrap = Bootstrap(ztps_default_config=True)
smarthost = "%s:2525" % str(bootstrap.server).split(':')[0]
bootstrap.ztps.set_definition_response(
actions=[{'action' : 'startup_config_action'},
{'action' : 'test_action',
'attributes' : {
'smarthost': smarthost,
'sender': 'ztps@localhost',
'receivers': ['ztps@localhost']}}])
bootstrap.ztps.set_action_response('startup_config_action',
startup_config_action())
action = get_action('send_email')
bootstrap.ztps.set_action_response('test_action',
action)
bootstrap.start_test()
try:
self.failUnless(bootstrap.success())
except Exception as exception: #pylint: disable=W0703
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(exception)
finally:
bootstrap.end_test()
def test_bad_file_status(self):
bootstrap = Bootstrap(ztps_default_config=True)
config = random_string()
url = 'http://%s/%s' % (bootstrap.server, config)
bootstrap.ztps.set_definition_response(actions=[{
'action': 'test_action',
'attributes': {
'url': url
}
}])
bootstrap.ztps.set_action_response('test_action',
get_action('replace_config'))
contents = random_string()
bootstrap.ztps.set_file_response(config,
contents,
status=STATUS_NOT_FOUND)
bootstrap.start_test()
try:
self.failUnless(bootstrap.action_failure())
msg = [x for x in bootstrap.output.split('\n') if x][-1]
self.failUnless('Unable to retrieve config from URL' in msg)
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
bootstrap.end_test()
def test_success(self):
bootstrap = Bootstrap(ztps_default_config=True)
config = random_string()
url = config
bootstrap.ztps.set_definition_response(actions=[{
'action': 'test_action',
'attributes': {
'url': url
}
}])
bootstrap.ztps.set_action_response('test_action',
get_action('add_config'))
contents = random_string()
bootstrap.ztps.set_file_response(config, contents)
bootstrap.start_test()
try:
self.failUnless(os.path.isfile(bootstrap.startup_config))
self.failUnless(
contents.split() == file_log(bootstrap.startup_config))
self.failUnless(bootstrap.success())
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
bootstrap.end_test()
def bootstrap(cls, xmpp):
bootstrap = Bootstrap()
bootstrap.ztps.set_config_response(xmpp=xmpp)
bootstrap.ztps.set_node_check_response()
bootstrap.ztps.set_definition_response()
bootstrap.start_test()
return bootstrap
def test_replace(self):
bootstrap = Bootstrap(ztps_default_config=True)
source = random_string()
destination = '/tmp/%s' % random_string()
url = 'http://%s/%s' % (bootstrap.server, source)
attributes = {'src_url' : url,
'dst_url' : destination}
# 'replace' is the default
if bool(random.getrandbits(1)):
attributes['overwrite'] = 'replace'
mode = None
if True or bool(random.getrandbits(1)):
mode = random_permissions()
attributes['mode'] = mode
bootstrap.ztps.set_definition_response(
actions=[{'action' : 'startup_config_action'},
{'action' : 'test_action',
'attributes' : attributes}])
bootstrap.ztps.set_action_response(
'startup_config_action', startup_config_action())
action = get_action('copy_file')
# Make the destinaton persistent
action = action.replace('PERSISTENT_STORAGE = [',
'PERSISTENT_STORAGE = [\'%s\', ' %
destination)
bootstrap.ztps.set_action_response('test_action',
action)
contents = random_string()
bootstrap.ztps.set_file_response(source, contents)
destination_path = '%s/%s' % (destination, source)
bootstrap.start_test()
try:
self.failUnless(os.path.isfile(destination_path))
self.failUnless([contents] ==
file_log(destination_path))
self.failIf(os.path.isfile(bootstrap.rc_eos))
if mode:
self.failUnless(mode ==
oct(os.stat(destination_path)[ST_MODE])[-3:])
self.failUnless(bootstrap.success())
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
remove_file(destination_path)
shutil.rmtree(destination)
bootstrap.end_test()
def test_keep_backup(self):
bootstrap = Bootstrap(ztps_default_config=True)
source = random_string()
destination = '/tmp/%s' % random_string()
url = 'http://%s/%s' % (bootstrap.server, source)
bootstrap.ztps.set_definition_response(
actions=[{'action' : 'startup_config_action'},
{'action' : 'test_action',
'attributes' : {'src_url' : url,
'dst_url' : destination,
'overwrite' : 'backup'}}])
bootstrap.ztps.set_action_response(
'startup_config_action', startup_config_action())
action = get_action('copy_file')
# Make the destinaton persistent
action = action.replace('PERSISTENT_STORAGE = [',
'PERSISTENT_STORAGE = [\'%s\', ' %
destination)
bootstrap.ztps.set_action_response('test_action',
action)
contents = random_string()
bootstrap.ztps.set_file_response(source, contents)
destination_path = '%s/%s' % (destination, source)
backup_contents = random_string()
os.makedirs(destination)
file_descriptor = open(destination_path, 'w')
file_descriptor.write(backup_contents)
file_descriptor.close()
bootstrap.start_test()
backup_path = '%s.backup' % destination_path
try:
self.failUnless(os.path.isfile(destination_path))
self.failUnless([contents] ==
file_log(destination_path))
self.failUnless(os.path.isfile(backup_path))
self.failUnless([backup_contents] ==
file_log(backup_path))
self.failIf(os.path.isfile(bootstrap.rc_eos))
self.failUnless(bootstrap.success())
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
remove_file(destination_path)
remove_file(backup_path)
shutil.rmtree(destination)
bootstrap.end_test()
def test_replace(self):
bootstrap = Bootstrap(ztps_default_config=True)
source = random_string()
destination = random_string()
url = 'http://%s/%s' % (bootstrap.server, source)
attributes = {'src_url' : url,
'dst_url' : destination}
# 'replace' is the default
if bool(random.getrandbits(1)):
attributes['overwrite'] = 'replace'
mode = None
if True or bool(random.getrandbits(1)):
mode = random_permissions()
attributes['mode'] = mode
bootstrap.ztps.set_definition_response(
actions=[{'action' : 'startup_config_action'},
{'action' : 'test_action',
'attributes' : attributes}])
bootstrap.ztps.set_action_response(
'startup_config_action', startup_config_action())
persistent_dir = '/tmp'
action = get_action('copy_file')
action = action.replace('/mnt/flash/.ztp-files',
persistent_dir)
bootstrap.ztps.set_action_response('test_action',
action)
contents = random_string()
bootstrap.ztps.set_file_response(source, contents)
destination_path = '%s/%s' % (persistent_dir, source)
bootstrap.start_test()
try:
self.failUnless(os.path.isfile(destination_path))
self.failUnless([contents] ==
file_log(destination_path))
self.failUnless(os.path.isfile(RC_EOS))
log = file_log(RC_EOS)
self.failUnless('#!/bin/bash' in log)
self.failUnless('sudo cp %s %s' %
(destination_path, destination) in log)
if mode:
self.failUnless('sudo chmod %s %s' %
(mode, destination) in log)
self.failUnless(bootstrap.success())
except AssertionError:
raise
finally:
remove_file(destination_path)
bootstrap.end_test()
def test_replace(self):
bootstrap = Bootstrap(ztps_default_config=True)
source = random_string()
destination = '/tmp/%s' % random_string()
url = 'http://%s/%s' % (bootstrap.server, source)
attributes = {'src_url': url, 'dst_url': destination}
# 'replace' is the default
if bool(random.getrandbits(1)):
attributes['overwrite'] = 'replace'
mode = None
if True or bool(random.getrandbits(1)):
mode = random_permissions()
attributes['mode'] = mode
bootstrap.ztps.set_definition_response(
actions=[{
'action': 'startup_config_action'
}, {
'action': 'test_action',
'attributes': attributes
}])
bootstrap.ztps.set_action_response('startup_config_action',
startup_config_action())
action = get_action('copy_file')
# Make the destinaton persistent
action = action.replace('PERSISTENT_STORAGE = [',
'PERSISTENT_STORAGE = [\'%s\', ' % destination)
bootstrap.ztps.set_action_response('test_action', action)
contents = random_string()
bootstrap.ztps.set_file_response(source, contents)
destination_path = '%s/%s' % (destination, source)
bootstrap.start_test()
try:
self.failUnless(os.path.isfile(destination_path))
self.failUnless([contents] == file_log(destination_path))
self.failIf(os.path.isfile(bootstrap.rc_eos))
if mode:
self.failUnless(
mode == oct(os.stat(destination_path)[ST_MODE])[-3:])
self.failUnless(bootstrap.success())
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
remove_file(destination_path)
shutil.rmtree(destination)
bootstrap.end_test()
def test_keep_backup(self):
bootstrap = Bootstrap(ztps_default_config=True)
source = random_string()
destination = random_string()
url = 'http://%s/%s' % (bootstrap.server, source)
bootstrap.ztps.set_definition_response(
actions=[{
'action': 'startup_config_action'
}, {
'action': 'test_action',
'attributes': {
'src_url': url,
'dst_url': destination,
'overwrite': 'backup'
}
}])
bootstrap.ztps.set_action_response('startup_config_action',
startup_config_action())
persistent_dir = '/tmp'
action = get_action('copy_file')
action = action.replace('/mnt/flash/.ztp-files', persistent_dir)
bootstrap.ztps.set_action_response('test_action', action)
contents = random_string()
bootstrap.ztps.set_file_response(source, contents)
bootstrap.start_test()
destination_path = '%s/%s' % (persistent_dir, source)
try:
self.failUnless(os.path.isfile(destination_path))
self.failUnless([contents] == file_log(destination_path))
self.failUnless(os.path.isfile(bootstrap.rc_eos))
log = file_log(bootstrap.rc_eos)
self.failUnless('#!/bin/bash' in log)
self.failUnless('sudo cp %s %s' %
(destination_path, destination) in log)
self.failUnless('[ -f %s ] && sudo mv %s %s.backup' %
(destination, destination, destination) in log)
self.failUnless(bootstrap.success())
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
remove_file(destination_path)
bootstrap.end_test()
def test(self):
bootstrap = Bootstrap(server='127.0.0.2')
bootstrap.start_test()
try:
self.failUnless(bootstrap.server_connection_failure())
self.failIf(bootstrap.error)
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
bootstrap.end_test()
def test(self):
bootstrap = Bootstrap(server='127.0.0.2')
bootstrap.start_test()
try:
self.failUnless(bootstrap.server_connection_failure())
self.failIf(bootstrap.error)
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
bootstrap.end_test()
def test_keep_backup(self):
bootstrap = Bootstrap(ztps_default_config=True)
source = random_string()
destination = random_string()
url = 'http://%s/%s' % (bootstrap.server, source)
bootstrap.ztps.set_definition_response(
actions=[{'action' : 'startup_config_action'},
{'action' : 'test_action',
'attributes' : {'src_url' : url,
'dst_url' : destination,
'overwrite' : 'backup'}}])
bootstrap.ztps.set_action_response(
'startup_config_action', startup_config_action())
persistent_dir = '/tmp'
action = get_action('copy_file')
action = action.replace('/mnt/flash/.ztp-files',
persistent_dir)
bootstrap.ztps.set_action_response('test_action',
action)
contents = random_string()
bootstrap.ztps.set_file_response(source, contents)
bootstrap.start_test()
destination_path = '%s/%s' % (persistent_dir, source)
try:
self.failUnless(os.path.isfile(destination_path))
self.failUnless([contents] ==
file_log(destination_path))
self.failUnless(os.path.isfile(bootstrap.rc_eos))
log = file_log(bootstrap.rc_eos)
self.failUnless('#!/bin/bash' in log)
self.failUnless('sudo cp %s %s' %
(destination_path, destination) in log)
self.failUnless('[ -f %s ] && sudo mv %s %s.backup' %
(destination, destination,
destination) in log)
self.failUnless(bootstrap.success())
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
remove_file(destination_path)
bootstrap.end_test()
def test_content_type(self):
bootstrap = Bootstrap()
bootstrap.ztps.set_config_response(content_type='text/plain')
bootstrap.start_test()
try:
self.failUnless(bootstrap.unexpected_response_failure())
self.failIf(bootstrap.error)
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
bootstrap.end_test()
def test_success(self):
bootstrap = Bootstrap(ztps_default_config=True)
source = random_string()
destination = '/tmp/%s' % random_string()
ztps_server = 'http://%s' % bootstrap.server
url = 'http://%s/%s' % (bootstrap.server, source)
bootstrap.ztps.set_definition_response(
actions=[{
'action': 'startup_config_action'
}, {
'action': 'test_action',
'attributes': {
'src_url': url,
'dst_url': destination,
'ztps_server': ztps_server
}
}])
bootstrap.ztps.set_action_response('startup_config_action',
startup_config_action())
action = get_action('copy_file')
# Make the destinaton persistent
action = action.replace('PERSISTENT_STORAGE = [',
'PERSISTENT_STORAGE = [\'%s\', ' % destination)
bootstrap.ztps.set_action_response('test_action', action)
contents = random_string()
bootstrap.ztps.set_file_response(source, contents)
bootstrap.start_test()
destination_path = '%s/%s' % (destination, source)
try:
self.failUnless(os.path.isfile(destination_path))
self.failUnless([contents] == file_log(destination_path))
self.failIf(os.path.isfile(RC_EOS))
self.failUnless(bootstrap.success())
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
remove_file(destination_path)
shutil.rmtree(destination)
bootstrap.end_test()
def test_content_type(self):
bootstrap = Bootstrap()
bootstrap.ztps.set_config_response(content_type='text/plain')
bootstrap.start_test()
try:
self.failUnless(bootstrap.unexpected_response_failure())
self.failIf(bootstrap.error)
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
bootstrap.end_test()
def test(self):
filenames = {
'DEBUG' : '/tmp/ztps-log-%s-debug' % os.getpid(),
'ERROR' : '/tmp/ztps-log-%s-error' % os.getpid(),
'INFO' : '/tmp/ztps-log-%s-info' % os.getpid(),
'bogus' : '/tmp/ztps-log-%s-bogus' % os.getpid()
}
logging = []
for level, filename in filenames.iteritems():
logging += {'destination' : 'file:%s' % filename,
'level' : level},
for filename in filenames.itervalues():
self.failIf(os.path.isfile(filename))
bootstrap = Bootstrap()
bootstrap.ztps.set_config_response(logging=logging)
bootstrap.ztps.set_node_check_response()
bootstrap.ztps.set_definition_response()
bootstrap.start_test()
try:
self.failUnless(bootstrap.eapi_node_information_collected())
self.failUnless(bootstrap.missing_startup_config_failure())
for filename in filenames.itervalues():
self.failUnless(file_log(filename))
self.assertEquals(file_log(filenames['DEBUG'],
ignore_string='SyslogManager'),
file_log(filenames['bogus'],
ignore_string='SyslogManager'))
self.assertEquals(file_log(filenames['DEBUG'],
ignore_string='SyslogManager'),
file_log(filenames['INFO'],
ignore_string='SyslogManager'))
self.failIfEqual(file_log(filenames['DEBUG'],
ignore_string='SyslogManager'),
file_log(filenames['ERROR'],
ignore_string='SyslogManager'))
self.failUnless(set(file_log(filenames['ERROR'])).issubset(
set(file_log(filenames['DEBUG']))))
for filename in filenames.itervalues():
remove_file(filename)
self.failIf(bootstrap.error)
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise assertion
finally:
bootstrap.end_test()
def test_multi_lines(self):
bootstrap = Bootstrap(ztps_default_config=True)
config = random_string()
url = 'http://%s/%s' % (bootstrap.server, config)
bootstrap.ztps.set_definition_response(
actions=[{
'action': 'startup_config_action'
}, {
'action': 'test_action',
'attributes': {
'url': url
}
}])
bootstrap.ztps.set_action_response('test_action',
get_action('add_config'))
startup_config_lines = [
random_string(),
random_string(),
random_string(),
random_string()
]
bootstrap.ztps.set_action_response(
'startup_config_action',
startup_config_action(lines=startup_config_lines))
contents = '\n'.join([
random_string(),
random_string(),
random_string(),
random_string(),
random_string(),
random_string()
])
bootstrap.ztps.set_file_response(config, contents)
bootstrap.start_test()
try:
self.failUnless(os.path.isfile(bootstrap.startup_config))
log = file_log(bootstrap.startup_config)
all_lines = startup_config_lines + contents.split()
for line in all_lines:
self.failUnless(line in log)
self.failUnless(bootstrap.success())
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
bootstrap.end_test()
def test(self):
bootstrap = Bootstrap(eapi_port=54321)
bootstrap.ztps.set_config_response()
bootstrap.start_test()
try:
self.failUnless(bootstrap.eapi_configured())
self.failUnless(bootstrap.eapi_failure())
self.failIf(bootstrap.error)
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise assertion
finally:
bootstrap.end_test()
def test(self):
bootstrap = Bootstrap(eapi_port=54321)
bootstrap.ztps.set_config_response()
bootstrap.start_test()
try:
self.failUnless(bootstrap.eapi_configured())
self.failUnless(bootstrap.eapi_failure())
self.failIf(bootstrap.error)
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
bootstrap.end_test()
def test_status(self):
bootstrap = Bootstrap()
bootstrap.ztps.set_config_response(status=201)
bootstrap.start_test()
try:
self.failUnless(bootstrap.unexpected_response_failure())
self.assertEquals(cli_log(), [])
self.failIf(bootstrap.error)
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise assertion
finally:
bootstrap.end_test()
def test_ztps_path_success(self):
bootstrap = Bootstrap(ztps_default_config=True)
plugin = url = random_string()
ztps_server = 'http://%s' % bootstrap.server
bootstrap.ztps.set_definition_response(
actions=[{
'action': 'startup_config_action'
}, {
'action': 'test_action',
'attributes': {
'url': url,
'ztps_server': ztps_server
}
}])
bootstrap.ztps.set_action_response('startup_config_action',
startup_config_action())
plugin_dir = '/tmp'
persistent_dir = '/tmp/persistent'
action = get_action('install_cli_plugin')
action = action.replace('/usr/lib/python2.7/site-packages/CliPlugin',
plugin_dir)
action = action.replace('/mnt/flash/.ztp-plugins', persistent_dir)
bootstrap.ztps.set_action_response('test_action', action)
contents = random_string()
bootstrap.ztps.set_file_response(plugin, contents)
bootstrap.start_test()
try:
self.failUnless(os.path.isfile(bootstrap.rc_eos))
log = file_log(bootstrap.rc_eos)
self.failUnless('#!/bin/bash' in log)
self.failUnless('sudo cp %s/%s %s' %
(persistent_dir, plugin, plugin_dir) in log)
self.failUnless(contents in file_log('%s/%s' %
(persistent_dir, plugin)))
self.failUnless(bootstrap.success())
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
shutil.rmtree(persistent_dir)
bootstrap.end_test()
def test_definition_missing(self):
bootstrap = Bootstrap()
bootstrap.ztps.set_config_response()
bootstrap.ztps.set_node_check_response()
bootstrap.start_test()
try:
self.failUnless(bootstrap.eapi_node_information_collected())
self.failUnless(bootstrap.server_connection_failure())
self.failIf(bootstrap.error)
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
bootstrap.end_test()
def test_definition_missing(self):
bootstrap = Bootstrap()
bootstrap.ztps.set_config_response()
bootstrap.ztps.set_node_check_response()
bootstrap.start_test()
try:
self.failUnless(bootstrap.eapi_node_information_collected())
self.failUnless(bootstrap.server_connection_failure())
self.failIf(bootstrap.error)
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise assertion
finally:
bootstrap.end_test()
def test_node_not_found(self):
bootstrap = Bootstrap()
bootstrap.ztps.set_config_response()
bootstrap.ztps.set_node_check_response(status=400)
bootstrap.start_test()
try:
self.failUnless(bootstrap.eapi_node_information_collected())
self.failUnless(bootstrap.node_not_found_failure())
self.failIf(bootstrap.error)
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
bootstrap.end_test()
def test(self):
filenames = {
'DEBUG': '/tmp/ztps-log-%s-debug' % os.getpid(),
'ERROR': '/tmp/ztps-log-%s-error' % os.getpid(),
'INFO': '/tmp/ztps-log-%s-info' % os.getpid(),
'bogus': '/tmp/ztps-log-%s-bogus' % os.getpid()
}
logging = []
for level, filename in filenames.iteritems():
logging += {'destination': 'file:%s' % filename, 'level': level},
for filename in filenames.itervalues():
self.failIf(os.path.isfile(filename))
bootstrap = Bootstrap()
bootstrap.ztps.set_config_response(logging=logging)
bootstrap.ztps.set_node_check_response()
bootstrap.ztps.set_definition_response()
bootstrap.start_test()
try:
self.failUnless(bootstrap.eapi_node_information_collected())
self.failUnless(bootstrap.missing_startup_config_failure())
for filename in filenames.itervalues():
self.failUnless(file_log(filename))
self.assertEquals(
file_log(filenames['DEBUG'], ignore_string='SyslogManager'),
file_log(filenames['bogus'], ignore_string='SyslogManager'))
self.assertEquals(
file_log(filenames['DEBUG'], ignore_string='SyslogManager'),
file_log(filenames['INFO'], ignore_string='SyslogManager'))
self.failIfEqual(
file_log(filenames['DEBUG'], ignore_string='SyslogManager'),
file_log(filenames['ERROR'], ignore_string='SyslogManager'))
self.failUnless(
set(file_log(filenames['ERROR'])).issubset(
set(file_log(filenames['DEBUG']))))
for filename in filenames.itervalues():
remove_file(filename)
self.failIf(bootstrap.error)
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
bootstrap.end_test()
def test_append_server(self):
bootstrap = Bootstrap(ztps_default_config=True)
source = random_string()
destination = '/tmp/%s' % random_string()
ztps_server = 'http://%s' % bootstrap.server
bootstrap.ztps.set_definition_response(
actions=[{'action' : 'startup_config_action'},
{'action' : 'test_action',
'attributes' : {'src_url' : source,
'dst_url' : destination,
'ztps_server': ztps_server}}])
bootstrap.ztps.set_action_response(
'startup_config_action', startup_config_action())
action = get_action('copy_file')
# Make the destinaton persistent
action = action.replace('PERSISTENT_STORAGE = [',
'PERSISTENT_STORAGE = [\'%s\', ' %
destination)
bootstrap.ztps.set_action_response('test_action',
action)
contents = random_string()
bootstrap.ztps.set_file_response(source, contents)
bootstrap.start_test()
destination_path = '%s/%s' % (destination, source)
try:
self.failUnless(os.path.isfile(destination_path))
self.failUnless([contents] ==
file_log(destination_path))
self.failIf(os.path.isfile(bootstrap.rc_eos))
self.failUnless(bootstrap.success())
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
remove_file(destination_path)
shutil.rmtree(destination)
bootstrap.end_test()
def test_success(self):
bootstrap = Bootstrap(ztps_default_config=True)
source = random_string()
destination = random_string()
url = 'http://%s/%s' % (bootstrap.server, source)
bootstrap.ztps.set_definition_response(
actions=[{'action' : 'startup_config_action'},
{'action' : 'test_action',
'attributes' : {'src_url' : url,
'dst_url' : destination}}])
bootstrap.ztps.set_action_response(
'startup_config_action', startup_config_action())
persistent_dir = '/tmp'
action = get_action('copy_file')
action = action.replace('/mnt/flash/.ztp-files',
persistent_dir)
bootstrap.ztps.set_action_response('test_action',
action)
contents = random_string()
bootstrap.ztps.set_file_response(source, contents)
bootstrap.start_test()
destination_path = '%s/%s' % (persistent_dir, source)
try:
self.failUnless(os.path.isfile(destination_path))
self.failUnless([contents] ==
file_log(destination_path))
self.failUnless(os.path.isfile(RC_EOS))
log = file_log(RC_EOS)
self.failUnless('#!/bin/bash' in log)
self.failUnless('sudo cp %s %s' %
(destination_path, destination) in log)
self.failUnless(bootstrap.success())
except AssertionError:
raise
finally:
remove_file(destination_path)
bootstrap.end_test()
def test_ztps_path_success(self):
bootstrap = Bootstrap(ztps_default_config=True)
plugin = url = random_string()
ztps_server = 'http://%s' % bootstrap.server
bootstrap.ztps.set_definition_response(
actions=[{'action' : 'startup_config_action'},
{'action' : 'test_action',
'attributes' : {'url' : url,
'ztps_server': ztps_server}}])
bootstrap.ztps.set_action_response(
'startup_config_action', startup_config_action())
plugin_dir = '/tmp'
persistent_dir = '/tmp/persistent'
action = get_action('install_cli_plugin')
action = action.replace('/usr/lib/python2.7/site-packages/CliPlugin',
plugin_dir)
action = action.replace('/mnt/flash/.ztp-plugins',
persistent_dir)
bootstrap.ztps.set_action_response('test_action',
action)
contents = random_string()
bootstrap.ztps.set_file_response(plugin, contents)
bootstrap.start_test()
try:
self.failUnless(os.path.isfile(RC_EOS))
log = file_log(RC_EOS)
self.failUnless('#!/bin/bash' in log)
self.failUnless('sudo cp %s/%s %s' %
(persistent_dir, plugin, plugin_dir) in log)
self.failUnless(contents in
file_log('%s/%s' % (persistent_dir, plugin)))
self.failUnless(bootstrap.success())
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
shutil.rmtree(persistent_dir)
bootstrap.end_test()
def test_topology_check(self):
bootstrap = Bootstrap()
bootstrap.ztps.set_config_response()
bootstrap.ztps.set_node_check_response()
bootstrap.ztps.set_definition_response(status=400,
content_type='text/html')
bootstrap.start_test()
try:
self.failUnless(bootstrap.eapi_node_information_collected())
self.failUnless(bootstrap.toplogy_check_failure())
self.failIf(bootstrap.error)
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
bootstrap.end_test()
def test_action_success_log(self):
log = '/tmp/ztps-log-%s-debug' % os.getpid()
bootstrap = Bootstrap()
bootstrap.ztps.set_config_response(logging=[
{
'destination': 'file:%s' % log,
'level': 'DEBUG'
},
])
bootstrap.ztps.set_node_check_response()
text_onstart = random_string()
text_onsuccess = random_string()
text_onfailure = random_string()
bootstrap.ztps.set_definition_response(
actions=[{
'action': 'startup_config_action',
'onstart': text_onstart,
'onsuccess': text_onsuccess,
'onfailure': text_onfailure,
}])
bootstrap.ztps.set_action_response('startup_config_action',
startup_config_action())
bootstrap.start_test()
try:
self.failUnless(bootstrap.eapi_node_information_collected())
self.failUnless(bootstrap.success())
self.failIf(bootstrap.error)
log = ''.join(file_log(log))
self.failUnless('startup_config_action: %s' % text_onstart in log)
self.failUnless('startup_config_action: %s' %
text_onsuccess in log)
self.failUnless('startup_config_action: %s' %
text_onfailure not in log)
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
bootstrap.end_test()
def action_fail_test(self, action):
bootstrap = Bootstrap()
bootstrap.ztps.set_config_response()
bootstrap.ztps.set_node_check_response()
bootstrap.ztps.set_definition_response(actions=[{
'action': 'test_action'
}])
bootstrap.ztps.set_action_response('test_action', action)
bootstrap.start_test()
try:
self.failUnless(bootstrap.eapi_node_information_collected())
self.failUnless(bootstrap.action_failure())
self.failIf(bootstrap.error)
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
bootstrap.end_test()
def test_action_failure(self):
bootstrap = Bootstrap()
bootstrap.ztps.set_config_response()
bootstrap.ztps.set_node_check_response()
bootstrap.ztps.set_definition_response(
actions=[{'action' : 'test_action'}])
flash_filename = random_string()
bootstrap.ztps.set_action_response(
'test_action',
fail_flash_file_action(bootstrap.flash,
flash_filename))
open(bootstrap.rc_eos, 'w').write(random_string())
open(bootstrap.startup_config, 'w').write(random_string())
open(bootstrap.boot_extensions, 'w').write(random_string())
os.mkdir(bootstrap.boot_extensions_folder)
open('%s/my_extension' %
bootstrap.boot_extensions_folder, 'w').write(random_string())
bootstrap.start_test()
try:
self.failUnless(bootstrap.eapi_node_information_collected())
self.failUnless(bootstrap.action_failure())
self.failIf(bootstrap.error)
self.failIf(os.path.isfile('%s/%s' %
(bootstrap.flash,
flash_filename)))
self.failIf(os.path.isfile(bootstrap.rc_eos))
self.failIf(os.path.isfile(bootstrap.startup_config))
self.failIf(os.path.isfile(bootstrap.boot_extensions))
self.failIf(os.path.isdir(bootstrap.boot_extensions_folder))
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
bootstrap.end_test()
def test_variables(self):
bootstrap = Bootstrap(ztps_default_config=True)
config = random_string()
url = config
var_dict = {'a': 'A', 'b': 'A', 'xxx': '999', 'dummy': 'DUMMY'}
bootstrap.ztps.set_definition_response(
actions=[{
'action': 'startup_config_action'
}, {
'action': 'test_action',
'attributes': {
'url': url,
'substitution_mode': 'strict',
'variables': var_dict
}
}])
bootstrap.ztps.set_action_response('startup_config_action',
startup_config_action())
bootstrap.ztps.set_action_response('test_action',
get_action('run_bash_script'))
print_string = '$a 1234 $b 4 321 $xxx$a'
contents = '''#!/usr/bin/env bash
echo %s''' % print_string
bootstrap.ztps.set_file_response(config, contents)
bootstrap.start_test()
expected_contents = Template(print_string).substitute(var_dict)
try:
self.failUnless(os.path.isfile(bootstrap.startup_config))
self.failUnless(expected_contents in bootstrap.output)
self.failUnless(bootstrap.success())
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
bootstrap.end_test()