refactor(06-02): update CLI and test imports to kb_workflow.pipeline
- Remove sys.path manipulation from kb_workflow.py - Update CLI imports to use kb_workflow.pipeline prefix - Update all test file imports and patch paths Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
@@ -15,7 +15,7 @@ import pytest
|
||||
from pathlib import Path
|
||||
from datetime import datetime
|
||||
|
||||
from kb_pipeline.schemas import WorkflowState, StepRecord, StepStatus
|
||||
from kb_workflow.pipeline.schemas import WorkflowState, StepRecord, StepStatus
|
||||
|
||||
|
||||
class TestWorkflowStateIsStepCompleted:
|
||||
@@ -157,7 +157,7 @@ class TestAtomicWrite:
|
||||
|
||||
def test_save_creates_state_file(self, tmp_path):
|
||||
"""save creates workflow_state.json in content directory."""
|
||||
from kb_pipeline.state import WorkflowStateManager
|
||||
from kb_workflow.pipeline.state import WorkflowStateManager
|
||||
|
||||
manager = WorkflowStateManager(content_dir=tmp_path)
|
||||
state = WorkflowState(
|
||||
@@ -173,7 +173,7 @@ class TestAtomicWrite:
|
||||
|
||||
def test_save_creates_parent_directories(self, tmp_path):
|
||||
"""save creates parent directories if they don't exist."""
|
||||
from kb_pipeline.state import WorkflowStateManager
|
||||
from kb_workflow.pipeline.state import WorkflowStateManager
|
||||
|
||||
nested_dir = tmp_path / "deeply" / "nested" / "content"
|
||||
manager = WorkflowStateManager(content_dir=nested_dir)
|
||||
@@ -190,7 +190,7 @@ class TestAtomicWrite:
|
||||
|
||||
def test_save_rotates_backup(self, tmp_path):
|
||||
"""save rotates backup (workflow_state.json -> workflow_state.json.bak)."""
|
||||
from kb_pipeline.state import WorkflowStateManager
|
||||
from kb_workflow.pipeline.state import WorkflowStateManager
|
||||
|
||||
manager = WorkflowStateManager(content_dir=tmp_path)
|
||||
state1 = WorkflowState(
|
||||
@@ -228,7 +228,7 @@ class TestAtomicWrite:
|
||||
|
||||
def test_no_temp_files_remain_after_save(self, tmp_path):
|
||||
"""No temp files remain after successful save."""
|
||||
from kb_pipeline.state import WorkflowStateManager
|
||||
from kb_workflow.pipeline.state import WorkflowStateManager
|
||||
|
||||
manager = WorkflowStateManager(content_dir=tmp_path)
|
||||
state = WorkflowState(
|
||||
@@ -248,7 +248,7 @@ class TestBackupFallback:
|
||||
|
||||
def test_load_returns_state_for_valid_file(self, tmp_path):
|
||||
"""load returns WorkflowState for valid file."""
|
||||
from kb_pipeline.state import WorkflowStateManager
|
||||
from kb_workflow.pipeline.state import WorkflowStateManager
|
||||
|
||||
manager = WorkflowStateManager(content_dir=tmp_path)
|
||||
original = WorkflowState(
|
||||
@@ -266,7 +266,7 @@ class TestBackupFallback:
|
||||
|
||||
def test_load_returns_none_for_missing_file(self, tmp_path):
|
||||
"""load returns None when no state file exists."""
|
||||
from kb_pipeline.state import WorkflowStateManager
|
||||
from kb_workflow.pipeline.state import WorkflowStateManager
|
||||
|
||||
manager = WorkflowStateManager(content_dir=tmp_path)
|
||||
loaded = manager.load()
|
||||
@@ -275,7 +275,7 @@ class TestBackupFallback:
|
||||
|
||||
def test_load_falls_back_to_backup_on_corruption(self, tmp_path):
|
||||
"""load falls back to backup when main file is corrupted."""
|
||||
from kb_pipeline.state import WorkflowStateManager
|
||||
from kb_workflow.pipeline.state import WorkflowStateManager
|
||||
|
||||
manager = WorkflowStateManager(content_dir=tmp_path)
|
||||
state_file = tmp_path / "workflow_state.json"
|
||||
@@ -312,7 +312,7 @@ class TestBackupFallback:
|
||||
|
||||
def test_load_from_backup_when_main_missing(self, tmp_path):
|
||||
"""load loads from backup when main file is missing."""
|
||||
from kb_pipeline.state import WorkflowStateManager
|
||||
from kb_workflow.pipeline.state import WorkflowStateManager
|
||||
|
||||
manager = WorkflowStateManager(content_dir=tmp_path)
|
||||
state_file = tmp_path / "workflow_state.json"
|
||||
@@ -347,7 +347,7 @@ class TestStateManagerGetOrCreate:
|
||||
|
||||
def test_get_or_create_returns_existing_state(self, tmp_path):
|
||||
"""get_or_create returns existing state when it exists."""
|
||||
from kb_pipeline.state import WorkflowStateManager
|
||||
from kb_workflow.pipeline.state import WorkflowStateManager
|
||||
|
||||
manager = WorkflowStateManager(content_dir=tmp_path)
|
||||
|
||||
@@ -372,7 +372,7 @@ class TestStateManagerGetOrCreate:
|
||||
|
||||
def test_get_or_create_creates_new_state(self, tmp_path):
|
||||
"""get_or_create creates new state when it doesn't exist."""
|
||||
from kb_pipeline.state import WorkflowStateManager
|
||||
from kb_workflow.pipeline.state import WorkflowStateManager
|
||||
|
||||
manager = WorkflowStateManager(content_dir=tmp_path)
|
||||
|
||||
@@ -390,7 +390,7 @@ class TestStateManagerGetOrCreate:
|
||||
|
||||
def test_get_or_create_is_idempotent(self, tmp_path):
|
||||
"""get_or_create is idempotent - calling twice returns same state."""
|
||||
from kb_pipeline.state import WorkflowStateManager
|
||||
from kb_workflow.pipeline.state import WorkflowStateManager
|
||||
|
||||
manager = WorkflowStateManager(content_dir=tmp_path)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user