File tree Expand file tree Collapse file tree 2 files changed +4
-6
lines changed Expand file tree Collapse file tree 2 files changed +4
-6
lines changed Original file line number Diff line number Diff line change 10
10
import time
11
11
12
12
# Third-Party Libraries
13
+ from cyhy_db import initialize_db
13
14
import docker
14
15
from motor .core import AgnosticClient
15
16
import pytest
16
17
from rich .logging import RichHandler
17
18
from rich .traceback import install as traceback_install
18
19
19
- # cisagov Libraries
20
- from cyhy_db import initialize_db
21
-
22
20
MONGO_INITDB_ROOT_USERNAME = os .environ .get ("MONGO_INITDB_ROOT_USERNAME" , "mongoadmin" )
23
21
MONGO_INITDB_ROOT_PASSWORD = os .environ .get ("MONGO_INITDB_ROOT_PASSWORD" , "secret" )
24
22
DATABASE_NAME = os .environ .get ("DATABASE_NAME" , "test" )
Original file line number Diff line number Diff line change 1
1
"""Test database connection."""
2
2
3
3
# Third-Party Libraries
4
+ from cyhy_db .models import KEVDoc
4
5
from motor .motor_asyncio import AsyncIOMotorClient
5
6
6
7
# cisagov Libraries
7
- from cyhy_kevsync import DEFAULT_KEV_URL
8
- from cyhy_kevsync .sync import fetch_kev_data , create_kev_doc , process_kev_json
9
- from cyhy_db .models import KEVDoc
8
+ from cyhy_kevsync import DEFAULT_KEV_SCHEMA_URL , DEFAULT_KEV_URL
9
+ from cyhy_kevsync .sync import create_kev_doc , fetch_kev_data , process_kev_json
10
10
11
11
CVE_1 = "CVE-2024-123456"
12
12
VULN_1 = {"cveID" : CVE_1 , "knownRansomwareCampaignUse" : "Known" }
You can’t perform that action at this time.
0 commit comments