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)