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)