mirror of https://github.com/OWASP/Nettacker.git
46 lines
1.5 KiB
Python
46 lines
1.5 KiB
Python
from unittest.mock import patch, MagicMock
|
|
|
|
import pytest
|
|
from sqlalchemy import create_engine, inspect
|
|
|
|
from nettacker.config import Config
|
|
from nettacker.database.models import Base
|
|
from nettacker.database.sqlite import sqlite_create_tables
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_config():
|
|
Config.db = MagicMock()
|
|
yield Config.db
|
|
|
|
|
|
@patch("nettacker.database.sqlite.create_engine")
|
|
def test_sqlite_create_tables(mock_create_engine, mock_config):
|
|
mock_config.as_dict.return_value = {"name": "/path/to/test.db"}
|
|
|
|
mock_engine = MagicMock()
|
|
mock_create_engine.return_value = mock_engine
|
|
|
|
with patch.object(Base.metadata, "create_all") as mock_create_all:
|
|
sqlite_create_tables()
|
|
|
|
mock_create_engine.assert_called_once_with(
|
|
"sqlite:////path/to/test.db", connect_args={"check_same_thread": False}
|
|
)
|
|
mock_create_all.assert_called_once_with(mock_engine)
|
|
|
|
|
|
def test_sqlite_create_tables_integration(mock_config):
|
|
engine = create_engine("sqlite:///:memory:")
|
|
mock_config.as_dict.return_value = {"name": ":memory:"}
|
|
|
|
with patch("nettacker.database.sqlite.create_engine", return_value=engine):
|
|
sqlite_create_tables()
|
|
|
|
inspector = inspect(engine)
|
|
tables = inspector.get_table_names()
|
|
|
|
assert "reports" in tables, "Reports table was not created"
|
|
assert "temp_events" in tables, "Temp events table was not created"
|
|
assert "scan_events" in tables, "Scan events table was not created"
|