32 lines
549 B
Go
32 lines
549 B
Go
|
package testgen
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestGetTestPrompt(t *testing.T) {
|
||
|
|
tests := []struct {
|
||
|
|
lang string
|
||
|
|
expected string
|
||
|
|
}{
|
||
|
|
{"Go", "expert Go test writer"},
|
||
|
|
{"Python", "pytest expert"},
|
||
|
|
{"C", "Check framework"},
|
||
|
|
{"C++", "Google Test"},
|
||
|
|
{"Unknown", ""},
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, tt := range tests {
|
||
|
|
t.Run(tt.lang, func(t *testing.T) {
|
||
|
|
prompt := GetTestPrompt(tt.lang)
|
||
|
|
if tt.expected == "" {
|
||
|
|
assert.Empty(t, prompt)
|
||
|
|
} else {
|
||
|
|
assert.Contains(t, prompt, tt.expected)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|