mirror of
https://github.com/QuantumNous/new-api.git
synced 2026-09-10 22:20:25 +00:00
feat(task): replace built-in task adaptors with a sandboxed JS plugin system (#7076)
This commit is contained in:
@@ -0,0 +1,27 @@
|
||||
package constant
|
||||
|
||||
import (
|
||||
"testing"
|
||||
|
||||
"github.com/stretchr/testify/assert"
|
||||
)
|
||||
|
||||
func TestNormalizeTaskAction(t *testing.T) {
|
||||
tests := map[string]string{
|
||||
"generate": TaskActionImageToVideo,
|
||||
"textGenerate": TaskActionTextToVideo,
|
||||
"firstTailGenerate": TaskActionFirstTailToVideo,
|
||||
"referenceGenerate": TaskActionReferenceToVideo,
|
||||
"remixGenerate": TaskActionRemix,
|
||||
TaskActionTextToVideo: TaskActionTextToVideo,
|
||||
"MUSIC": "MUSIC",
|
||||
"custom_action": "custom_action",
|
||||
"": "",
|
||||
}
|
||||
|
||||
for input, expected := range tests {
|
||||
t.Run(input, func(t *testing.T) {
|
||||
assert.Equal(t, expected, NormalizeTaskAction(input))
|
||||
})
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user