mirror of https://github.com/MISP/misp-dashboard
76 lines
2.0 KiB
Python
76 lines
2.0 KiB
Python
|
#!/usr/bin/env python3.5
|
||
|
|
import configparser
|
||
|
|
import redis
|
||
|
|
import sys,os
|
||
|
|
import datetime
|
||
|
|
sys.path.append('..')
|
||
|
|
|
||
|
|
configfile = 'test_config.cfg'
|
||
|
|
cfg = configparser.ConfigParser()
|
||
|
|
cfg.read(configfile)
|
||
|
|
|
||
|
|
serv_redis_db = redis.StrictRedis(
|
||
|
|
host='localhost',
|
||
|
|
port=6260,
|
||
|
|
db=1)
|
||
|
|
|
||
|
|
from helpers import geo_helper
|
||
|
|
geo_helper = geo_helper.Geo_helper(serv_redis_db, cfg)
|
||
|
|
|
||
|
|
categ = 'Network Activity'
|
||
|
|
|
||
|
|
def wipeRedis():
|
||
|
|
serv_redis_db.flushall()
|
||
|
|
|
||
|
|
def errorExit():
|
||
|
|
wipeRedis()
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
def test():
|
||
|
|
today = datetime.datetime.now()
|
||
|
|
# IP -> Coord
|
||
|
|
supposed_ip = '8.8.8.8'
|
||
|
|
geo_helper.getCoordFromIpAndPublish(supposed_ip, categ)
|
||
|
|
rep = geo_helper.getTopCoord(today)
|
||
|
|
excpected_result = [['{"lat": 37.751, "lon": -97.822}', 1.0]]
|
||
|
|
if rep != excpected_result:
|
||
|
|
print('ip to coord result not matching')
|
||
|
|
errorExit()
|
||
|
|
|
||
|
|
# gethitmap
|
||
|
|
rep = geo_helper.getHitMap(today)
|
||
|
|
excpected_result = [['US', 1.0]]
|
||
|
|
if rep != excpected_result:
|
||
|
|
print('getHitMap result not matching')
|
||
|
|
errorExit()
|
||
|
|
|
||
|
|
# getCoordsByRadius
|
||
|
|
rep = geo_helper.getCoordsByRadius(today, today, 0.000, 0.000, '1')
|
||
|
|
excpected_result = []
|
||
|
|
if rep != excpected_result:
|
||
|
|
print('getCoordsByRadius result not matching')
|
||
|
|
errorExit()
|
||
|
|
|
||
|
|
rep = geo_helper.getCoordsByRadius(today, today, 37.750, -97.821, '10')
|
||
|
|
excpected_result = [[['{"categ": "Network Activity", "value": "8.8.8.8"}'], [-97.82200008630753, 37.75100012475438]]]
|
||
|
|
if rep != excpected_result:
|
||
|
|
print('getCoordsByRadius result not matching')
|
||
|
|
errorExit()
|
||
|
|
|
||
|
|
wipeRedis()
|
||
|
|
|
||
|
|
|
||
|
|
# Phone -> Coord
|
||
|
|
phoneNumber = '(+352) 247-82000'
|
||
|
|
geo_helper.getCoordFromPhoneAndPublish(phoneNumber, categ)
|
||
|
|
rep = geo_helper.getTopCoord(datetime.datetime.now())[0]
|
||
|
|
excpected_result = ['{"lat": "49.7500", "lon": "6.1667"}', 1.0]
|
||
|
|
if rep != excpected_result:
|
||
|
|
print('Phone to coord result not matching')
|
||
|
|
errorExit()
|
||
|
|
|
||
|
|
|
||
|
|
test()
|
||
|
|
wipeRedis()
|
||
|
|
print('Tests succeeded')
|