mirror of https://github.com/portainer/portainer
42 lines
1.1 KiB
Go
42 lines
1.1 KiB
Go
|
package middlewares
|
||
|
|
|
||
|
|
import (
|
||
|
|
"io"
|
||
|
|
"net/http"
|
||
|
|
"net/http/httptest"
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func Test_demoEnvironment_shouldFail(t *testing.T) {
|
||
|
|
r := httptest.NewRequest(http.MethodPost, "/", strings.NewReader(`{}`))
|
||
|
|
w := httptest.NewRecorder()
|
||
|
|
|
||
|
|
h := http.HandlerFunc(func(rw http.ResponseWriter, r *http.Request) {})
|
||
|
|
|
||
|
|
RestrictDemoEnv(func() bool { return true }).Middleware(h).ServeHTTP(w, r)
|
||
|
|
|
||
|
|
response := w.Result()
|
||
|
|
defer response.Body.Close()
|
||
|
|
|
||
|
|
assert.Equal(t, http.StatusBadRequest, response.StatusCode)
|
||
|
|
|
||
|
|
body, _ := io.ReadAll(response.Body)
|
||
|
|
assert.Contains(t, string(body), "This feature is not available in the demo version of Portainer")
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_notDemoEnvironment_shouldSucceed(t *testing.T) {
|
||
|
|
r := httptest.NewRequest(http.MethodPost, "/", strings.NewReader(`{}`))
|
||
|
|
w := httptest.NewRecorder()
|
||
|
|
|
||
|
|
h := http.HandlerFunc(func(rw http.ResponseWriter, r *http.Request) {})
|
||
|
|
|
||
|
|
RestrictDemoEnv(func() bool { return false }).Middleware(h).ServeHTTP(w, r)
|
||
|
|
|
||
|
|
response := w.Result()
|
||
|
|
assert.Equal(t, http.StatusOK, response.StatusCode)
|
||
|
|
|
||
|
|
}
|