def test_vendor_index_search(test_db, test_vuln_data):
# This slow test ensures that every data in the main database is indexed
table = test_db
docs = db.list_all(table)
assert len(docs) == 0
docs = db.store(test_db, test_vuln_data)
assert len(docs) > 0
all_data = db.list_all(table)
assert all_data
tmp_list = []
for d in all_data[:40]:
vendor, _, _ = parse_cpe(d["details"]["cpe_uri"])
tmp_list.append({
"vendor":
vendor,
"name":
d["details"]["package"],
"version":
d["details"]["max_affected_version_including"],
})
res = db.bulk_index_search(tmp_list)
assert len(res)
for r in res:
name_ver = r.split("|")
fullres = db.index_search(name_ver[1], name_ver[2])
assert fullres
def test_search_fast(test_db, test_vuln_data):
table = test_db
docs = db.list_all(table)
assert len(docs) == 0
docs = db.store(test_db, test_vuln_data)
assert len(docs) > 0
all_data = db.list_all(table)
assert all_data
search_list = [
d["details"]["package"] + "|" +
d["details"]["max_affected_version_including"] for d in all_data
]
res = db.pkg_bulk_search(test_db, search_list)
assert len(res) > len(set(search_list))
assert res[0].to_dict()["package_issue"]
def test_gha_search_bulk(test_db, test_gha_data):
table = test_db
docs = db.list_all(table)
assert len(docs) == 0
docs = db.store(test_db, test_gha_data)
assert len(docs) > 0
all_data = db.list_all(table)
assert all_data
tmp_list = [{
"name": d["details"]["package"],
"version": d["details"]["max_affected_version_including"],
} for d in all_data
if d["details"]["max_affected_version_including"] != "*"]
res = db.bulk_index_search(tmp_list)
assert len(res)
def test_search_slow(test_db, test_vuln_data):
table = test_db
docs = db.list_all(table)
assert len(docs) == 0
docs = db.store(test_db, test_vuln_data)
assert len(docs) > 0
all_data = db.list_all(table)
assert all_data
for d in all_data:
res = db.pkg_search(
table,
d["details"]["package"],
d["details"]["max_affected_version_including"],
)
assert len(res)
assert res[0].to_dict()["package_issue"]
def test_gha_vendor_search(test_db, test_gha_data):
table = test_db
docs = db.list_all(table)
assert len(docs) == 0
docs = db.store(test_db, test_gha_data)
assert len(docs) > 0
all_data = db.list_all(table)
assert all_data
for d in all_data:
vendor, _, _ = parse_cpe(d["details"]["cpe_uri"])
version = d["details"]["max_affected_version_including"]
if version and version != "*":
res = db.vendor_pkg_search(
table,
vendor,
d["details"]["package"],
version,
)
assert len(res)
assert res[0].to_dict()["package_issue"]
def test_gha_create(test_db, test_gha_data):
docs = db.store(test_db, test_gha_data)
assert len(docs) > len(test_gha_data)
def store(self, data):
"""Store data in the database
"""
docs = dbLib.store(db, data)
return docs