evan/overseer
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Slight refactor

Browse Source
This commit is contained in:
Evan Reichard
2021-03-22 12:59:11 -04:00
parent d93cfd183c
commit 5ab87e1c6a
10 changed files with 56 additions and 66 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
tests/overseer/test_database.py
View File

@@ -1,14 +1,12 @@
# import pytest
import ipaddress
from overseer import scan_manager
from overseer.database import DatabaseConnector
# We're not testing this & this will stall tests
scan_manager.shutdown()
DB_TYPE = "MEMORY"
def test_create_scan_target():
db = DatabaseConnector(None, in_memory=True)
db = DatabaseConnector(DB_TYPE)
hostname = db.create_scan_target(hostname="google.com")
ip_address = db.create_scan_target(ip_addr="1.1.1.1")
@@ -19,7 +17,7 @@ def test_create_scan_target():
def test_get_scan_target():
db = DatabaseConnector(None, in_memory=True)
db = DatabaseConnector(DB_TYPE)
created_target = db.create_scan_target(hostname="google.com")
found_target = db.get_scan_target(hostname="google.com")
@@ -31,7 +29,7 @@ def test_get_scan_target():
def test_get_all_scan_targets():
db = DatabaseConnector(None, in_memory=True)
db = DatabaseConnector(DB_TYPE)
for i in range(1, 6):
db.create_scan_target(ip_addr="127.0.0." + str(i))
@@ -47,7 +45,7 @@ def test_get_all_scan_targets():
def test_create_scan_result():
db = DatabaseConnector(None, in_memory=True)
db = DatabaseConnector(DB_TYPE)
scan_target = db.create_scan_target(ip_addr="127.0.0.1")
scan_history = db.create_scan_result("IN_PROGRESS", ip_addr="127.0.0.1")
scan_history_2 = db.create_scan_result("COMPLETE", ip_addr="127.0.0.2")
@@ -62,7 +60,7 @@ def test_create_scan_result():
def test_update_scan_result():
db = DatabaseConnector(None, in_memory=True)
db = DatabaseConnector(DB_TYPE)
scan_history = db.create_scan_result("IN_PROGRESS", ip_addr="127.0.0.1")
updated_scan_history = db.update_scan_result(
scan_history.id, "COMPLETE", ["53 UDP", "53 TCP"]
@@ -75,7 +73,7 @@ def test_update_scan_result():
def test_get_scan_results_by_target():
db = DatabaseConnector(None, in_memory=True)
db = DatabaseConnector(DB_TYPE)
for i in range(1, 6):
db.create_scan_result("IN_PROGRESS", ip_addr="127.0.0.1")