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_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_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(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)
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_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_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_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_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(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(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_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()
