mirror of
https://git.vectorsigma.ru/public/atlantis.git
synced 2026-07-28 19:38:24 +00:00
126 lines
3.6 KiB
Go
126 lines
3.6 KiB
Go
// Copyright 2017 HootSuite Media Inc.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the License);
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an AS IS BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
// Modified hereafter by contributors to runatlantis/atlantis.
|
|
|
|
package server_test
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/gorilla/mux"
|
|
. "github.com/petergtz/pegomock"
|
|
"github.com/runatlantis/atlantis/server"
|
|
"github.com/runatlantis/atlantis/server/events/locking/mocks"
|
|
"github.com/runatlantis/atlantis/server/events/models"
|
|
sMocks "github.com/runatlantis/atlantis/server/mocks"
|
|
. "github.com/runatlantis/atlantis/testing"
|
|
)
|
|
|
|
func TestNewServer(t *testing.T) {
|
|
t.Log("Run through NewServer constructor")
|
|
tmpDir, err := ioutil.TempDir("", "")
|
|
Ok(t, err)
|
|
_, err = server.NewServer(server.UserConfig{
|
|
DataDir: tmpDir,
|
|
}, server.Config{})
|
|
Ok(t, err)
|
|
}
|
|
|
|
func TestIndex_LockErr(t *testing.T) {
|
|
t.Log("index should return a 503 if unable to list locks")
|
|
RegisterMockTestingT(t)
|
|
l := mocks.NewMockLocker()
|
|
When(l.List()).ThenReturn(nil, errors.New("err"))
|
|
s := server.Server{
|
|
Locker: l,
|
|
}
|
|
req, _ := http.NewRequest("GET", "", bytes.NewBuffer(nil))
|
|
w := httptest.NewRecorder()
|
|
s.Index(w, req)
|
|
responseContains(t, w, 503, "Could not retrieve locks: err")
|
|
}
|
|
|
|
func TestIndex_Success(t *testing.T) {
|
|
t.Log("Index should render the index template successfully.")
|
|
RegisterMockTestingT(t)
|
|
l := mocks.NewMockLocker()
|
|
// These are the locks that we expect to be rendered.
|
|
now := time.Now()
|
|
locks := map[string]models.ProjectLock{
|
|
"id1": {
|
|
Pull: models.PullRequest{
|
|
Num: 9,
|
|
},
|
|
Project: models.Project{
|
|
RepoFullName: "owner/repo",
|
|
},
|
|
Time: now,
|
|
},
|
|
}
|
|
When(l.List()).ThenReturn(locks, nil)
|
|
it := sMocks.NewMockTemplateWriter()
|
|
r := mux.NewRouter()
|
|
atlantisVersion := "0.3.1"
|
|
// Need to create a lock route since the server expects this route to exist.
|
|
r.NewRoute().Path("").Name(server.LockViewRouteName)
|
|
s := server.Server{
|
|
Locker: l,
|
|
IndexTemplate: it,
|
|
Router: r,
|
|
AtlantisVersion: atlantisVersion,
|
|
}
|
|
req, _ := http.NewRequest("GET", "", bytes.NewBuffer(nil))
|
|
w := httptest.NewRecorder()
|
|
s.Index(w, req)
|
|
it.VerifyWasCalledOnce().Execute(w, server.IndexData{
|
|
Locks: []server.LockIndexData{
|
|
{
|
|
LockURL: "",
|
|
RepoFullName: "owner/repo",
|
|
PullNum: 9,
|
|
Time: now,
|
|
},
|
|
},
|
|
AtlantisVersion: atlantisVersion,
|
|
})
|
|
responseContains(t, w, http.StatusOK, "")
|
|
}
|
|
|
|
func TestHealthz(t *testing.T) {
|
|
s := server.Server{}
|
|
req, _ := http.NewRequest("GET", "/healthz", bytes.NewBuffer(nil))
|
|
w := httptest.NewRecorder()
|
|
s.Healthz(w, req)
|
|
Equals(t, http.StatusOK, w.Result().StatusCode)
|
|
body, _ := ioutil.ReadAll(w.Result().Body)
|
|
Equals(t, "application/json", w.Result().Header["Content-Type"][0])
|
|
Equals(t,
|
|
`{
|
|
"status": "ok"
|
|
}`, string(body))
|
|
}
|
|
|
|
func responseContains(t *testing.T, r *httptest.ResponseRecorder, status int, bodySubstr string) {
|
|
t.Helper()
|
|
body, err := ioutil.ReadAll(r.Result().Body)
|
|
Ok(t, err)
|
|
Assert(t, status == r.Result().StatusCode, "exp %d got %d, body: %s", status, r.Result().StatusCode, string(body))
|
|
Assert(t, strings.Contains(string(body), bodySubstr), "exp %q to be contained in %q", bodySubstr, string(body))
|
|
}
|