refactor: update agent name

This commit is contained in:
GH05TCREW
2025-12-19 10:25:58 -07:00
parent fe51aeed6f
commit c9866a407b
91 changed files with 3120 additions and 500 deletions

View File

@@ -1 +1 @@
# GhostCrew Tests
# PentestAgent Tests

View File

@@ -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

View File

@@ -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:

View File

@@ -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."""

View File

@@ -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:

View File

@@ -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()

View File

@@ -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
)