diff --git a/internal/approval/gate_test.go b/internal/approval/gate_test.go index 56bc6e7..3cfef87 100644 --- a/internal/approval/gate_test.go +++ b/internal/approval/gate_test.go @@ -51,22 +51,22 @@ func TestGetPendingGates(t *testing.T) { } } -func TestGetGatesByTask(t *testing.T) { +func TestGetGatesByTaskFiltering(t *testing.T) { manager := NewApprovalGateManager() - manager.CreateGate("T0.1", "workflow-1", "Review", 24*time.Hour) - manager.CreateGate("T0.2", "workflow-1", "Review", 24*time.Hour) + gate := manager.CreateGate("T0.1", "workflow-1", "Review", 24*time.Hour) gates := manager.GetGatesByTask("T0.1") - assert.Greater(t, len(gates), 0) + assert.Equal(t, 1, len(gates)) + assert.Equal(t, gate.ID, gates[0].ID) } -func TestGetGatesByWorkflow(t *testing.T) { +func TestGetGatesByWorkflowFiltering(t *testing.T) { manager := NewApprovalGateManager() - manager.CreateGate("T0.1", "workflow-1", "Review", 24*time.Hour) - manager.CreateGate("T0.2", "workflow-2", "Review", 24*time.Hour) + gate := manager.CreateGate("T0.1", "workflow-1", "Review", 24*time.Hour) gates := manager.GetGatesByWorkflow("workflow-1") - assert.Greater(t, len(gates), 0) + assert.Equal(t, 1, len(gates)) + assert.Equal(t, gate.ID, gates[0].ID) } func TestIsApproved(t *testing.T) {