def test_setup_logging_file(self):
fh = False
logger = changeme.setup_logging(False, False, "/tmp/foo.log")
for i in logger.handlers:
if isinstance(i, logging.FileHandler):
fh = True
assert fh
def test_setup_logging_file(self):
fh = False
logger = changeme.setup_logging(False, False, "/tmp/foo.log")
for i in logger.handlers:
if isinstance(i, logging.FileHandler):
fh = True
assert fh
def test_do_scan_get(self):
responses.add(** mock.ipcamera_fp)
responses.add(** mock.ipcamera_auth)
changeme.logger = changeme.setup_logging(True, True, None)
matches = changeme.do_scan(mock.ipcamera_fp['url'], self.creds, self.config)
assert len(matches) == 1
def test_check_basic_auth_tomcat_fail(self):
responses.add(** mock.tomcat_fp)
cred = self.get_cred(self.tomcat_name)
assert cred['name'] == self.tomcat_name
changeme.logger = changeme.setup_logging(False, False, None)
matches = changeme.check_basic_auth(mock.tomcat_fp['url'], cred, False, False)
assert len(matches) == 0
def test_do_scan_get(self):
responses.add(**mock.ipcamera_fp)
responses.add(**mock.ipcamera_auth)
changeme.logger = changeme.setup_logging(True, True, None)
matches = changeme.do_scan(mock.ipcamera_fp['url'], self.creds,
self.config)
assert len(matches) == 1
def test_check_basic_auth_tomcat_fail(self):
responses.add(**mock.tomcat_fp)
cred = self.get_cred(self.tomcat_name)
assert cred['name'] == self.tomcat_name
changeme.logger = changeme.setup_logging(False, False, None)
matches = changeme.check_basic_auth(mock.tomcat_fp['url'], cred, False,
False)
assert len(matches) == 0
def test_check_basic_auth_tomcat_fail(self):
responses.add(**mock.tomcat_fp)
responses.add(**mock.tomcat_fp_alt)
cred = self.get_cred(self.tomcat_name)
assert cred['name'] == self.tomcat_name
changeme.logger = changeme.setup_logging(False, False, None)
s = requests.Session()
matches = changeme.check_basic_auth(mock.tomcat_fp['url'], s, cred,
self.config, False, False)
assert len(matches) == 0
def test_do_scan_idrac(self):
"""
This test will makes sure the regex in
iDRAC success body works
"""
responses.add(** mock.idrac_fp)
responses.add(** mock.idrac_auth)
changeme.logger = changeme.setup_logging(True, True, None)
matches = changeme.do_scan(mock.idrac_fp['url'], self.creds, self.config)
assert len(matches) == 1
assert matches[0]['name'] == self.idrac_name
def test_do_scan_idrac(self):
"""
This test will makes sure the regex in
iDRAC success body works
"""
responses.add(**mock.idrac_fp)
responses.add(**mock.idrac_auth)
changeme.logger = changeme.setup_logging(True, True, None)
matches = changeme.do_scan(mock.idrac_fp['url'], self.creds,
self.config)
assert len(matches) == 1
assert matches[0]['name'] == self.idrac_name
def test_do_scan_get(self):
responses.add(**mock.ipcamera_fp)
responses.add(**mock.ipcamera_auth)
changeme.logger = changeme.setup_logging(True, True, None)
tlist = changeme.build_target_list(('127.0.0.1', ), self.creds, None,
None)
self.config['fingerprint'] = False
matches = changeme.do_scan(tlist['fingerprints'], self.creds,
self.config)
assert isinstance(matches, list)
assert len(matches) == 1
print "matches[0]: ", matches[0]
assert matches[0]['name'] == 'MayGion Camera'
def setup_class(cls):
changeme.logger = changeme.setup_logging(True, True, None)
def test_load_creds_good(self):
changeme.logger = changeme.setup_logging(False, False, None)
changeme.load_creds(None, None)
def test_setup_logging_debug(self):
logger = changeme.setup_logging(False, True, None)
assert logger.isEnabledFor(logging.INFO)
assert logger.isEnabledFor(logging.DEBUG)
def test_setup_logging_verbose(self):
logger = changeme.setup_logging(True, False, None)
assert logger.isEnabledFor(logging.INFO)
def test_setup_logging(self):
logger = changeme.setup_logging(False, False, None)
assert logger.isEnabledFor(logging.WARNING)
def test_load_creds_good(self):
changeme.logger = changeme.setup_logging(False, False, None)
changeme.load_creds(None, None)
def setup_class(cls):
changeme.logger = changeme.setup_logging(True, True, None)
def test_setup_logging(self):
logger = changeme.setup_logging(False, False, None)
assert logger.isEnabledFor(logging.WARNING)
def test_setup_logging_verbose(self):
logger = changeme.setup_logging(True, False, None)
assert logger.isEnabledFor(logging.INFO)
def test_setup_logging_debug(self):
logger = changeme.setup_logging(False, True, None)
assert logger.isEnabledFor(logging.INFO)
assert logger.isEnabledFor(logging.DEBUG)