from flask import Flask
from os import environ
import database.database as database
# Connect to Redis if the right env vars are set,
# Otherwise generate and use a test DB
if 'REDIS_HOST' in environ:
from redis import Redis
if 'REDIS_PORT' not in environ:
environ['REDIS_PORT'] = '6379'
db = database.redisDB()
else:
db = database.testDB()
app = Flask(__name__)
from urlinfo import urlinfo
def test_testDB_get_threat_id_safe_url():
testDB = database.testDB()
assert testDB.get_threat_id('www.risky.biz:443', 'docs/notspyware.doc') is None
def test_testDB_get_threat_id_unsafe_url_root_path():
testDB = database.testDB()
assert testDB.get_threat_id('www.darknetdiaries.com:443', '') == 1
def test_testDB_get_threat_id_unsafe_url():
testDB = database.testDB()
assert testDB.get_threat_id('www.risky.biz:443', 'virus/funnycats.exe') == 0
def test_testDB_get_threat_details():
testDB = database.testDB()
expected = dict()
expected['name'] = 'Trojan.Win32.FunnyCats'
assert expected == testDB.get_threat_details('#0')
def test_testDB_is_a_database():
testDB = database.testDB()
assert isinstance(testDB, database.Database)