Carlos Florêncio
68 lines
2.3 KiB
Go
68 lines
2.3 KiB
Go
package lifecycle
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestShouldPersistMetadataKey(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
key string
|
|
want bool
|
|
}{
|
|
{name: "exact match sprite_name", key: "sprite_name", want: true},
|
|
{name: "exact match is_remote", key: MetadataKeyIsRemote, want: true},
|
|
{name: "exact match cleanup_script", key: MetadataKeyCleanupScript, want: true},
|
|
{name: "exact match executor_profile_id", key: "executor_profile_id", want: true},
|
|
{name: "prefix env_secret_id_", key: "env_secret_id_SPRITES_API_TOKEN", want: true},
|
|
{name: "prefix env_secret_id_ another key", key: "env_secret_id_OPENAI_KEY", want: true},
|
|
{name: "not persistent task_description", key: "task_description", want: false},
|
|
{name: "not persistent session_id", key: "session_id", want: false},
|
|
{name: "not persistent empty", key: "", want: false},
|
|
{name: "not persistent arbitrary key", key: "some_random_key", want: false},
|
|
{name: "partial prefix match (no underscore)", key: "env_secret_id", want: false},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
got := ShouldPersistMetadataKey(tt.key)
|
|
require.Equal(t, tt.want, got)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestFilterPersistentMetadata(t *testing.T) {
|
|
t.Run("nil input returns nil", func(t *testing.T) {
|
|
require.Nil(t, FilterPersistentMetadata(nil))
|
|
})
|
|
|
|
t.Run("empty map returns nil", func(t *testing.T) {
|
|
require.Nil(t, FilterPersistentMetadata(map[string]interface{}{}))
|
|
})
|
|
|
|
t.Run("no persistent keys returns nil", func(t *testing.T) {
|
|
src := map[string]interface{}{
|
|
"task_description": "do something",
|
|
"session_id": "abc",
|
|
}
|
|
require.Nil(t, FilterPersistentMetadata(src))
|
|
})
|
|
|
|
t.Run("filters to persistent keys only", func(t *testing.T) {
|
|
src := map[string]interface{}{
|
|
"sprite_name": "kandev-abc",
|
|
"task_description": "should be dropped",
|
|
"env_secret_id_SPRITES_API_TOKEN": "secret-123",
|
|
MetadataKeyIsRemote: true,
|
|
}
|
|
got := FilterPersistentMetadata(src)
|
|
require.NotNil(t, got)
|
|
require.Equal(t, "kandev-abc", got["sprite_name"])
|
|
require.Equal(t, "secret-123", got["env_secret_id_SPRITES_API_TOKEN"])
|
|
require.Equal(t, true, got[MetadataKeyIsRemote])
|
|
require.NotContains(t, got, "task_description")
|
|
})
|
|
}
|