Add some conifguration and logging for the hub
This commit is contained in:
14
appconfig/endpoints.go
Normal file
14
appconfig/endpoints.go
Normal file
@@ -0,0 +1,14 @@
|
|||||||
|
package appconfig
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"net/http"
|
||||||
|
)
|
||||||
|
|
||||||
|
func handleAuthzConfig(w http.ResponseWriter, req *http.Request) {
|
||||||
|
headers := w.Header()
|
||||||
|
headers.Set("content-type", "text/plain")
|
||||||
|
|
||||||
|
w.WriteHeader(http.StatusOK)
|
||||||
|
fmt.Fprintf(w, "hello, world")
|
||||||
|
}
|
||||||
33
appconfig/log.go
Normal file
33
appconfig/log.go
Normal file
@@ -0,0 +1,33 @@
|
|||||||
|
package appconfig
|
||||||
|
|
||||||
|
import "fmt"
|
||||||
|
|
||||||
|
type logger interface {
|
||||||
|
Output(int, string) error
|
||||||
|
}
|
||||||
|
|
||||||
|
type ilogger interface {
|
||||||
|
logger
|
||||||
|
Print(...interface{})
|
||||||
|
Printf(string, ...interface{})
|
||||||
|
Println(...interface{})
|
||||||
|
}
|
||||||
|
|
||||||
|
type internalLog struct {
|
||||||
|
logger
|
||||||
|
}
|
||||||
|
|
||||||
|
// Println replicates the behaviour of the standard logger.
|
||||||
|
func (t internalLog) Println(v ...interface{}) {
|
||||||
|
t.Output(2, fmt.Sprintln(v...))
|
||||||
|
}
|
||||||
|
|
||||||
|
// Printf replicates the behaviour of the standard logger.
|
||||||
|
func (t internalLog) Printf(format string, v ...interface{}) {
|
||||||
|
t.Output(2, fmt.Sprintf(format, v...))
|
||||||
|
}
|
||||||
|
|
||||||
|
// Print replicates the behaviour of the standard logger.
|
||||||
|
func (t internalLog) Print(v ...interface{}) {
|
||||||
|
t.Output(2, fmt.Sprint(v...))
|
||||||
|
}
|
||||||
31
appconfig/provider.go
Normal file
31
appconfig/provider.go
Normal file
@@ -0,0 +1,31 @@
|
|||||||
|
package appconfig
|
||||||
|
|
||||||
|
import (
|
||||||
|
"log"
|
||||||
|
"net/http"
|
||||||
|
"os"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Provider struct {
|
||||||
|
handler http.Handler
|
||||||
|
log ilogger
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewProvider() Provider {
|
||||||
|
return Provider{
|
||||||
|
handler: router(),
|
||||||
|
log: log.New(os.Stderr, "servicedemon/appconfig - ", log.LstdFlags|log.Lshortfile),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p Provider) Handler() http.Handler {
|
||||||
|
return p.handler
|
||||||
|
}
|
||||||
|
|
||||||
|
func router() *http.ServeMux {
|
||||||
|
mux := http.NewServeMux()
|
||||||
|
|
||||||
|
mux.HandleFunc("/config/authz", handleAuthzConfig)
|
||||||
|
|
||||||
|
return mux
|
||||||
|
}
|
||||||
@@ -4,28 +4,17 @@ import (
|
|||||||
"context"
|
"context"
|
||||||
"crypto/tls"
|
"crypto/tls"
|
||||||
"crypto/x509"
|
"crypto/x509"
|
||||||
|
"fmt"
|
||||||
"io"
|
"io"
|
||||||
"log"
|
"log"
|
||||||
"net/http"
|
"net/http"
|
||||||
"os"
|
"os"
|
||||||
|
|
||||||
"go.fixergrid.net/servicedemon/pkg/pubsub"
|
"go.fixergrid.net/servicedemon/appconfig"
|
||||||
"go.fixergrid.net/servicedemon/pkg/registrar"
|
"go.fixergrid.net/servicedemon/pubsub"
|
||||||
|
"go.fixergrid.net/servicedemon/registrar"
|
||||||
)
|
)
|
||||||
|
|
||||||
type noopHandler struct {
|
|
||||||
http.HandlerFunc
|
|
||||||
}
|
|
||||||
|
|
||||||
func wrapHandlefunc(h http.HandlerFunc) noopHandler {
|
|
||||||
return noopHandler{
|
|
||||||
HandlerFunc: h,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (h noopHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
|
|
||||||
h.HandlerFunc(w, req)
|
|
||||||
}
|
|
||||||
func main() {
|
func main() {
|
||||||
ctx, cancel := context.WithCancel(context.Background())
|
ctx, cancel := context.WithCancel(context.Background())
|
||||||
defer cancel()
|
defer cancel()
|
||||||
@@ -39,41 +28,79 @@ func main() {
|
|||||||
r := registrar.NewRegistrar(
|
r := registrar.NewRegistrar(
|
||||||
pubsub,
|
pubsub,
|
||||||
repo,
|
repo,
|
||||||
registrar.WithLogger(log.New(os.Stdout, "registrar: ", log.LstdFlags|log.Lshortfile)),
|
|
||||||
)
|
)
|
||||||
|
|
||||||
al := registrar.NewApprovalListener(
|
al := registrar.NewApprovalListener(
|
||||||
pubsub,
|
pubsub,
|
||||||
nil,
|
nil,
|
||||||
repo,
|
repo,
|
||||||
registrar.OptionLog(log.New(os.Stdout, "approvalListener: ", log.LstdFlags|log.Lshortfile)),
|
|
||||||
)
|
)
|
||||||
go al.Run(ctx)
|
go al.Run(ctx)
|
||||||
|
|
||||||
|
appConfig := appconfig.NewProvider()
|
||||||
|
|
||||||
mux := http.NewServeMux()
|
mux := http.NewServeMux()
|
||||||
|
logger.Println("Registering endpoints...")
|
||||||
|
mux.HandleFunc("/register", postjson(r.HandleRegistration))
|
||||||
|
logger.Println("POST /register")
|
||||||
|
|
||||||
mux.HandleFunc("/register", r.HandleRegistration)
|
mux.Handle("/approvals/", http.StripPrefix("/approvals/", wrapHandleFunc(postjson(r.HandleApproval))))
|
||||||
mux.Handle("/approvals/", http.StripPrefix("/approvals/", wrapHandlefunc(r.HandleApproval)))
|
logger.Println("POST /approvals/:id")
|
||||||
|
|
||||||
certFile, err := os.Open("./certs/ca.pem")
|
mux.Handle("/application/", http.StripPrefix("/application", appConfig.Handler()))
|
||||||
|
logger.Println("GET /application/config/authz")
|
||||||
|
|
||||||
|
server, err := newServer()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Fatalf("failed to open ca.pem: %v", err)
|
logger.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
server.Handler = mux
|
||||||
|
log.Println(server.ListenAndServeTLS("", ""))
|
||||||
|
}
|
||||||
|
|
||||||
|
func newServer() (*http.Server, error) {
|
||||||
|
// "./certs/combined.pem", "./certs/server-key.pem"
|
||||||
|
requiredVars := map[string]string{
|
||||||
|
"HUB_CA_CERT_FILE": "",
|
||||||
|
"HUB_SERVER_CERT_FILE": "",
|
||||||
|
"HUB_SERVER_KEY_FILE": "",
|
||||||
|
}
|
||||||
|
|
||||||
|
for k, _ := range requiredVars {
|
||||||
|
val, isSet := os.LookupEnv(k)
|
||||||
|
if !isSet {
|
||||||
|
return nil, fmt.Errorf("hub: required environment variable is unset: %s", k)
|
||||||
|
}
|
||||||
|
requiredVars[k] = val
|
||||||
|
}
|
||||||
|
|
||||||
|
certFile, err := os.Open(requiredVars["HUB_CA_CERT_FILE"])
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("hub: failed to open ca.pem: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
caCert, err := io.ReadAll(certFile)
|
caCert, err := io.ReadAll(certFile)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logger.Fatalf("failed to read in ca: %v", err)
|
return nil, fmt.Errorf("hub: failed to read in ca: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
pool := x509.NewCertPool()
|
pool := x509.NewCertPool()
|
||||||
pool.AppendCertsFromPEM(caCert)
|
pool.AppendCertsFromPEM(caCert)
|
||||||
|
|
||||||
|
serverCert, err := tls.LoadX509KeyPair(requiredVars["HUB_SERVER_CERT_FILE"], requiredVars["HUB_SERVER_KEY_FILE"])
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("hub: failed to load server certs: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
server := &http.Server{
|
server := &http.Server{
|
||||||
Addr: ":3001",
|
Addr: ":3001",
|
||||||
TLSConfig: &tls.Config{
|
TLSConfig: &tls.Config{
|
||||||
ClientCAs: pool,
|
ClientCAs: pool,
|
||||||
ClientAuth: tls.RequireAndVerifyClientCert,
|
ClientAuth: tls.RequireAndVerifyClientCert,
|
||||||
|
Certificates: []tls.Certificate{serverCert},
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
server.Handler = mux
|
|
||||||
log.Println(server.ListenAndServeTLS("./certs/combined.pem", "./certs/server-key.pem"))
|
return server, nil
|
||||||
}
|
}
|
||||||
|
|||||||
46
cmd/hub/middlewares.go
Normal file
46
cmd/hub/middlewares.go
Normal file
@@ -0,0 +1,46 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"net/http"
|
||||||
|
)
|
||||||
|
|
||||||
|
func postjson(f http.HandlerFunc) http.HandlerFunc {
|
||||||
|
return contentJSON(methodPost(f))
|
||||||
|
}
|
||||||
|
|
||||||
|
func methodPost(f http.HandlerFunc) http.HandlerFunc {
|
||||||
|
return func(w http.ResponseWriter, req *http.Request) {
|
||||||
|
if req.Method != http.MethodPost {
|
||||||
|
w.WriteHeader(http.StatusMethodNotAllowed)
|
||||||
|
fmt.Fprintf(w, `{"errors": ["method not allowed"]}`)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
f(w, req)
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func contentJSON(f http.HandlerFunc) http.HandlerFunc {
|
||||||
|
return func(w http.ResponseWriter, req *http.Request) {
|
||||||
|
headers := w.Header()
|
||||||
|
headers.Set("content-type", "application/json")
|
||||||
|
|
||||||
|
f(w, req)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
type noopHandler struct {
|
||||||
|
http.HandlerFunc
|
||||||
|
}
|
||||||
|
|
||||||
|
func wrapHandleFunc(h http.HandlerFunc) noopHandler {
|
||||||
|
return noopHandler{
|
||||||
|
HandlerFunc: h,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h noopHandler) ServeHTTP(w http.ResponseWriter, req *http.Request) {
|
||||||
|
h.HandlerFunc(w, req)
|
||||||
|
}
|
||||||
@@ -1,39 +0,0 @@
|
|||||||
package discord
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"io"
|
|
||||||
"net/http"
|
|
||||||
|
|
||||||
"golang.org/x/oauth2"
|
|
||||||
)
|
|
||||||
|
|
||||||
type ChannelInfo struct{}
|
|
||||||
|
|
||||||
type DiscordClient struct {
|
|
||||||
client *http.Client
|
|
||||||
defaultChannel string
|
|
||||||
baseUrl string
|
|
||||||
apiVersion string
|
|
||||||
}
|
|
||||||
|
|
||||||
func NewClient(ctx context.Context, tokenSrc oauth2.TokenSource) DiscordClient {
|
|
||||||
return DiscordClient{
|
|
||||||
client: oauth2.NewClient(ctx, tokenSrc),
|
|
||||||
baseUrl: "https://discord.com",
|
|
||||||
apiVersion: "v10",
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (dc DiscordClient) WithDefaultChannel(channelID string) DiscordClient {
|
|
||||||
dc.defaultChannel = channelID
|
|
||||||
return dc
|
|
||||||
}
|
|
||||||
|
|
||||||
func (dc DiscordClient) GetChannel() ChannelInfo {
|
|
||||||
return ChannelInfo{}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (dc DiscordClient) doRequest(path string, method string, body io.Reader) (*http.Response, error) {
|
|
||||||
return dc.client.Do(method, path, body)
|
|
||||||
}
|
|
||||||
@@ -1,10 +0,0 @@
|
|||||||
package registrar
|
|
||||||
|
|
||||||
import (
|
|
||||||
persist "go.fixergrid.net/servicedemon/pkg/registrar/internal/persistence"
|
|
||||||
svc "go.fixergrid.net/servicedemon/pkg/registrar/internal/services"
|
|
||||||
)
|
|
||||||
|
|
||||||
func NewRepo() svc.AppRepo {
|
|
||||||
return persist.NewFakeRepo()
|
|
||||||
}
|
|
||||||
@@ -4,9 +4,11 @@ import (
|
|||||||
"context"
|
"context"
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"log"
|
||||||
|
"os"
|
||||||
|
|
||||||
"github.com/google/uuid"
|
"github.com/google/uuid"
|
||||||
svc "go.fixergrid.net/servicedemon/pkg/registrar/internal/services"
|
svc "go.fixergrid.net/servicedemon/registrar/internal/services"
|
||||||
)
|
)
|
||||||
|
|
||||||
const APPROVAL_TOPIC = "net.fixergrid.events.app.approved"
|
const APPROVAL_TOPIC = "net.fixergrid.events.app.approved"
|
||||||
@@ -15,7 +17,7 @@ type ApprovalListener struct {
|
|||||||
svc.ApprovalRequester
|
svc.ApprovalRequester
|
||||||
svc.PubSub
|
svc.PubSub
|
||||||
svc.AppRepo
|
svc.AppRepo
|
||||||
log internalLog
|
log ilogger
|
||||||
}
|
}
|
||||||
|
|
||||||
type option func(r *ApprovalListener)
|
type option func(r *ApprovalListener)
|
||||||
@@ -31,6 +33,7 @@ func NewApprovalListener(ps svc.PubSub, requester svc.ApprovalRequester, repo sv
|
|||||||
PubSub: ps,
|
PubSub: ps,
|
||||||
ApprovalRequester: requester,
|
ApprovalRequester: requester,
|
||||||
AppRepo: repo,
|
AppRepo: repo,
|
||||||
|
log: log.New(os.Stderr, "servicedemon/registrar - ", log.LstdFlags|log.Lshortfile),
|
||||||
}
|
}
|
||||||
|
|
||||||
for _, opt := range options {
|
for _, opt := range options {
|
||||||
@@ -3,10 +3,12 @@ package registrar
|
|||||||
import (
|
import (
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"log"
|
||||||
"net/http"
|
"net/http"
|
||||||
|
"os"
|
||||||
|
|
||||||
"github.com/google/uuid"
|
"github.com/google/uuid"
|
||||||
svc "go.fixergrid.net/servicedemon/pkg/registrar/internal/services"
|
svc "go.fixergrid.net/servicedemon/registrar/internal/services"
|
||||||
)
|
)
|
||||||
|
|
||||||
const (
|
const (
|
||||||
@@ -18,7 +20,7 @@ var validApprover string = "Nautilus Admins"
|
|||||||
type Registrar struct {
|
type Registrar struct {
|
||||||
svc.Publisher
|
svc.Publisher
|
||||||
repo svc.AppRepo
|
repo svc.AppRepo
|
||||||
log internalLog
|
log ilogger
|
||||||
}
|
}
|
||||||
|
|
||||||
func WithLogger(l logger) RegistrarOpt {
|
func WithLogger(l logger) RegistrarOpt {
|
||||||
@@ -33,6 +35,7 @@ func NewRegistrar(publisher svc.Publisher, repo svc.AppRepo, options ...Registra
|
|||||||
r := Registrar{
|
r := Registrar{
|
||||||
Publisher: publisher,
|
Publisher: publisher,
|
||||||
repo: repo,
|
repo: repo,
|
||||||
|
log: log.New(os.Stderr, "servicedemon/registrar - ", log.LstdFlags|log.Lshortfile),
|
||||||
}
|
}
|
||||||
for _, opt := range options {
|
for _, opt := range options {
|
||||||
opt(&r)
|
opt(&r)
|
||||||
@@ -68,9 +71,6 @@ func (r Registrar) HandleRegistration(resp http.ResponseWriter, req *http.Reques
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (r Registrar) HandleApproval(resp http.ResponseWriter, req *http.Request) {
|
func (r Registrar) HandleApproval(resp http.ResponseWriter, req *http.Request) {
|
||||||
headers := resp.Header()
|
|
||||||
headers.Set("content-type", "application/json")
|
|
||||||
|
|
||||||
in := req.URL.Path
|
in := req.URL.Path
|
||||||
cert := req.TLS.PeerCertificates[0]
|
cert := req.TLS.PeerCertificates[0]
|
||||||
OUs := cert.Subject.OrganizationalUnit
|
OUs := cert.Subject.OrganizationalUnit
|
||||||
@@ -10,71 +10,11 @@ import (
|
|||||||
"testing"
|
"testing"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/google/uuid"
|
|
||||||
"github.com/stretchr/testify/assert"
|
"github.com/stretchr/testify/assert"
|
||||||
"github.com/stretchr/testify/require"
|
"github.com/stretchr/testify/require"
|
||||||
"go.fixergrid.net/servicedemon/pkg/pubsub"
|
"go.fixergrid.net/servicedemon/pubsub"
|
||||||
)
|
)
|
||||||
|
|
||||||
type TestRequest http.Request
|
|
||||||
|
|
||||||
func NewTestRequest(t *testing.T, method string, url string, body io.Reader) TestRequest {
|
|
||||||
req, err := http.NewRequest(method, url, body)
|
|
||||||
require.NoError(t, err)
|
|
||||||
return TestRequest(*req)
|
|
||||||
}
|
|
||||||
func (tr TestRequest) WithFakeTLSState(dnsName string) TestRequest {
|
|
||||||
cert := x509.Certificate{
|
|
||||||
DNSNames: []string{
|
|
||||||
dnsName,
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
tr.TLS = &tls.ConnectionState{
|
|
||||||
PeerCertificates: []*x509.Certificate{&cert},
|
|
||||||
}
|
|
||||||
|
|
||||||
return tr
|
|
||||||
}
|
|
||||||
|
|
||||||
func (tr TestRequest) Create() *http.Request {
|
|
||||||
req := http.Request(tr)
|
|
||||||
return &req
|
|
||||||
}
|
|
||||||
|
|
||||||
type AppState struct {
|
|
||||||
Name string
|
|
||||||
State string
|
|
||||||
}
|
|
||||||
type TestAppRepo []AppState
|
|
||||||
|
|
||||||
func (repo TestAppRepo) IsRegistered(name string) bool {
|
|
||||||
for _, app := range repo {
|
|
||||||
if name == app.Name && app.State == "registered" {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
func (repo TestAppRepo) PendingApprovalCount() int {
|
|
||||||
count := 0
|
|
||||||
for _, app := range repo {
|
|
||||||
if app.State == "pending_approval" {
|
|
||||||
count += 1
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return count
|
|
||||||
}
|
|
||||||
|
|
||||||
func (repo *TestAppRepo) StartAppRegistration(name string) uuid.UUID {
|
|
||||||
*repo = append(*repo, AppState{
|
|
||||||
Name: name,
|
|
||||||
State: "pending_approval",
|
|
||||||
})
|
|
||||||
return uuid.New()
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestHandleRegisterPendingApproval(t *testing.T) {
|
func TestHandleRegisterPendingApproval(t *testing.T) {
|
||||||
resp := httptest.NewRecorder()
|
resp := httptest.NewRecorder()
|
||||||
|
|
||||||
@@ -82,7 +22,7 @@ func TestHandleRegisterPendingApproval(t *testing.T) {
|
|||||||
WithFakeTLSState("dev-app-1.delivery.engineering").
|
WithFakeTLSState("dev-app-1.delivery.engineering").
|
||||||
Create()
|
Create()
|
||||||
|
|
||||||
repo := &TestAppRepo{}
|
repo := NewRepo()
|
||||||
pubsub := pubsub.New()
|
pubsub := pubsub.New()
|
||||||
subscriber := pubsub.Subscribe(REGISTRATION_TOPIC)
|
subscriber := pubsub.Subscribe(REGISTRATION_TOPIC)
|
||||||
|
|
||||||
@@ -112,12 +52,9 @@ func TestHandleRegisterAlreadyRegistered(t *testing.T) {
|
|||||||
Create()
|
Create()
|
||||||
|
|
||||||
pubsub := pubsub.New()
|
pubsub := pubsub.New()
|
||||||
repo := &TestAppRepo{
|
repo := NewRepo()
|
||||||
AppState{
|
appID := repo.StartAppRegistration("dev-app-1.delivery.engineering")
|
||||||
Name: "dev-app-1.delivery.engineering",
|
repo.ApproveApp(appID)
|
||||||
State: "registered",
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
NewRegistrar(pubsub, repo).HandleRegistration(resp, req)
|
NewRegistrar(pubsub, repo).HandleRegistration(resp, req)
|
||||||
|
|
||||||
@@ -129,6 +66,27 @@ func TestHandleRegisterAlreadyRegistered(t *testing.T) {
|
|||||||
assert.Equal(t, []byte(`{"status": "registered"}`), body)
|
assert.Equal(t, []byte(`{"status": "registered"}`), body)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestHandleRegisterNotYetApproved(t *testing.T) {
|
||||||
|
resp := httptest.NewRecorder()
|
||||||
|
|
||||||
|
req := NewTestRequest(t, http.MethodPost, "http://example.com/v1/register", nil).
|
||||||
|
WithFakeTLSState("dev-app-1.delivery.engineering").
|
||||||
|
Create()
|
||||||
|
|
||||||
|
pubsub := pubsub.New()
|
||||||
|
repo := NewRepo()
|
||||||
|
repo.StartAppRegistration("dev-app-1.delivery.engineering")
|
||||||
|
|
||||||
|
NewRegistrar(pubsub, repo).HandleRegistration(resp, req)
|
||||||
|
|
||||||
|
assert.Equal(t, http.StatusOK, resp.Code)
|
||||||
|
|
||||||
|
body, err := io.ReadAll(resp.Body)
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
assert.Equal(t, []byte(`{"status": "pending_approval"}`), body)
|
||||||
|
}
|
||||||
|
|
||||||
func assertMessage(t *testing.T, expected map[string]string, sender <-chan string) {
|
func assertMessage(t *testing.T, expected map[string]string, sender <-chan string) {
|
||||||
timer := time.After(2 * time.Second)
|
timer := time.After(2 * time.Second)
|
||||||
|
|
||||||
@@ -153,3 +111,29 @@ func compareMessage(t *testing.T, expected map[string]string, observed string) {
|
|||||||
assert.Equal(t, v, observedEvent[k])
|
assert.Equal(t, v, observedEvent[k])
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type TestRequest http.Request
|
||||||
|
|
||||||
|
func NewTestRequest(t *testing.T, method string, url string, body io.Reader) TestRequest {
|
||||||
|
req, err := http.NewRequest(method, url, body)
|
||||||
|
require.NoError(t, err)
|
||||||
|
return TestRequest(*req)
|
||||||
|
}
|
||||||
|
func (tr TestRequest) WithFakeTLSState(dnsName string) TestRequest {
|
||||||
|
cert := x509.Certificate{
|
||||||
|
DNSNames: []string{
|
||||||
|
dnsName,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
tr.TLS = &tls.ConnectionState{
|
||||||
|
PeerCertificates: []*x509.Certificate{&cert},
|
||||||
|
}
|
||||||
|
|
||||||
|
return tr
|
||||||
|
}
|
||||||
|
|
||||||
|
func (tr TestRequest) Create() *http.Request {
|
||||||
|
req := http.Request(tr)
|
||||||
|
return &req
|
||||||
|
}
|
||||||
@@ -2,7 +2,7 @@ package persistence
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/google/uuid"
|
"github.com/google/uuid"
|
||||||
svc "go.fixergrid.net/servicedemon/pkg/registrar/internal/services"
|
svc "go.fixergrid.net/servicedemon/registrar/internal/services"
|
||||||
)
|
)
|
||||||
|
|
||||||
type FakeRepo struct {
|
type FakeRepo struct {
|
||||||
10
registrar/repo.go
Normal file
10
registrar/repo.go
Normal file
@@ -0,0 +1,10 @@
|
|||||||
|
package registrar
|
||||||
|
|
||||||
|
import (
|
||||||
|
persist "go.fixergrid.net/servicedemon/registrar/internal/persistence"
|
||||||
|
svc "go.fixergrid.net/servicedemon/registrar/internal/services"
|
||||||
|
)
|
||||||
|
|
||||||
|
func NewRepo() svc.AppRepo {
|
||||||
|
return persist.NewFakeRepo()
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user