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_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_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_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_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_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_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_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_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_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 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)
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_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_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_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_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_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_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 = '/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 = '/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_url_failure(self):
action = get_action('copy_file')
action = action.replace('/mnt/flash/.ztp-files',
'/tmp')
self.basic_test('copy_file', 4,
attributes={'src_url' :
random_string(),
'dst_url' :
random_string()},
action_value=action)
def test_url_failure(self):
action = get_action('copy_file')
action = action.replace('/mnt/flash/.ztp-files', '/tmp')
self.basic_test('copy_file',
'Unable to retrieve file from URL',
attributes={
'src_url': random_string(),
'dst_url': random_string()
},
action_value=action)
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_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_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_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_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_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_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('replace_config'))
contents = random_string()
bootstrap.ztps.set_file_response(config, contents)
bootstrap.start_test()
try:
self.failUnless(os.path.isfile(STARTUP_CONFIG))
self.failUnless(contents.split() == file_log(STARTUP_CONFIG))
self.failUnless(bootstrap.success())
except AssertionError:
raise
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_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()
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:
raise
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('return code 2' in msg)
except AssertionError:
raise
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()
def test_url_success(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)
bootstrap.start_test()
try:
self.failUnless(os.path.isfile(STARTUP_CONFIG))
self.failUnless(contents.split() == file_log(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_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_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' % (bootstrap.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_success(self):
bootstrap = Bootstrap(ztps_default_config=True)
extension = random_string()
url = extension
extension_force = random_string()
url_force = 'http://%s/%s' % (bootstrap.server, extension_force)
bootstrap.ztps.set_definition_response(
actions=[{
'action': 'startup_config_action'
}, {
'action': 'test_action',
'attributes': {
'url': url
}
}, {
'action': 'test_action_force',
'attributes': {
'url': url_force,
'force': True
}
}])
bootstrap.ztps.set_action_response('startup_config_action',
startup_config_action())
extensions_dir = '/tmp/extensions'
boot_extensions = '/tmp/boot-extensions'
action = get_action('install_extension')
action = action.replace('/mnt/flash/.extensions', extensions_dir)
action = action.replace('/mnt/flash/boot-extensions', boot_extensions)
bootstrap.ztps.set_action_response('test_action', action)
contents = random_string()
bootstrap.ztps.set_file_response(extension, contents)
bootstrap.ztps.set_action_response('test_action_force', action)
contents_force = random_string()
bootstrap.ztps.set_file_response(extension_force, contents_force)
bootstrap.start_test()
try:
ext_filename = '%s/%s' % (extensions_dir, extension)
self.failUnless(os.path.isfile(ext_filename))
self.failUnless([contents] == file_log(ext_filename))
self.failUnless(extension in file_log(boot_extensions))
ext_filename_force = '%s/%s' % (extensions_dir, extension_force)
self.failUnless(os.path.isfile(ext_filename_force))
self.failUnless([contents_force] == file_log(ext_filename_force))
self.failUnless('%s force' %
extension_force in file_log(boot_extensions))
self.failUnless(bootstrap.success())
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
shutil.rmtree(extensions_dir)
remove_file(boot_extensions)
bootstrap.end_test()
def test_success(self):
bootstrap = Bootstrap(ztps_default_config=True)
extension = random_string()
url = extension
extension_force = random_string()
url_force = 'http://%s/%s' % (bootstrap.server, extension_force)
bootstrap.ztps.set_definition_response(
actions=[{'action' : 'startup_config_action'},
{'action' : 'test_action',
'attributes' : {'url' : url}},
{'action' : 'test_action_force',
'attributes' :
{'url' : url_force,
'force' : True}}
])
bootstrap.ztps.set_action_response(
'startup_config_action', startup_config_action())
extensions_dir = '/tmp/extensions'
boot_extensions = '/tmp/boot-extensions'
action = get_action('install_extension')
action = action.replace('/mnt/flash/.extensions',
extensions_dir)
action = action.replace('/mnt/flash/boot-extensions',
boot_extensions)
bootstrap.ztps.set_action_response('test_action',
action)
contents = random_string()
bootstrap.ztps.set_file_response(extension, contents)
bootstrap.ztps.set_action_response('test_action_force',
action)
contents_force = random_string()
bootstrap.ztps.set_file_response(extension_force, contents_force)
bootstrap.start_test()
try:
ext_filename = '%s/%s' % (extensions_dir, extension)
self.failUnless(os.path.isfile(ext_filename))
self.failUnless([contents] ==
file_log(ext_filename))
self.failUnless(extension in file_log(boot_extensions))
ext_filename_force = '%s/%s' % (extensions_dir, extension_force)
self.failUnless(os.path.isfile(ext_filename_force))
self.failUnless([contents_force] ==
file_log(ext_filename_force))
self.failUnless('%s force' % extension_force in
file_log(boot_extensions))
self.failUnless(bootstrap.success())
except AssertionError as assertion:
print 'Output: %s' % bootstrap.output
print 'Error: %s' % bootstrap.error
raise_exception(assertion)
finally:
shutil.rmtree(extensions_dir)
remove_file(boot_extensions)
bootstrap.end_test()
