def test_idrac_scan_success(mock_args):
responses.reset()
responses.add(**MockResponses.idrac_fp)
responses.add(**MockResponses.idrac_auth)
reset_handlers()
se = core.main()
assert se.found_q.qsize() == 1
def test_idrac_scan_success(mock_args):
responses.reset()
responses.add(**MockResponses.idrac_fp)
responses.add(**MockResponses.idrac_auth)
reset_handlers()
se = core.main()
assert se.found_q.qsize() == 1
def test_tomcat_fingerprint(mock_args):
responses.add(**MockResponses.tomcat_fp)
reset_handlers()
se = core.main()
print(("Scanners:",se.scanners.qsize()))
assert se.scanners.qsize() == 34
core.remove_queues()
def test_tomcat_fingerprint(mock_args):
responses.add(**MockResponses.tomcat_fp)
reset_handlers()
se = core.main()
print(("Scanners:",se.scanners.qsize()))
assert se.scanners.qsize() == 34
core.remove_queues()
def test_targets_scan_success(mock_args):
responses.reset()
responses.add(**MockResponses.idrac_fp)
responses.add(**MockResponses.idrac_auth)
with open(targets_args['target'], 'w') as fout:
fout.write('127.0.0.1' + '\n')
reset_handlers()
se = core.main()
assert se.found_q.qsize() == 1
def test_targets_scan_success(mock_args):
responses.reset()
responses.add(**MockResponses.idrac_fp)
responses.add(**MockResponses.idrac_auth)
with open(targets_args['target'], 'w') as fout:
fout.write('127.0.0.1' + '\n')
reset_handlers()
se = core.main()
assert se.found_q.qsize() == 1
def memcached(mock_args):
reset_handlers()
se = core.main()
try:
assert se.found_q.qsize() == 1
except Exception as e:
# Raise an assertion error if we're in Travis CI and fail
if os.environ.get('TRAVIS', None):
raise e
# Warn if we're not Travis CI
else:
logger.warning('memcached failed')
def memcached(mock_args):
reset_handlers()
se = core.main()
try:
assert se.found_q.qsize() == 1
except Exception as e:
# Raise an assertion error if we're in Travis CI and fail
if os.environ.get('TRAVIS', None):
raise e
# Warn if we're not Travis CI
else:
logger.warning('memcached failed')
def test_json_output(mock_args):
responses.add(**MockResponses.jboss_fp)
responses.add(**MockResponses.jboss_auth)
reset_handlers()
se = core.main()
assert se.found_q.qsize() == 1
assert os.path.isfile(json_args['output'])
i = 0
with open(json_args['output'], 'r') as json_file:
j = json.loads(json_file.read())
assert j["results"][0]['name'] == 'JBoss AS 6 Alt'
assert j['results'][0]['username'] == 'admin'
assert j['results'][0]['password'] == 'admin'
assert j['results'][0]['target'] == 'http://127.0.0.1:8080/admin-console/login.seam'
def test_json_output(mock_args):
responses.add(**MockResponses.jboss_fp)
responses.add(**MockResponses.jboss_auth)
reset_handlers()
se = core.main()
assert se.found_q.qsize() == 1
assert os.path.isfile(json_args['output'])
i = 0
with open(json_args['output'], 'r') as json_file:
j = json.loads(json_file.read())
assert j["results"][0]['name'] == 'JBoss AS 6 Alt'
assert j['results'][0]['username'] == 'admin'
assert j['results'][0]['password'] == 'admin'
assert j['results'][0]['target'] == 'http://127.0.0.1:8080/admin-console/login.seam'
def test_csv_output(mock_args):
responses.add(**MockResponses.jboss_fp)
responses.add(**MockResponses.jboss_auth)
reset_handlers()
se = core.main()
assert se.found_q.qsize() == 1
assert os.path.isfile(csv_args['output'])
i = 0
with open(csv_args['output'], 'r') as csvfile:
reader = csv.reader(csvfile)
for line in reader:
if i == 1:
assert line[0] == 'JBoss AS 6 Alt'
assert line[1] == 'admin'
assert line[2] == 'admin'
assert line[3] == 'http://127.0.0.1:8080/admin-console/login.seam'
i += 1
assert os.path.isfile(csv_args['log'])
def test_csv_output(mock_args):
responses.add(**MockResponses.jboss_fp)
responses.add(**MockResponses.jboss_auth)
reset_handlers()
se = core.main()
print(se.found_q.qsize())
assert se.found_q.qsize() == 1
assert os.path.isfile(csv_args['output'])
i = 0
with open(csv_args['output'], 'r') as csvfile:
reader = csv.reader(csvfile)
for line in reader:
if i == 1:
assert line[0] == 'JBoss AS 6 Alt'
assert line[1] == 'admin'
assert line[2] == 'admin'
assert line[3] == 'http://127.0.0.1:8080/admin-console/login.seam'
i += 1
assert os.path.isfile(csv_args['log'])
def test_jboss_scan_success_subnet(mock_args):
responses.add(**MockResponses.jboss_fp)
responses.add(**MockResponses.jboss_auth)
reset_handlers()
se = core.main()
assert se.found_q.qsize() == 1
def test_es_scan_success(mock_args):
responses.reset()
responses.add(**MockResponses.elasticsearch)
reset_handlers()
se = core.main()
assert se.found_q.qsize() == 1
def test_dryrun(mock_args):
reset_handlers()
se = core.main()
assert se.found_q.qsize() == 0
def test_jboss_csrf_fail(mock_args):
responses.add(**MockResponses.jboss_fp_no_csrf)
reset_handlers()
se = core.main()
assert se.found_q.qsize() == 0
def test_jboss_csrf_fail(mock_args):
responses.add(**MockResponses.jboss_fp_no_csrf)
reset_handlers()
se = core.main()
assert se.found_q.qsize() == 0
def test_jboss_scan_success_subnet(mock_args):
responses.add(**MockResponses.jboss_fp)
responses.add(**MockResponses.jboss_auth)
reset_handlers()
se = core.main()
assert se.found_q.qsize() == 1
def test_tomcat_invalid_creds(mock_args):
responses.add(**MockResponses.tomcat_fp)
reset_handlers()
se = core.main()
assert se.found_q.qsize() == 0
def test_tomcat_invalid_creds(mock_args):
responses.add(**MockResponses.tomcat_fp)
reset_handlers()
se = core.main()
assert se.found_q.qsize() == 0
def test_redis(mock_args):
reset_handlers()
se = core.main()
assert se.found_q.qsize() == 1
def test_snmp_proto(mock_args):
reset_handlers()
se = core.main()
assert se.found_q.qsize() == 2
#!/usr/bin/env python
from changeme import core
if __name__ == '__main__':
core.main()
def test_snmp_proto(mock_args):
reset_handlers()
se = core.main()
assert se.found_q.qsize() == 2
def test_es_scan_success(mock_args):
responses.reset()
responses.add(**MockResponses.elasticsearch)
reset_handlers()
se = core.main()
assert se.found_q.qsize() == 1
