mirror of
https://github.com/GH05TCREW/pentestagent.git
synced 2026-04-28 18:50:03 +00:00
refactor: update agent name
This commit is contained in:
@@ -1 +1 @@
|
||||
# GhostCrew Tests
|
||||
# PentestAgent Tests
|
||||
|
||||
@@ -1,4 +1,4 @@
|
||||
"""Test fixtures for GhostCrew tests."""
|
||||
"""Test fixtures for PentestAgent tests."""
|
||||
|
||||
import pytest
|
||||
import asyncio
|
||||
@@ -6,9 +6,9 @@ from pathlib import Path
|
||||
from typing import Generator, AsyncGenerator
|
||||
from unittest.mock import MagicMock, AsyncMock
|
||||
|
||||
from ghostcrew.config import Settings
|
||||
from ghostcrew.agents.state import AgentState, AgentStateManager
|
||||
from ghostcrew.tools import get_all_tools, Tool, ToolSchema
|
||||
from pentestagent.config import Settings
|
||||
from pentestagent.agents.state import AgentState, AgentStateManager
|
||||
from pentestagent.tools import get_all_tools, Tool, ToolSchema
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
|
||||
@@ -3,7 +3,7 @@
|
||||
import pytest
|
||||
from datetime import datetime
|
||||
|
||||
from ghostcrew.agents.state import AgentState, AgentStateManager, StateTransition
|
||||
from pentestagent.agents.state import AgentState, AgentStateManager, StateTransition
|
||||
|
||||
|
||||
class TestAgentState:
|
||||
|
||||
@@ -2,7 +2,7 @@
|
||||
|
||||
import pytest
|
||||
import networkx as nx
|
||||
from ghostcrew.knowledge.graph import ShadowGraph, GraphNode, GraphEdge
|
||||
from pentestagent.knowledge.graph import ShadowGraph, GraphNode, GraphEdge
|
||||
|
||||
class TestShadowGraph:
|
||||
"""Tests for ShadowGraph class."""
|
||||
|
||||
@@ -5,7 +5,7 @@ import numpy as np
|
||||
from pathlib import Path
|
||||
from unittest.mock import patch
|
||||
|
||||
from ghostcrew.knowledge.rag import RAGEngine, Document
|
||||
from pentestagent.knowledge.rag import RAGEngine, Document
|
||||
|
||||
|
||||
class TestDocument:
|
||||
|
||||
@@ -6,7 +6,7 @@ import asyncio
|
||||
from pathlib import Path
|
||||
from unittest.mock import MagicMock, patch
|
||||
|
||||
from ghostcrew.tools.notes import notes, set_notes_file, get_all_notes, _notes
|
||||
from pentestagent.tools.notes import notes, set_notes_file, get_all_notes, _notes
|
||||
|
||||
# We need to reset the global state for tests
|
||||
@pytest.fixture(autouse=True)
|
||||
@@ -18,7 +18,7 @@ def reset_notes_state(tmp_path):
|
||||
|
||||
# Clear the global dictionary (it's imported from the module)
|
||||
# We need to clear the actual dictionary object in the module
|
||||
from ghostcrew.tools.notes import _notes
|
||||
from pentestagent.tools.notes import _notes
|
||||
_notes.clear()
|
||||
|
||||
yield
|
||||
@@ -148,7 +148,7 @@ async def test_legacy_migration(tmp_path):
|
||||
set_notes_file(legacy_file)
|
||||
|
||||
# Trigger load (get_all_notes calls _load_notes_unlocked if empty, but we need to clear first)
|
||||
from ghostcrew.tools.notes import _notes
|
||||
from pentestagent.tools.notes import _notes
|
||||
_notes.clear()
|
||||
|
||||
all_notes = await get_all_notes()
|
||||
|
||||
@@ -2,7 +2,7 @@
|
||||
|
||||
import pytest
|
||||
|
||||
from ghostcrew.tools import (
|
||||
from pentestagent.tools import (
|
||||
Tool, ToolSchema, register_tool, get_all_tools, get_tool,
|
||||
enable_tool, disable_tool, get_tool_names
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user