137 lines
4.4 KiB
Go
137 lines
4.4 KiB
Go
// SPDX-License-Identifier: EUPL-1.2
|
|
|
|
package agentic
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
core "dappco.re/go/core"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestState_HandleStateSet_Good(t *testing.T) {
|
|
subsystem := testPrepWithPlatformServer(t, nil, "")
|
|
result := subsystem.handleStateSet(context.Background(), core.NewOptions(
|
|
core.Option{Key: "plan_slug", Value: "ax-follow-up"},
|
|
core.Option{Key: "key", Value: "pattern"},
|
|
core.Option{Key: "value", Value: `{"name":"observer"}`},
|
|
))
|
|
require.True(t, result.OK)
|
|
|
|
output, ok := result.Value.(StateOutput)
|
|
require.True(t, ok)
|
|
assert.Equal(t, "pattern", output.State.Key)
|
|
|
|
states, err := readPlanStates("ax-follow-up")
|
|
require.NoError(t, err)
|
|
require.Len(t, states, 1)
|
|
assert.Equal(t, "observer", anyMapValue(states[0].Value)["name"])
|
|
}
|
|
|
|
func TestState_HandleStateSet_Bad(t *testing.T) {
|
|
subsystem := testPrepWithPlatformServer(t, nil, "")
|
|
result := subsystem.handleStateSet(context.Background(), core.NewOptions(
|
|
core.Option{Key: "plan_slug", Value: "ax-follow-up"},
|
|
))
|
|
assert.False(t, result.OK)
|
|
}
|
|
|
|
func TestState_HandleStateSet_Ugly_Upsert(t *testing.T) {
|
|
subsystem := testPrepWithPlatformServer(t, nil, "")
|
|
first := subsystem.handleStateSet(context.Background(), core.NewOptions(
|
|
core.Option{Key: "plan_slug", Value: "ax-follow-up"},
|
|
core.Option{Key: "key", Value: "pattern"},
|
|
core.Option{Key: "value", Value: "observer"},
|
|
))
|
|
require.True(t, first.OK)
|
|
|
|
second := subsystem.handleStateSet(context.Background(), core.NewOptions(
|
|
core.Option{Key: "plan_slug", Value: "ax-follow-up"},
|
|
core.Option{Key: "key", Value: "pattern"},
|
|
core.Option{Key: "value", Value: "pipeline"},
|
|
))
|
|
require.True(t, second.OK)
|
|
|
|
states, err := readPlanStates("ax-follow-up")
|
|
require.NoError(t, err)
|
|
require.Len(t, states, 1)
|
|
assert.Equal(t, "pipeline", stringValue(states[0].Value))
|
|
}
|
|
|
|
func TestState_HandleStateGet_Good(t *testing.T) {
|
|
subsystem := testPrepWithPlatformServer(t, nil, "")
|
|
require.NoError(t, writePlanStates("ax-follow-up", []PlanState{{
|
|
Key: "pattern",
|
|
Value: "observer",
|
|
Category: "general",
|
|
}}))
|
|
|
|
result := subsystem.handleStateGet(context.Background(), core.NewOptions(
|
|
core.Option{Key: "plan_slug", Value: "ax-follow-up"},
|
|
core.Option{Key: "key", Value: "pattern"},
|
|
))
|
|
require.True(t, result.OK)
|
|
|
|
output, ok := result.Value.(StateOutput)
|
|
require.True(t, ok)
|
|
assert.Equal(t, "observer", stringValue(output.State.Value))
|
|
}
|
|
|
|
func TestState_HandleStateGet_Bad(t *testing.T) {
|
|
subsystem := testPrepWithPlatformServer(t, nil, "")
|
|
result := subsystem.handleStateGet(context.Background(), core.NewOptions(
|
|
core.Option{Key: "plan_slug", Value: "ax-follow-up"},
|
|
))
|
|
assert.False(t, result.OK)
|
|
}
|
|
|
|
func TestState_HandleStateGet_Ugly_CorruptStateFile(t *testing.T) {
|
|
subsystem := testPrepWithPlatformServer(t, nil, "")
|
|
require.True(t, fs.EnsureDir(stateRoot()).OK)
|
|
require.True(t, fs.Write(statePath("ax-follow-up"), `[{broken`).OK)
|
|
|
|
result := subsystem.handleStateGet(context.Background(), core.NewOptions(
|
|
core.Option{Key: "plan_slug", Value: "ax-follow-up"},
|
|
core.Option{Key: "key", Value: "pattern"},
|
|
))
|
|
assert.False(t, result.OK)
|
|
}
|
|
|
|
func TestState_HandleStateList_Good(t *testing.T) {
|
|
subsystem := testPrepWithPlatformServer(t, nil, "")
|
|
require.NoError(t, writePlanStates("ax-follow-up", []PlanState{
|
|
{Key: "pattern", Value: "observer", Category: "general"},
|
|
{Key: "risk", Value: "auth", Category: "security"},
|
|
}))
|
|
|
|
result := subsystem.handleStateList(context.Background(), core.NewOptions(
|
|
core.Option{Key: "plan_slug", Value: "ax-follow-up"},
|
|
core.Option{Key: "category", Value: "security"},
|
|
))
|
|
require.True(t, result.OK)
|
|
|
|
output, ok := result.Value.(StateListOutput)
|
|
require.True(t, ok)
|
|
assert.Equal(t, 1, output.Total)
|
|
require.Len(t, output.States, 1)
|
|
assert.Equal(t, "risk", output.States[0].Key)
|
|
}
|
|
|
|
func TestState_HandleStateList_Bad(t *testing.T) {
|
|
subsystem := testPrepWithPlatformServer(t, nil, "")
|
|
result := subsystem.handleStateList(context.Background(), core.NewOptions())
|
|
assert.False(t, result.OK)
|
|
}
|
|
|
|
func TestState_HandleStateList_Ugly_CorruptStateFile(t *testing.T) {
|
|
subsystem := testPrepWithPlatformServer(t, nil, "")
|
|
require.True(t, fs.EnsureDir(stateRoot()).OK)
|
|
require.True(t, fs.Write(statePath("ax-follow-up"), `{broken`).OK)
|
|
|
|
result := subsystem.handleStateList(context.Background(), core.NewOptions(
|
|
core.Option{Key: "plan_slug", Value: "ax-follow-up"},
|
|
))
|
|
assert.False(t, result.OK)
|
|
}
|