frontmatter_command_test.go

package markdown

import (
	"testing"

	"github.com/kitproj/coding-context-cli/pkg/codingcontext/urn"
	"gopkg.in/yaml.v3"
)

// mustParseURN is a test helper that parses a URN string and panics on error
func mustParseURNCommand(s string) *urn.URN {
	u, ok := urn.Parse([]byte(s))
	if !ok {
		panic("failed to parse URN: " + s)
	}
	return u
}

func TestCommandFrontMatter_Marshal(t *testing.T) {
	tests := []struct {
		name    string
		command CommandFrontMatter
		want    string
	}{
		{
			name:    "minimal command",
			command: CommandFrontMatter{},
			want:    "{}\n",
		},
		{
			name: "command with standard id, name, description",
			command: CommandFrontMatter{
				BaseFrontMatter: BaseFrontMatter{
					URN:         mustParseURNCommand("urn:agents:command:standard"),
					Name:        "Standard Command",
					Description: "This is a standard command with metadata",
				},
			},
			want: `urn: urn:agents:command:standard
name: Standard Command
description: This is a standard command with metadata
`,
		},
		{
			name: "command with expand false",
			command: CommandFrontMatter{
				BaseFrontMatter: BaseFrontMatter{
					URN:         mustParseURNCommand("urn:agents:command:no-expand"),
					Name:        "No Expand Command",
					Description: "Command with expansion disabled",
				},
				ExpandParams: func() *bool {
					b := false
					return &b
				}(),
			},
			want: `urn: urn:agents:command:no-expand
name: No Expand Command
description: Command with expansion disabled
expand: false
`,
		},
		{
			name: "command with selectors",
			command: CommandFrontMatter{
				BaseFrontMatter: BaseFrontMatter{
					URN:         mustParseURNCommand("urn:agents:command:selector"),
					Name:        "Selector Command",
					Description: "Command with selectors",
				},
				Selectors: map[string]any{
					"database": "postgres",
					"feature":  "auth",
				},
			},
			want: `urn: urn:agents:command:selector
name: Selector Command
description: Command with selectors
selectors:
    database: postgres
    feature: auth
`,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			got, err := yaml.Marshal(&tt.command)
			if err != nil {
				t.Fatalf("Marshal() error = %v", err)
			}
			if string(got) != tt.want {
				t.Errorf("Marshal() = %q, want %q", string(got), tt.want)
			}
		})
	}
}

func TestCommandFrontMatter_Unmarshal(t *testing.T) {
	tests := []struct {
		name    string
		yaml    string
		want    CommandFrontMatter
		wantErr bool
	}{
		{
			name: "command with standard id, name, description",
			yaml: `urn: urn:agents:command:named
name: Named Command
description: A command with standard fields
`,
			want: CommandFrontMatter{
				BaseFrontMatter: BaseFrontMatter{
					URN:         mustParseURNCommand("urn:agents:command:named"),
					Name:        "Named Command",
					Description: "A command with standard fields",
				},
			},
		},
		{
			name: "command with expand false",
			yaml: `urn: urn:agents:command:no-expand
name: No Expand
description: No expansion
expand: false
`,
			want: CommandFrontMatter{
				BaseFrontMatter: BaseFrontMatter{
					URN:         mustParseURNCommand("urn:agents:command:no-expand"),
					Name:        "No Expand",
					Description: "No expansion",
				},
				ExpandParams: func() *bool {
					b := false
					return &b
				}(),
			},
		},
		{
			name: "command with selectors",
			yaml: `urn: urn:agents:command:selector
name: Selector Command
description: Has selectors
selectors:
  database: postgres
  feature: auth
`,
			want: CommandFrontMatter{
				BaseFrontMatter: BaseFrontMatter{
					URN:         mustParseURNCommand("urn:agents:command:selector"),
					Name:        "Selector Command",
					Description: "Has selectors",
				},
				Selectors: map[string]any{
					"database": "postgres",
					"feature":  "auth",
				},
			},
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			var got CommandFrontMatter
			err := yaml.Unmarshal([]byte(tt.yaml), &got)
			if (err != nil) != tt.wantErr {
				t.Fatalf("Unmarshal() error = %v, wantErr %v", err, tt.wantErr)
			}
			if err != nil {
				return
			}

			// Compare fields individually
			if (got.URN == nil) != (tt.want.URN == nil) {
				t.Errorf("URN = %v, want %v", got.URN, tt.want.URN)
			} else if got.URN != nil && tt.want.URN != nil && !got.URN.Equal(tt.want.URN) {
				t.Errorf("URN = %q, want %q", got.URN, tt.want.URN)
			}
			if got.Name != tt.want.Name {
				t.Errorf("Name = %q, want %q", got.Name, tt.want.Name)
			}
			if got.Description != tt.want.Description {
				t.Errorf("Description = %q, want %q", got.Description, tt.want.Description)
			}
			if (got.ExpandParams == nil) != (tt.want.ExpandParams == nil) {
				t.Errorf("ExpandParams nil mismatch: got %v, want %v", got.ExpandParams == nil, tt.want.ExpandParams == nil)
			} else if got.ExpandParams != nil && tt.want.ExpandParams != nil {
				if *got.ExpandParams != *tt.want.ExpandParams {
					t.Errorf("ExpandParams = %v, want %v", *got.ExpandParams, *tt.want.ExpandParams)
				}
			}
		})
	}
}