mirror of
https://git.vectorsigma.ru/public/atlantis.git
synced 2026-07-29 14:58:19 +00:00
7ac8106f8a
This is two changes: Replacing all usages of SimpleLogger with it's interface SimpleLogging which makes it easier to swap out loggers going forward Nukes SimpleLogger in favor of StructuredLogger which allows us to better analyze logs in our log management system we choose.
83 lines
1.9 KiB
Go
83 lines
1.9 KiB
Go
package server_test
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/runatlantis/atlantis/server"
|
|
"github.com/runatlantis/atlantis/server/events"
|
|
"github.com/runatlantis/atlantis/server/logging"
|
|
. "github.com/runatlantis/atlantis/testing"
|
|
)
|
|
|
|
func TestStatusController_Startup(t *testing.T) {
|
|
logger := logging.NewNoopLogger(t)
|
|
r, _ := http.NewRequest("GET", "/status", bytes.NewBuffer(nil))
|
|
w := httptest.NewRecorder()
|
|
dr := &events.Drainer{}
|
|
d := &server.StatusController{
|
|
Logger: logger,
|
|
Drainer: dr,
|
|
}
|
|
d.Get(w, r)
|
|
|
|
var result server.StatusResponse
|
|
body, err := ioutil.ReadAll(w.Result().Body)
|
|
Ok(t, err)
|
|
Equals(t, 200, w.Result().StatusCode)
|
|
err = json.Unmarshal(body, &result)
|
|
Ok(t, err)
|
|
Equals(t, false, result.ShuttingDown)
|
|
Equals(t, 0, result.InProgressOps)
|
|
}
|
|
|
|
func TestStatusController_InProgress(t *testing.T) {
|
|
logger := logging.NewNoopLogger(t)
|
|
r, _ := http.NewRequest("GET", "/status", bytes.NewBuffer(nil))
|
|
w := httptest.NewRecorder()
|
|
dr := &events.Drainer{}
|
|
dr.StartOp()
|
|
|
|
d := &server.StatusController{
|
|
Logger: logger,
|
|
Drainer: dr,
|
|
}
|
|
d.Get(w, r)
|
|
|
|
var result server.StatusResponse
|
|
body, err := ioutil.ReadAll(w.Result().Body)
|
|
Ok(t, err)
|
|
Equals(t, 200, w.Result().StatusCode)
|
|
err = json.Unmarshal(body, &result)
|
|
Ok(t, err)
|
|
Equals(t, false, result.ShuttingDown)
|
|
Equals(t, 1, result.InProgressOps)
|
|
}
|
|
|
|
func TestStatusController_Shutdown(t *testing.T) {
|
|
logger := logging.NewNoopLogger(t)
|
|
r, _ := http.NewRequest("GET", "/status", bytes.NewBuffer(nil))
|
|
w := httptest.NewRecorder()
|
|
dr := &events.Drainer{}
|
|
dr.ShutdownBlocking()
|
|
|
|
d := &server.StatusController{
|
|
Logger: logger,
|
|
Drainer: dr,
|
|
}
|
|
d.Get(w, r)
|
|
|
|
var result server.StatusResponse
|
|
body, err := ioutil.ReadAll(w.Result().Body)
|
|
Ok(t, err)
|
|
Equals(t, 200, w.Result().StatusCode)
|
|
err = json.Unmarshal(body, &result)
|
|
Ok(t, err)
|
|
Equals(t, true, result.ShuttingDown)
|
|
Equals(t, 0, result.InProgressOps)
|
|
}
|