def test_help_contents(self):
shell = ExecShell(MockVuln(), None, None)
_help = shell.help(None)
self.assertIn('execute', _help)
self.assertIn('upload', _help)
def __init__(self, vuln, uri_opener, worker_pool, exploit_dc):
'''
Create the obj
'''
PortScanShell.__init__(self, vuln, uri_opener, worker_pool)
ExecShell.__init__(self, vuln, uri_opener, worker_pool)
self._exploit_dc = exploit_dc
def __init__(self, vuln, uri_opener, worker_pool, exploit_dc):
'''
Create the obj
'''
PortScanShell.__init__(self, vuln, uri_opener, worker_pool)
ExecShell.__init__(self, vuln, uri_opener, worker_pool)
self._exploit_dc = exploit_dc
def test_help_format(self):
shell = ExecShell(MockVuln(), None, None)
_help = shell.help(None)
self.assertFalse(_help.startswith(' '))
self.assertIn(' help', _help)
# Note that I add an extra space
self.assertNotIn(' help', _help)