Skip to content

Commit eaf5fd8

Browse files
committed
also rename agent/use-case in unittest
1 parent 83c92f5 commit eaf5fd8

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

tests/integration_minimal_test.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11

22
from typing import Tuple
3-
from hackingBuddyGPT.usecases.minimal.agent import MinimalLinuxPrivesc, MinimalLinuxPrivescUseCase
4-
from hackingBuddyGPT.usecases.minimal.agent_with_state import MinimalLinuxTemplatedPrivesc, MinimalLinuxTemplatedPrivescUseCase
3+
from hackingBuddyGPT.usecases.examples.agent import ExPrivEscLinux, ExPrivEscLinuxUseCase
4+
from hackingBuddyGPT.usecases.examples.agent_with_state import ExPrivEscLinuxTemplated, ExPrivEscLinuxTemplatedUseCase
55
from hackingBuddyGPT.usecases.privesc.linux import LinuxPrivesc, LinuxPrivescUseCase
66
from hackingBuddyGPT.utils.console.console import Console
77
from hackingBuddyGPT.utils.db_storage.db_storage import DbStorage
@@ -101,8 +101,8 @@ def test_minimal_agent():
101101

102102
log_db.init()
103103

104-
priv_esc = MinimalLinuxPrivescUseCase(
105-
agent = MinimalLinuxPrivesc(
104+
priv_esc = ExPrivEscLinuxUseCase(
105+
agent = ExPrivEscLinux(
106106
conn=conn,
107107
llm=llm
108108
),
@@ -125,8 +125,8 @@ def test_minimal_agent_state():
125125

126126
log_db.init()
127127

128-
priv_esc = MinimalLinuxTemplatedPrivescUseCase(
129-
agent = MinimalLinuxTemplatedPrivesc(
128+
priv_esc = ExPrivEscLinuxTemplatedUseCase(
129+
agent = ExPrivEscLinuxTemplated(
130130
conn=conn,
131131
llm = llm,
132132
),

0 commit comments

Comments
 (0)