style: 项目重构
1.项目改名为kilostar(千星) 2.后端部分进行大规模重构 3.node功能进行大规模重新设计
This commit is contained in:
@@ -28,9 +28,9 @@ def mock_import(name, globals=None, locals=None, fromlist=(), level=0):
|
||||
|
||||
builtins.__import__ = mock_import
|
||||
for mod in list(sys.modules.keys()):
|
||||
if "pretor.core.workflow_running_engine.workflow_runner" in mod or "ray" in mod:
|
||||
if "kilostar.core.workflow_running_engine.workflow_runner" in mod or "ray" in mod:
|
||||
del sys.modules[mod]
|
||||
from pretor.core.workflow_running_engine.workflow_runner import ( # noqa: E402
|
||||
from kilostar.core.workflow_running_engine.workflow_runner import ( # noqa: E402
|
||||
WorkflowEngine,
|
||||
WorkflowRunningEngine,
|
||||
)
|
||||
@@ -40,7 +40,7 @@ builtins.__import__ = real_import
|
||||
|
||||
@pytest.fixture
|
||||
def mock_ray():
|
||||
with patch("pretor.core.workflow_running_engine.workflow_runner.ray") as mock_ray:
|
||||
with patch("kilostar.core.workflow_running_engine.workflow_runner.ray") as mock_ray:
|
||||
mock_ray.get = lambda x: x
|
||||
yield mock_ray
|
||||
|
||||
@@ -52,14 +52,14 @@ def test_workflow_engine_init():
|
||||
assert engine.workflow == mock_wf
|
||||
assert engine.consciousness_node == "conscious"
|
||||
assert engine.control_node == "control"
|
||||
assert engine.supervisory_node == "supervisor"
|
||||
assert engine.regulatory_node == "supervisor"
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_workflow_engine_run():
|
||||
from pretor.core.workflow.workflow import PretorWorkflow, WorkStep, WorkflowStatus
|
||||
from kilostar.core.workflow_running_engine.workflow import kilostarWorkflow, WorkStep, WorkflowStatus
|
||||
|
||||
mock_wf = MagicMock(spec=PretorWorkflow)
|
||||
mock_wf = MagicMock(spec=kilostarWorkflow)
|
||||
|
||||
step1 = MagicMock(spec=WorkStep)
|
||||
step1.step = 1
|
||||
@@ -97,7 +97,7 @@ async def test_workflow_engine_run():
|
||||
engine = WorkflowEngine(mock_wf, mock_conscious, mock_control, mock_supervisor)
|
||||
|
||||
with patch(
|
||||
"pretor.core.workflow_running_engine.workflow_runner.ray"
|
||||
"kilostar.core.workflow_running_engine.workflow_runner.ray"
|
||||
) as mock_ray_patch:
|
||||
mock_gsm = MagicMock()
|
||||
mock_ray_patch.get_actor.return_value = mock_gsm
|
||||
@@ -111,7 +111,7 @@ def test_workflow_running_engine_init():
|
||||
engine = WorkflowRunningEngine("conscious", "control", "supervisor")
|
||||
assert engine.consciousness_node == "conscious"
|
||||
assert engine.control_node == "control"
|
||||
assert engine.supervisory_node == "supervisor"
|
||||
assert engine.regulatory_node == "supervisor"
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
@@ -128,8 +128,8 @@ async def test_workflow_running_engine_submit():
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_workflow_running_engine_runner():
|
||||
from pretor.api.platform.event import PretorEvent
|
||||
from pretor.core.individual.consciousness_node.template import ForWorkflowEngine
|
||||
from kilostar.api.platform.event import kilostarEvent
|
||||
from kilostar.core.individual.consciousness_node.template import ForWorkflowEngine
|
||||
|
||||
mock_consciousness = MagicMock()
|
||||
mock_wf = MagicMock()
|
||||
@@ -143,7 +143,7 @@ async def test_workflow_running_engine_runner():
|
||||
engine = WorkflowRunningEngine(mock_consciousness, "control", "supervisor")
|
||||
engine.workflow_queue = asyncio.Queue()
|
||||
|
||||
mock_event = PretorEvent(
|
||||
mock_event = kilostarEvent(
|
||||
platform="test_platform",
|
||||
user_id="test_user",
|
||||
user_name="test_user",
|
||||
@@ -167,11 +167,11 @@ async def test_workflow_running_engine_runner():
|
||||
|
||||
with (
|
||||
patch(
|
||||
"pretor.core.workflow_running_engine.workflow_runner.WorkflowEngine"
|
||||
"kilostar.core.workflow_running_engine.workflow_runner.WorkflowEngine"
|
||||
) as mock_wf_engine_cls,
|
||||
patch("builtins.open", new_callable=MagicMock) as mock_open,
|
||||
patch(
|
||||
"pretor.core.workflow_running_engine.workflow_runner.ray_actor_hook"
|
||||
"kilostar.core.workflow_running_engine.workflow_runner.ray_actor_hook"
|
||||
) as mock_hook,
|
||||
):
|
||||
# Instead of patching hook, we inject it directly
|
||||
|
||||
Reference in New Issue
Block a user