Viewing file:  test_system_info.py (1.01 KB) -rw-r--r--
Select action/file-type:
 (+) |  (+) |  (+) | Code (+) | Session (+) |  (+) | SDB (+) |  (+) |  (+) |  (+) |  (+) |  (+) |
from tests.base_test import BaseTest
from testfixtures import log_capture
from weevely.core.weexceptions import ArgparseError
from weevely.core import modules
from weevely.core.sessions import SessionURL
import os
class SystemInfo(BaseTest):
def setUp(self):
session = SessionURL(self.url, self.password, volatile = True)
modules.load_modules(session)
self.run_argv = modules.loaded['system_info'].run_argv
@log_capture()
def test_commands(self, log_captured):
# Get all infos, returns a dict
vectors_names = [v.name for v in modules.loaded['system_info'].vectors ]
self.assertEqual(set(self.run_argv([ '-extended' ]).keys()), set(vectors_names));
# Get just one info, returns a string
self.assertEqual(
os.path.split(self.run_argv(["-info", "script"]))[1],
os.path.split(self.path)[1]
);
# Pass unexistant info
self.assertRaises(ArgparseError, self.run_argv, ["-info", "BOGUS"]);
|