72 lines
2.3 KiB
Go
72 lines
2.3 KiB
Go
package service
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
// ========== parseJSONResponse Tests ==========
|
|
|
|
func TestParseJSONResponse_DirectJSON(t *testing.T) {
|
|
content := `{"suggestions": [{"labels": ["a", "b"], "priority": "high", "reason": "test reason"}]}`
|
|
var result LabelSuggestionResult
|
|
err := parseJSONResponse(content, &result)
|
|
|
|
require.NoError(t, err)
|
|
assert.Len(t, result.Suggestions, 1)
|
|
assert.Equal(t, "high", result.Suggestions[0].Priority)
|
|
assert.Len(t, result.Suggestions[0].Labels, 2)
|
|
}
|
|
|
|
func TestParseJSONResponse_MarkdownCodeBlock(t *testing.T) {
|
|
content := "```json\n{\"suggestions\": [{\"labels\": [\"test\"], \"priority\": \"low\", \"reason\": \"test reason\"}]}\n```"
|
|
var result LabelSuggestionResult
|
|
err := parseJSONResponse(content, &result)
|
|
|
|
require.NoError(t, err)
|
|
assert.Len(t, result.Suggestions, 1)
|
|
assert.Equal(t, "low", result.Suggestions[0].Priority)
|
|
assert.Equal(t, "test reason", result.Suggestions[0].Reason)
|
|
}
|
|
|
|
func TestParseJSONResponse_PlainCodeBlock(t *testing.T) {
|
|
content := "```json\n{\"suggestions\": [{\"labels\": [], \"priority\": \"medium\"}]}\n```"
|
|
var result LabelSuggestionResult
|
|
err := parseJSONResponse(content, &result)
|
|
|
|
require.NoError(t, err)
|
|
assert.Len(t, result.Suggestions, 1)
|
|
assert.Equal(t, "medium", result.Suggestions[0].Priority)
|
|
}
|
|
|
|
func TestParseJSONResponse_EmbeddedJSON(t *testing.T) {
|
|
content := "Here is the analysis:\n{\"participants\": [], \"summary\": \"test summary\"}\nEnd of analysis."
|
|
var result ParticipantAnalysisResult
|
|
err := parseJSONResponse(content, &result)
|
|
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "test summary", result.Summary)
|
|
}
|
|
|
|
func TestParseJSONResponse_InvalidJSON(t *testing.T) {
|
|
content := "this is not json at all"
|
|
var result LabelSuggestionResult
|
|
err := parseJSONResponse(content, &result)
|
|
|
|
assert.Error(t, err)
|
|
}
|
|
|
|
func TestExtractLabelSuggestionFromText(t *testing.T) {
|
|
content := "Label: bug, feature\nPriority: high\nReason: urgent issue"
|
|
result := extractLabelSuggestionFromText(content)
|
|
|
|
assert.NotEmpty(t, result.Suggestions)
|
|
if len(result.Suggestions) > 0 {
|
|
assert.Contains(t, result.Suggestions[0].Labels, "bug")
|
|
assert.Contains(t, result.Suggestions[0].Labels, "feature")
|
|
assert.Equal(t, "high", result.Suggestions[0].Priority)
|
|
assert.Equal(t, "urgent issue", result.Suggestions[0].Reason)
|
|
}
|
|
} |