new-api/service/midjourney_helpers_test.go

package service

import (
	"testing"

	"github.com/stretchr/testify/require"
)

func TestConvertSimpleChangeParamsParsesSupportedActions(t *testing.T) {
	upscale := ConvertSimpleChangeParams("task-1 U3")
	require.NotNil(t, upscale)
	require.Equal(t, "task-1", upscale.TaskId)
	require.Equal(t, "UPSCALE", upscale.Action)
	require.Equal(t, 3, upscale.Index)

	reroll := ConvertSimpleChangeParams("task-1 R")
	require.NotNil(t, reroll)
	require.Equal(t, "REROLL", reroll.Action)
}

func TestConvertSimpleChangeParamsRejectsBlankOrMalformedAction(t *testing.T) {
	require.Nil(t, ConvertSimpleChangeParams("task-1 "))
	require.Nil(t, ConvertSimpleChangeParams("task-1 X1"))
	require.Nil(t, ConvertSimpleChangeParams("task-1 U0"))
}