def test_remove_plugins():
""" Remove plugins with valid dirs, invalid dirs """
path_dirs = test_env.PathDirs()
invalid_dirs = test_env.PathDirs(base_dir="/tmp/")
plugin_parser.remove_plugins(
path_dirs, "https://github.com/CyberReboot/vent-plugins.git")
plugin_parser.remove_plugins(invalid_dirs, "vent-plugins")
plugin_parser.remove_plugins(
path_dirs, "https://github.com/Joecakes4u/test_template_file_ignore")
#removing a git repo that isn't installed
plugin_parser.remove_plugins(
path_dirs, "https://github.com/Joecakes4u/test_template_file_ignore")
def test_add_plugins():
""" Test with valid dirs, invalid dirs, empty dirs, non-extistent plugins, duplicate plugins """
path_dirs = test_env.PathDirs()
invalid_dirs = test_env.PathDirs(base_dir="/tmp/")
invalid2_dirs = test_env.PathDirs(plugin_repos="core/")
plugin_parser.add_plugins(
path_dirs, "https://github.com/CyberReboot/vent-plugins.git")
plugin_parser.add_plugins(
invalid_dirs, "https://github.com/CyberReboot/vent-plugins.git")
plugin_parser.add_plugins(path_dirs, "test")
plugin_parser.add_plugins(path_dirs, "")
plugin_parser.add_plugins(invalid2_dirs,
"https://github.com/template-change")
plugin_parser.add_plugins(
path_dirs, "https://github.com/Joecakes4u/test_template_file_ignore")
def test_running_remove_plugin():
""" testing running the menu and removing a plugin """
path_dirs = test_env.PathDirs()
cmd = "python2.7 vent/menu_launcher.py " + path_dirs.base_dir + " " + path_dirs.info_dir + " " + path_dirs.data_dir
child1 = pexpect.spawn(cmd)
### Plugins Menu ###
# go to plugins menu
child1.sendline('2')
child1.expect('Return to Vent menu')
# remove plugin menu
child1.sendline('2')
child1.expect('Return to Plugins menu')
# remove plugin
child1.sendline('1')
child1.expect('Press any key to continue...')
# press a key
# go to plugins menu
child1.send('q')
child1.expect('Return to Vent menu')
# go to main menu
child1.sendline('5')
child1.expect('Exit')
child1.sendline('8')
child1.read()
child1.close()
def test_running_configure_template():
""" testing running the menu and configuring a template file """
path_dirs = test_env.PathDirs()
cmd = "python2.7 vent/menu_launcher.py " + path_dirs.base_dir + " " + path_dirs.info_dir + " " + path_dirs.data_dir
child1 = pexpect.spawn(cmd)
# go to modes menu
child1.sendline('1')
child1.expect('Return to Vent menu')
# go to configure menu
child1.sendline('5')
child1.expect('Return to Mode menu')
# configure a template
#child1.sendline('1')
#child1.expect('Suplemon')
# quit out of editing the template
#child1.sendcontrol('q')
#child1.expect('Return to Mode menu')
# return to modes menu
child1.sendline('6')
child1.expect('Return to Vent menu')
# go to main menu
child1.sendline('6')
child1.expect('Exit')
child1.sendline('8')
child1.read()
child1.close()
def test_mimetypes_flag():
""" tests get_namespaces using -m flag """
path_dirs = test_env.PathDirs()
parser = get_namespaces.set_parser()
get_namespaces.parse_args(parser.parse_args(['-m']), parser)
get_namespaces.parse_args(
parser.parse_args(['-m', '-t', path_dirs.template_dir]), parser)
def test_get_installed_plugin_repos():
""" Test get_installed_plugin_repos function with valid and invalid directories """
path_dirs = test_env.PathDirs()
invalid_dirs = test_env.PathDirs(base_dir="/tmp/")
menu_launcher.get_installed_plugin_repos(path_dirs, "INFO", "remove")
menu_launcher.get_installed_plugin_repos(path_dirs, "INFO", "update")
menu_launcher.get_installed_plugin_repos(path_dirs, "INFO", "")
menu_launcher.get_installed_plugin_repos(invalid_dirs, "INFO", "remove")
menu_launcher.get_installed_plugin_repos(invalid_dirs, "INFO", "update")
menu_launcher.get_installed_plugin_repos(invalid_dirs, "INFO", "")
# Test with installed plugins
url = "https://github.com/CyberReboot/vent-plugins.git"
env = test_env.TestEnv()
env.add_plugin(path_dirs, url)
menu_launcher.get_installed_plugin_repos(path_dirs, "INFO", "remove")
menu_launcher.get_installed_plugin_repos(path_dirs, "INFO", "update")
menu_launcher.get_installed_plugin_repos(path_dirs, "INFO", "")
env.remove_plugin(path_dirs, url)
def test_private_repos():
""" Tests the entrypoint of plugin_parser with private repo args """
path_dirs = test_env.PathDirs()
os.system(
"python2.7 " + path_dirs.base_dir +
"plugin_parser.py add_plugins https://github.com/CyberReboot/vent-plugins.git user pass private "
+ path_dirs.base_dir + " " + path_dirs.data_dir)
os.system(
"python2.7 " + path_dirs.base_dir +
"plugin_parser.py update_plugins https://github.com/CyberReboot/vent-plugins.git user pass private "
+ path_dirs.base_dir + " " + path_dirs.data_dir)
def test_update_images():
""" Test update_images function """
path_dirs = test_env.PathDirs()
plugin_parser.update_images(path_dirs)
# Add then Remove plugin & call update_images
plugin_parser.add_plugins(
path_dirs, "https://github.com/CyberReboot/vent-plugins.git")
plugin_parser.update_images(path_dirs)
plugin_parser.remove_plugins(
path_dirs, "https://github.com/CyberReboot/vent-plugins.git")
def test_visualization_endpoints():
""" testing running the menu and viewing visualization endpoints """
path_dirs = test_env.PathDirs()
cmd = "python2.7 vent/menu_launcher.py " + path_dirs.base_dir + " " + path_dirs.info_dir + " " + path_dirs.data_dir
child1 = pexpect.spawn(cmd)
# go to visualization endoints menu
child1.sendline('7')
child1.expect('Return to Vent menu')
# go to main menu
child1.sendline('1')
child1.expect('Exit')
child1.sendline('8')
child1.read()
child1.close()
def test_template_queue():
""" Tests simulation of new/modified template """
path_dirs = test_env.PathDirs()
os.environ['HOSTNAME'] = "test"
os.system(
'docker run -d alpine:latest /bin/sh -c "echo core hello world;"')
os.system(
'docker run --name core-template-queue1 -d alpine:latest /bin/sh -c "while true; do echo core hello world; sleep 1; done"'
)
os.system(
'docker run --name active-template-queue1 -d alpine:latest /bin/sh -c "while true; do echo core hello world; sleep 1; done"'
)
os.system(
'docker run --name passive-template-queue1 -d alpine:latest /bin/sh -c "while true; do echo core hello world; sleep 1; done"'
)
os.system(
'docker run --name visualization-template-queue1 -d alpine:latest /bin/sh -c "while true; do echo core hello world; sleep 1; done"'
)
file_watch.template_queue("/dev/null")
file_watch.template_queue("/modes.template")
file_watch.template_queue("/core.template")
file_watch.template_queue("/collectors.template")
file_watch.template_queue("/visualization.template")
os.environ['HOSTNAME'] = subprocess.check_output(
'docker run --name core-template-queue2 -d alpine:latest /bin/sh -c "while true; do echo core hello world; sleep 1; done"',
shell=True)[:4]
file_watch.template_queue("/core.template")
os.environ['HOSTNAME'] = subprocess.check_output(
'docker run --name active-template-queue2 -d alpine:latest /bin/sh -c "while true; do echo core hello world; sleep 1; done"',
shell=True)[:4]
file_watch.template_queue("/collectors.template")
os.environ['HOSTNAME'] = subprocess.check_output(
'docker run --name passive-template-queue2 -d alpine:latest /bin/sh -c "while true; do echo core hello world; sleep 1; done"',
shell=True)[:4]
file_watch.template_queue("/collectors.template")
os.environ['HOSTNAME'] = subprocess.check_output(
'docker run --name visualization-template-queue2 -d alpine:latest /bin/sh -c "while true; do echo core hello world; sleep 1; done"',
shell=True)[:4]
file_watch.template_queue("/visualization.template")
file_watch.template_queue("/modes.template",
base_dir=path_dirs.base_dir + "/")
file_watch.template_queue("/core.template",
base_dir=path_dirs.base_dir + "/")
file_watch.template_queue("/collectors.template",
base_dir=path_dirs.base_dir + "/")
file_watch.template_queue("/visualization.template",
base_dir=path_dirs.base_dir + "/")
def test_entrypoint():
""" Tests the entrypoint of plugin_parser """
path_dirs = test_env.PathDirs()
plugin_parser.add_plugins(
path_dirs, "https://github.com/CyberReboot/vent-plugins.git")
os.system(
"python2.7 " + path_dirs.base_dir +
"plugin_parser.py update_plugins https://github.com/CyberReboot/vent-plugins.git "
+ path_dirs.base_dir + " " + path_dirs.data_dir)
os.system(
"python2.7 " + path_dirs.base_dir +
"plugin_parser.py remove_plugins https://github.com/CyberReboot/vent-plugins.git "
+ path_dirs.base_dir + " " + path_dirs.data_dir)
os.system("python2.7 " + path_dirs.base_dir + "plugin_parser.py")
os.system("python2.7 " + path_dirs.base_dir +
"plugin_parser.py invalid_type https://foo.git")
def test_file_queue():
""" Tests simulation of new file """
os.system(
'docker run -d alpine:latest /bin/sh -c "echo core hello world;"')
os.system(
'docker run -d alpine:latest /bin/sh -c "echo core hello world;"')
os.system(
'docker run -d alpine:latest /bin/sh -c "echo core hello world;"')
time.sleep(5)
file_watch.file_queue("/tmp")
file_watch.file_queue("/dev/null")
file_watch.file_queue("/dev/null", base_dir=os.getcwd() + "/")
# Test with installed plugins
url = "https://github.com/CyberReboot/vent-plugins.git"
env = test_env.TestEnv()
path_dirs = test_env.PathDirs()
env.add_plugin(path_dirs, url)
file_watch.file_queue("vent_/dev/null", base_dir=os.getcwd() + "/")
env.remove_plugin(path_dirs, url)
def test_running_add_plugin():
""" testing running the menu and adding a plugin """
path_dirs = test_env.PathDirs()
cmd = "python2.7 vent/menu_launcher.py " + path_dirs.base_dir + " " + path_dirs.info_dir + " " + path_dirs.data_dir
child1 = pexpect.spawn(cmd)
### Plugins Menu ###
# go to plugins menu
child1.sendline('2')
child1.expect('Return to Vent menu')
# add plugin
child1.sendline('1')
# send url
child1.sendline("bar.git")
child1.expect('Press any key to continue...')
# press a key
# go to plugins menu
child1.send('q')
child1.expect('Exit')
child1.sendline('8')
child1.read()
child1.close()
cmd = "python2.7 vent/menu_launcher.py " + path_dirs.base_dir + " " + path_dirs.info_dir + " " + path_dirs.data_dir
child1 = pexpect.spawn(cmd)
child1.timeout = 120
### Plugins Menu ###
# go to plugins menu
child1.sendline('2')
child1.expect('Return to Vent menu')
# add plugin
child1.sendline('1')
# send url
child1.sendline("https://github.com/CyberReboot/vent-plugins.git")
child1.expect('Press any key to continue...')
# press a key
# go to plugins menu
child1.send('q')
child1.expect('Exit')
child1.sendline('8')
child1.read()
child1.close()
def test_get_namespace_menu():
"""test get_namespace_menu"""
path_dirs = test_env.PathDirs()
menu_launcher.get_namespace_menu(path_dirs)
def test_update_plugins():
""" Test update_plugins function with valid and invalid directories """
path_dirs = test_env.PathDirs()
invalid_dirs = test_env.PathDirs(base_dir="/tmp/")
menu_launcher.update_plugins(path_dirs)
menu_launcher.update_plugins(invalid_dirs)
def test_build_menu_dict():
""" Test build_menu_dict """
path_dirs = test_env.PathDirs()
menu_launcher.build_menu_dict(path_dirs)
def test_run_plugins():
""" Test get_run_plugins function with valid and invalid directories """
# Prep
os.system("cp modes.backup templates/modes.template")
os.system("cp core.backup templates/core.template")
os.system("rm templates/core.template")
os.system("rm templates/modes.template")
env = test_env.TestEnv()
path_dirs = test_env.PathDirs()
env.initconfigs(path_dirs, False)
invalid_dirs = test_env.PathDirs(base_dir="/tmp/")
menu_launcher.run_plugins(path_dirs, "start")
menu_launcher.run_plugins(invalid_dirs, "start")
### Core Test ###
# Check if modes.template has core as a namespace
config = ConfigParser.RawConfigParser()
config.read(path_dirs.template_dir + 'modes.template')
if config.has_section("plugins"):
config.set("plugins", "core", "all")
with open(path_dirs.template_dir + 'modes.template', 'w') as f:
config.write(f)
menu_launcher.run_plugins(path_dirs, "start")
### Visualization Test ###
# Find modes.template
config = ConfigParser.RawConfigParser()
config.read(path_dirs.template_dir + 'modes.template')
# Check for valid sections/options
config.set("plugins", "vis_test", "all")
with open(path_dirs.template_dir + 'modes.template', 'w') as f:
config.write(f)
# Test with one visualization plugin (filewalk)
vis_test = "/vis_test"
if not os.path.exists(path_dirs.vis_dir + vis_test):
os.system("mkdir " + path_dirs.vis_dir + vis_test)
menu_launcher.run_plugins(path_dirs, "start")
# Cleanup
os.system("rm -rf " + path_dirs.vis_dir + vis_test)
os.system("cp modes.backup templates/modes.template")
os.system("cp core.backup templates/core.template")
### Collectors: Passive/Active Test ###
# Find core.template
os.system("touch " + path_dirs.template_dir + 'core.template')
config = ConfigParser.RawConfigParser()
config.read(path_dirs.template_dir + 'core.template')
# Check for valid sections/options
config.set("local-collection", "passive", "on")
config.set("local-collection", "active", "on")
with open(path_dirs.template_dir + 'core.template', 'w') as f:
config.write(f)
# Test with one passive/active collector.
active = "/active-test-collector"
passive = "/passive-test-collector"
if not os.path.exists(path_dirs.collectors_dir + active):
os.system("mkdir " + path_dirs.collectors_dir + active)
if not os.path.exists(path_dirs.collectors_dir + passive):
os.system("mkdir " + path_dirs.collectors_dir + passive)
menu_launcher.run_plugins(path_dirs, "start")
# Cleanup
#os.system("rm -rf "+path_dirs.collectors_dir+active)
#os.system("rm -rf "+path_dirs.collectors_dir+active)
os.system("cp core.backup templates/core.template")
os.system("cp modes.backup templates/modes.template")
menu_launcher.run_plugins(path_dirs, "start")
def test_running_menu():
""" test running the actual menu """
ESC = '\033' # escape key for returning to previous menu
cmd_invalid_path = "python2.7 vent/menu_launcher.py "
child0 = pexpect.spawn(cmd_invalid_path)
# expect main menu
child0.expect('Exit')
# go to mode
child0.sendline('1')
child0.expect('Return to Vent menu')
# go to main menu
child0.sendline(ESC)
child0.expect('Exit')
# exit
child0.sendline(ESC)
child0.read()
child0.close()
path_dirs = test_env.PathDirs()
cmd = "python2.7 vent/menu_launcher.py " + path_dirs.base_dir + " " + path_dirs.info_dir + " " + path_dirs.data_dir
invalid_url = "https://thisisinvalid-.git"
child = pexpect.spawn(cmd)
child.timeout = 120
# expect main menu
child.expect('Exit')
### Mode Menu ###
# go to mode
child.sendline('1')
child.expect('Return to Vent menu')
# go to start
child.sendline('1')
child.expect('Return to Mode menu')
# return to mode
child.sendline(ESC)
child.expect('Return to Vent menu')
# go to stop
child.sendline('2')
child.expect('Return to Mode menu')
# return to mode
child.sendline(ESC)
child.expect('Return to Vent menu')
# go to clean
child.sendline('3')
child.expect('Return to Mode menu')
# return to mode
child.sendline(ESC)
child.expect('Return to Vent menu')
# go to status
child.sendline('4')
child.expect('Return to Mode menu')
# return to mode
child.sendline(ESC)
child.expect('Return to Vent menu')
# go to configure
child.sendline('5')
child.expect('Return to Mode menu')
# return to mode
child.sendline(ESC)
child.expect('Return to Vent menu')
# return to main menu
child.sendline(ESC)
child.expect('Exit')
### Plugins Menu ###
# go to plugins menu
child.sendline('2')
child.expect('Return to Vent menu')
# add plugin
child.sendline('1')
# send url
child.sendline(invalid_url)
child.expect('Operation complete. Press any key to continue...')
# press a key
# go to plugins menu
child.send('q')
child.expect('Exit')
# go to plugins menu
child.sendline('2')
child.expect('Return to Vent menu')
# go to remove plugin
child.sendline('2')
child.expect('Return to Plugins menu')
# return to Plugins menu
child.sendline(ESC)
child.expect('Return to Vent menu')
# go to installed plugins
child.sendline('3')
child.expect('Return to Plugins menu')
# go to plugins menu
child.sendline(ESC)
child.expect('Return to Vent menu')
# go to update plugins
child.sendline('4')
child.expect('Return to Plugins menu')
# return to plugins menu
child.sendline(ESC)
child.expect('Return to Vent menu')
# go to main menu
child.sendline(ESC)
child.expect('Exit')
### System Info Menu ###
# go to System Info Menu
#child.sendline('3')
#child.expect('Return to Vent menu')
# return to Main Menu
#child.sendline(ESC)
#child.expect('Exit')
### Build Menu ###
# go to Build menu
child.sendline('4')
child.expect('Return to Vent menu')
# Build new plugins and core
child.sendline('1')
child.expect('Operation complete. Press any key to continue...')
# Send confirmation
child.send('q')
child.expect('Return to Vent menu')
# !! TODO - Test Force Rebuild
# return to Main Menu
child.sendline(ESC)
child.expect('Exit')
### System Commands Menu ###
# go to system commands
child.sendline('5')
child.expect('Return to Vent menu')
# choose reboot
child.sendline('4')
child.expect('Do you want to continue')
child.send('n')
# choose shutdown
child.sendline('5')
child.expect('Do you want to continue')
child.send('n')
# go to logs menu
child.sendline('1')
child.expect('Return to System Commands menu')
# # go to containers menu
# child.sendline('1')
# # curses blows up because the length of the menu exceeds the terminal size
# child.expect('error')
# go to namespace menu
#child.sendline('2')
#child.expect('Please select a namespace:')
# return to logs menu
#child.sendline(ESC)
#child.expect('Return to System Commands menu')
# go to files
child.sendline('3')
child.expect('Enter the name of the processed file to lookup logs for:')
# enter invalid file
child.sendline('')
child.expect('END')
# return to logs menu
child.send('q')
child.expect('Please select a group to view logs for:')
# go to all
child.sendline('4')
# return to logs menu
child.send('q')
child.expect('Return to System Commands menu')
# return to System Commands menu
child.sendline(ESC)
child.expect('Return to Vent menu')
# go to Service Stats
child.sendline('2')
child.expect('CONTAINER')
# return to System Commands Menu
child.sendcontrol('c')
child.expect('Return to Vent menu')
# return to Main Menu
child.sendline(ESC)
child.expect('Exit')
### Help ###
# go to Help
child.sendline('6')
child.expect('getting started')
# return to Main Menu
child.send('q')
child.expect('Exit')
# exit
child.sendline(ESC)
child.read()
child.close()
def test_get_container_menu():
"""test get_container_menu"""
path_dirs = test_env.PathDirs()
menu_launcher.get_container_menu(path_dirs)
def test_entrypoint():
""" test the entrypoint of get_namespaces """
path_dirs = test_env.PathDirs()
os.system("python2.7 " + path_dirs.info_dir + "get_namespaces.py -m -t " +
path_dirs.template_dir)
def test_get_plugin_status():
""" Test get_plugin_status function with valid and invalid directories """
path_dirs = test_env.PathDirs()
invalid_dirs = test_env.PathDirs(base_dir="/tmp/")
menu_launcher.get_plugin_status(path_dirs)
menu_launcher.get_plugin_status(invalid_dirs)
