api_key_auth_image_task_test.go 485 B

123456789101112131415
  1. package middleware
  2. import (
  3. "net/http"
  4. "testing"
  5. "github.com/stretchr/testify/require"
  6. )
  7. func TestIsAsyncImageTaskRead(t *testing.T) {
  8. require.True(t, isAsyncImageTaskRead(http.MethodGet, "/v1/images/tasks/imgtask_123"))
  9. require.True(t, isAsyncImageTaskRead(http.MethodGet, "/images/tasks/imgtask_123"))
  10. require.False(t, isAsyncImageTaskRead(http.MethodPost, "/v1/images/tasks/imgtask_123"))
  11. require.False(t, isAsyncImageTaskRead(http.MethodGet, "/v1/images/generations"))
  12. }