Files
new-api-custom/service/task_artifact_store_test.go
T
Codex bb7968b612
Publish Docker image (Multi-arch) / Build & push (amd64) (push) Failing after 9m20s
Release (Linux, macOS, Windows) / Linux Release (push) Failing after 55m7s
Publish Docker image (Multi-arch) / Build & push (arm64) (push) Canceled after 0s
Release (Linux, macOS, Windows) / macOS Release (push) Canceled after 0s
Release (Linux, macOS, Windows) / Windows Release (push) Canceled after 0s
Publish Docker image (Multi-arch) / Create multi-arch manifests (push) Canceled after 0s
feat: publish v1.0.0-rc.36-custom.1
2026-09-11 05:07:48 +08:00

32 lines
959 B
Go

package service
import (
"strings"
"testing"
"github.com/QuantumNous/new-api/model"
"github.com/QuantumNous/new-api/types"
"github.com/gin-gonic/gin"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
func TestDisabledTaskArtifactStoreHasNoStorageBehavior(t *testing.T) {
store := GetTaskArtifactStore()
require.NotNil(t, store)
assert.False(t, store.Enabled())
task := &model.Task{TaskID: "task-disabled-store"}
ref, err := store.Resolve(task, "video")
require.NoError(t, err)
assert.Nil(t, ref)
ref, err = store.Persist(t.Context(), task, types.TaskArtifact{Key: "video", Type: "video"}, strings.NewReader("content"))
assert.Nil(t, ref)
assert.ErrorIs(t, err, ErrTaskArtifactStoreDisabled)
assert.ErrorIs(t, store.Serve(&gin.Context{}, task, &StoredArtifactRef{Backend: "s3"}), ErrTaskArtifactStoreDisabled)
assert.Same(t, store, GetTaskArtifactStore())
}
var _ TaskArtifactStore = disabledArtifactStore{}