mirror of https://github.com/MISP/misp-dashboard
126 lines
3.8 KiB
Python
Executable File
126 lines
3.8 KiB
Python
Executable File
#!/usr/bin/env python3.5
|
|
import configparser
|
|
import datetime
|
|
import os
|
|
import sys
|
|
import time
|
|
|
|
import redis
|
|
|
|
from helpers import users_helper
|
|
|
|
sys.path.append('..')
|
|
|
|
configfile = 'test_config.cfg'
|
|
cfg = configparser.ConfigParser()
|
|
cfg.read(configfile)
|
|
|
|
serv_redis_db = redis.StrictRedis(
|
|
host='localhost',
|
|
port=6260,
|
|
db=1)
|
|
|
|
users_helper = users_helper.Users_helper(serv_redis_db, cfg)
|
|
|
|
|
|
def wipeRedis():
|
|
serv_redis_db.flushall()
|
|
|
|
def errorExit():
|
|
sys.exit(1)
|
|
|
|
# return if array are equal even if they are unordered
|
|
def checkArrayEquality(arr1, arr2):
|
|
temp = arr2[:]
|
|
for i in arr1:
|
|
if i in temp:
|
|
temp.remove(i)
|
|
else:
|
|
return False
|
|
return True
|
|
|
|
|
|
def test():
|
|
flag_error = False
|
|
now = int(time.time())
|
|
today = datetime.datetime.fromtimestamp(now)
|
|
twoDayAgo = today - datetime.timedelta(days=2)
|
|
DAY = 60*60*24
|
|
org = 'TEST_ORG'
|
|
org2 = 'TEST_ORG2'
|
|
|
|
# logged in dates
|
|
users_helper.add_user_login(now, org)
|
|
users_helper.add_user_login(now+5, org)
|
|
users_helper.add_user_login(now-DAY*2, org)
|
|
expected_result = [datetime.datetime.fromtimestamp(now-DAY*2), datetime.datetime.fromtimestamp(now+5), datetime.datetime.fromtimestamp(now)]
|
|
rep = users_helper.getDates(org)
|
|
|
|
if not checkArrayEquality(rep, expected_result):
|
|
print('getDates result not matching for all dates')
|
|
flag_error = True
|
|
expected_result = [datetime.datetime.fromtimestamp(now+5), datetime.datetime.fromtimestamp(now)]
|
|
rep = users_helper.getDates(org, datetime.datetime.now())
|
|
if not checkArrayEquality(rep, expected_result):
|
|
print('getDates result not matching for query 1')
|
|
flag_error = True
|
|
expected_result = []
|
|
rep = users_helper.getDates(org, datetime.datetime.now()-datetime.timedelta(days=7))
|
|
if not checkArrayEquality(rep, expected_result):
|
|
print('getDates result not matching for query 2')
|
|
flag_error = True
|
|
|
|
# all logged orgs
|
|
users_helper.add_user_login(now, org2)
|
|
expected_result = [datetime.datetime.fromtimestamp(now+5), datetime.datetime.fromtimestamp(now), datetime.datetime.fromtimestamp(now)]
|
|
rep = users_helper.getUserLogins(datetime.datetime.now())
|
|
if not checkArrayEquality(rep, expected_result):
|
|
print('getUserLogins result not matching')
|
|
flag_error = True
|
|
|
|
# all logged in org
|
|
expected_result = [org, org2]
|
|
rep = users_helper.getAllLoggedInOrgs(datetime.datetime.fromtimestamp(now+5), prev_days=7)
|
|
if not checkArrayEquality(rep, expected_result):
|
|
print('getAllLoggedInOrgs result not matching')
|
|
flag_error = True
|
|
|
|
# punchcard
|
|
expected_result = [ [0 for x in range(24)] for y in range(7)]
|
|
# set correct values
|
|
day = today.weekday()
|
|
hour = today.hour
|
|
expected_result[day][hour] = 3
|
|
day = twoDayAgo.weekday()
|
|
hour = twoDayAgo.hour
|
|
expected_result[day][hour] = 1
|
|
# swap: punchcard day starts on sunday
|
|
expected_result = [expected_result[6]]+expected_result[:6]
|
|
rep = users_helper.getUserLoginsForPunchCard(datetime.datetime.fromtimestamp(now), org=None, prev_days=6)
|
|
if not checkArrayEquality(rep, expected_result):
|
|
print('getUserLoginsForPunchCard result not matching')
|
|
flag_error = True
|
|
|
|
# overtime
|
|
rep = users_helper.getUserLoginsAndContribOvertime(datetime.datetime.fromtimestamp(now), org=None, prev_days=6)
|
|
t1 = all([tab[1]==0 for tab in rep['contrib']]) # no contribution
|
|
t2 = [True for tab in rep['login'] if tab[1] == 3]
|
|
t2 = t2[0] and len(t2)==1 # one login at 3, others at 0
|
|
if not (t1 and t2):
|
|
print('getUserLoginsAndContribOvertime result not matching')
|
|
flag_error = True
|
|
|
|
|
|
|
|
|
|
|
|
return flag_error
|
|
|
|
wipeRedis()
|
|
if test():
|
|
wipeRedis()
|
|
errorExit()
|
|
else:
|
|
wipeRedis()
|
|
print('Users tests succeeded')
|