105 lines
3.1 KiB
Go
105 lines
3.1 KiB
Go
// This file is part of MinIO Console Server
|
|
// Copyright (c) 2023 MinIO, Inc.
|
|
//
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU Affero General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU Affero General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU Affero General Public License
|
|
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
package api
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/minio/console/api/operations"
|
|
release "github.com/minio/console/api/operations/release"
|
|
"github.com/minio/console/models"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/suite"
|
|
)
|
|
|
|
type ReleasesTestSuite struct {
|
|
suite.Suite
|
|
assert *assert.Assertions
|
|
currentServer string
|
|
isServerSet bool
|
|
getServer *httptest.Server
|
|
withError bool
|
|
}
|
|
|
|
func (suite *ReleasesTestSuite) SetupSuite() {
|
|
suite.assert = assert.New(suite.T())
|
|
suite.getServer = httptest.NewServer(http.HandlerFunc(suite.getHandler))
|
|
suite.currentServer, suite.isServerSet = os.LookupEnv(releaseServiceHostEnvVar)
|
|
os.Setenv(releaseServiceHostEnvVar, suite.getServer.URL)
|
|
}
|
|
|
|
func (suite *ReleasesTestSuite) TearDownSuite() {
|
|
if suite.isServerSet {
|
|
os.Setenv(releaseServiceHostEnvVar, suite.currentServer)
|
|
} else {
|
|
os.Unsetenv(releaseServiceHostEnvVar)
|
|
}
|
|
}
|
|
|
|
func (suite *ReleasesTestSuite) getHandler(
|
|
w http.ResponseWriter, _ *http.Request,
|
|
) {
|
|
if suite.withError {
|
|
w.WriteHeader(400)
|
|
} else {
|
|
w.WriteHeader(200)
|
|
response := &models.ReleaseListResponse{}
|
|
bytes, _ := json.Marshal(response)
|
|
fmt.Fprint(w, string(bytes))
|
|
}
|
|
}
|
|
|
|
func (suite *ReleasesTestSuite) TestRegisterReleasesHandlers() {
|
|
api := &operations.ConsoleAPI{}
|
|
suite.assert.Nil(api.ReleaseListReleasesHandler)
|
|
registerReleasesHandlers(api)
|
|
suite.assert.NotNil(api.ReleaseListReleasesHandler)
|
|
}
|
|
|
|
func (suite *ReleasesTestSuite) TestGetReleasesWithError() {
|
|
api := &operations.ConsoleAPI{}
|
|
current := "mock"
|
|
registerReleasesHandlers(api)
|
|
params := release.NewListReleasesParams()
|
|
params.Current = ¤t
|
|
params.HTTPRequest = &http.Request{}
|
|
suite.withError = true
|
|
response := api.ReleaseListReleasesHandler.Handle(params, &models.Principal{})
|
|
_, ok := response.(*release.ListReleasesDefault)
|
|
suite.assert.True(ok)
|
|
}
|
|
|
|
func (suite *ReleasesTestSuite) TestGetReleasesWithoutError() {
|
|
api := &operations.ConsoleAPI{}
|
|
registerReleasesHandlers(api)
|
|
params := release.NewListReleasesParams()
|
|
params.HTTPRequest = &http.Request{}
|
|
suite.withError = false
|
|
response := api.ReleaseListReleasesHandler.Handle(params, &models.Principal{})
|
|
_, ok := response.(*release.ListReleasesOK)
|
|
suite.assert.True(ok)
|
|
}
|
|
|
|
func TestReleases(t *testing.T) {
|
|
suite.Run(t, new(ReleasesTestSuite))
|
|
}
|