mirror of
https://github.com/containrrr/watchtower.git
synced 2024-12-12 09:04:17 +02:00
2842b97df3
Co-authored-by: nils måsén <nils@piksel.se> Co-authored-by: Simon Aronsson <simme@arcticbit.se>
78 lines
2.4 KiB
Go
78 lines
2.4 KiB
Go
package mocks
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/docker/docker/api/types"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
// NewMockAPIServer returns a mocked docker api server that responds to some fixed requests
|
|
// used in the test suite.
|
|
func NewMockAPIServer() *httptest.Server {
|
|
return httptest.NewServer(http.HandlerFunc(
|
|
func(w http.ResponseWriter, r *http.Request) {
|
|
logrus.Debug("Mock server has received a HTTP call on ", r.URL)
|
|
var response = ""
|
|
|
|
if isRequestFor("filters=", r) {
|
|
|
|
Filters := r.URL.Query().Get("filters")
|
|
var result map[string]interface{}
|
|
json.Unmarshal([]byte(Filters), &result)
|
|
status := result["status"].(map[string]interface{})
|
|
|
|
response = getMockJSONFromDisk("./mocks/data/containers.json")
|
|
var x2 []types.Container
|
|
var containers []types.Container
|
|
json.Unmarshal([]byte(response), &containers)
|
|
for _, v := range containers {
|
|
for key := range status {
|
|
if v.State == key {
|
|
x2 = append(x2, v)
|
|
}
|
|
}
|
|
}
|
|
|
|
b, _ := json.Marshal(x2)
|
|
response = string(b)
|
|
|
|
} else if isRequestFor("containers/json?limit=0", r) {
|
|
response = getMockJSONFromDisk("./mocks/data/containers.json")
|
|
} else if isRequestFor("ae8964ba86c7cd7522cf84e09781343d88e0e3543281c747d88b27e246578b65", r) {
|
|
response = getMockJSONFromDisk("./mocks/data/container_stopped.json")
|
|
} else if isRequestFor("b978af0b858aa8855cce46b628817d4ed58e58f2c4f66c9b9c5449134ed4c008", r) {
|
|
response = getMockJSONFromDisk("./mocks/data/container_running.json")
|
|
} else if isRequestFor("ae8964ba86c7cd7522cf84e09781343d88e0e3543281c747d88b27e246578b67", r) {
|
|
response = getMockJSONFromDisk("./mocks/data/container_restarting.json")
|
|
} else if isRequestFor("sha256:19d07168491a3f9e2798a9bed96544e34d57ddc4757a4ac5bb199dea896c87fd", r) {
|
|
response = getMockJSONFromDisk("./mocks/data/image01.json")
|
|
} else if isRequestFor("sha256:4dbc5f9c07028a985e14d1393e849ea07f68804c4293050d5a641b138db72daa", r) {
|
|
response = getMockJSONFromDisk("./mocks/data/image02.json")
|
|
}
|
|
fmt.Fprintln(w, response)
|
|
},
|
|
))
|
|
}
|
|
|
|
func isRequestFor(urlPart string, r *http.Request) bool {
|
|
return strings.Contains(r.URL.String(), urlPart)
|
|
}
|
|
|
|
func getMockJSONFromDisk(relPath string) string {
|
|
absPath, _ := filepath.Abs(relPath)
|
|
logrus.Error(absPath)
|
|
buf, err := ioutil.ReadFile(absPath)
|
|
if err != nil {
|
|
logrus.Error(err)
|
|
return ""
|
|
}
|
|
return string(buf)
|
|
}
|