Chaim Lev-Ari
223494141a
* feat(server): add demo cli flag * feat(backup): block backup operations on demo version * fix(http/backup): fix tests * feat(backup): test demo block * refactor(backup): move demo flag * Trigger Build Co-authored-by: yi-portainer <yi.chen@portainer.io>
36 lines
924 B
Go
36 lines
924 B
Go
package backup
|
|
|
|
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()
|
|
|
|
restrictDemoEnv(true, http.DefaultServeMux).ServeHTTP(w, r)
|
|
|
|
response := w.Result()
|
|
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()
|
|
|
|
restrictDemoEnv(false, http.HandlerFunc(func(rw http.ResponseWriter, r *http.Request) {})).ServeHTTP(w, r)
|
|
|
|
response := w.Result()
|
|
assert.Equal(t, http.StatusOK, response.StatusCode)
|
|
|
|
}
|