def test_LineInfo_not_allowed(self): res = authparse.LineInfo(test_lines['not_allowed']) self.assertEqual(( res.line_type, res.key_, res.value, ), ('not_allowed', 'user', "root"))
def test_master_dict(self): master = authparse.IpDict() for key in test_lines: ip = authparse.get_ip(test_lines[key]) master.add_data(ip, authparse.LineInfo(test_lines[key])) keys = set([key for key in master.data]) self.assertEqual(keys, ip_set)
def test_LineInfo_pub_key(self): res = authparse.LineInfo(test_lines['pub_key']) self.assertEqual(( res.line_type, res.key_, res.value, ), ('pub_key', 'user', "alex"))
def test_LineInfo_invalid_user(self): res = authparse.LineInfo(test_lines['invalid_user']) self.assertEqual(( res.line_type, res.key_, res.value, ), ('invalid_user', 'user', "postgres"))
def test_add2ip_dict(self): ip_set = set() ip = "10.0.0.10" ip_set.add(ip) master = authparse.IpDict() master.add_data(ip, authparse.LineInfo(test_lines['invalid_user'])) self.assertEqual(master.data[ip].data['invalid_user'], ['postgres']) master.add_data(ip, authparse.LineInfo(test_line1)) self.assertEqual(master.data[ip].data['invalid_user'], [ 'postgres', 'apache', ]) master.add_data(ip, authparse.LineInfo(test_lines['no_id'])) self.assertEqual(master.data[ip].data['no_id'], 1) ip = "10.1.0.10" ip_set.add(ip) master.add_data(ip, authparse.LineInfo(test_lines['no_id'])) self.assertEqual(master.data[ip].data['no_id'], 1) self.assertEqual(set([key for key in master.data]), ip_set)
def test_add2ip_info1(self): ip_info = authparse.IpInfo() line_info0 = authparse.LineInfo(test_lines['invalid_user']) line_info1 = authparse.LineInfo(test_line1) line_info2 = authparse.LineInfo(test_line2) line_info3 = authparse.LineInfo(test_line3) line_info4 = authparse.LineInfo(test_line4) ip_info.add_entry(line_info0) ip_info.add_entry(line_info1) ip_info.add_entry(line_info2) ip_info.add_entry(line_info3) ip_info.add_entry(line_info4) self.assertEqual(ip_info.data['invalid_user'], [ "postgres", "apache", "india", "romanian", "brazil", ])
def setUp(self): self.args = { '--report': 1, '--demographics': False, '--frequency': False, } self.master = authparse.IpDict() for ip in w_ips: for k in test_lines.keys(): line_info = authparse.LineInfo(test_lines['pub_key']) self.master.add_data(ip, line_info)
def test_LineInfo_disconnect(self): res = authparse.LineInfo(test_lines['disconnect']) self.assertEqual(( res.line_type, res.key_, res.value, ), ( 'disconnect', None, None, ))
def test_LineInfo_closed(self): res = authparse.LineInfo(test_lines['closed']) self.assertEqual(( res.line_type, res.key_, res.value, ), ( 'closed', None, None, ))
def test_LineInfo_break_in(self): res = authparse.LineInfo(test_lines['break_in']) self.assertEqual(( res.line_type, res.key_, res.value, ), ( 'break_in', None, None, ))
def test_0(self): master = authparse.IpDict() ip = '10.0.0.1' line_info = authparse.LineInfo(test_lines['pub_key']) master.add_data(ip, line_info) show = master.show(self.args) # print('\n###\n{}\n###\n'.format(show)) # print('\n###\n{}\n###\n' # .format(master.data[ip].show(self.args, ip))) self.assertEqual( show, 'Remaining IP addresses (which are suspect:)\n10.0.0.1\n Appearances: 1.' )
def test_0(self): shown = ( '10.0.0.1', '10.0.0.1\n Appearances: 1.', "10.0.0.1\n Appearances: 1.\n pub_key: 1. =>['alex']", ) for n in range(3): args = { '--report': n, '--demographics': False, '--frequency': True, } line_info = authparse.LineInfo(test_lines['pub_key']) info = authparse.IpInfo() info.add_entry(line_info) s = info.show(args, '10.0.0.1') self.assertEqual(s, shown[n])