refactor handler out
parent
efb75f74cb
commit
a59857b549
|
|
@ -0,0 +1,30 @@
|
||||||
|
package handler
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"net/http"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Handler struct {
|
||||||
|
ctx context.Context
|
||||||
|
}
|
||||||
|
|
||||||
|
func New(ctx context.Context) Handler {
|
||||||
|
return Handler{ctx: ctx}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h Handler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
||||||
|
if err := h.serveHTTP(w, r); err != nil {
|
||||||
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h Handler) serveHTTP(w http.ResponseWriter, r *http.Request) error {
|
||||||
|
switch r.URL.Path {
|
||||||
|
case "/v1/vpntor":
|
||||||
|
return h.vpntor(r.Context(), r.Body)
|
||||||
|
default:
|
||||||
|
http.NotFound(w, r)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,21 @@
|
||||||
|
package handler_test
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"net/http"
|
||||||
|
"net/http/httptest"
|
||||||
|
"show-rss/src/cmd/server/handler"
|
||||||
|
"show-rss/src/db"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestHandler(t *testing.T) {
|
||||||
|
h := handler.New(db.Test(t, context.Background()))
|
||||||
|
|
||||||
|
w := httptest.NewRecorder()
|
||||||
|
r := httptest.NewRequest(http.MethodGet, "/not-found", nil)
|
||||||
|
h.ServeHTTP(w, r)
|
||||||
|
if w.Code != http.StatusNotFound {
|
||||||
|
t.Errorf("%+v", w)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -1,4 +1,4 @@
|
||||||
package server
|
package handler
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bytes"
|
"bytes"
|
||||||
|
|
@ -11,30 +11,6 @@ import (
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
type Handler struct {
|
|
||||||
ctx context.Context
|
|
||||||
}
|
|
||||||
|
|
||||||
func NewHandler(ctx context.Context) Handler {
|
|
||||||
return Handler{ctx: ctx}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (h Handler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
|
||||||
if err := h.serveHTTP(w, r); err != nil {
|
|
||||||
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (h Handler) serveHTTP(w http.ResponseWriter, r *http.Request) error {
|
|
||||||
switch r.URL.Path {
|
|
||||||
case "/v1/vpntor":
|
|
||||||
return h.vpntor(r.Context(), r.Body)
|
|
||||||
default:
|
|
||||||
http.NotFound(w, r)
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (h Handler) vpntor(ctx context.Context, r io.Reader) error {
|
func (h Handler) vpntor(ctx context.Context, r io.Reader) error {
|
||||||
var form struct {
|
var form struct {
|
||||||
Magnet string
|
Magnet string
|
||||||
|
|
@ -0,0 +1,66 @@
|
||||||
|
package handler_test
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"net/http"
|
||||||
|
"net/http/httptest"
|
||||||
|
"show-rss/src/cmd/server/handler"
|
||||||
|
"show-rss/src/db"
|
||||||
|
"strings"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestVpntor(t *testing.T) {
|
||||||
|
h := handler.New(db.Test(t, context.Background()))
|
||||||
|
|
||||||
|
t.Run("no body", func(t *testing.T) {
|
||||||
|
w := httptest.NewRecorder()
|
||||||
|
r := httptest.NewRequest(http.MethodGet, "/v1/vpntor", nil)
|
||||||
|
h.ServeHTTP(w, r)
|
||||||
|
if w.Code != http.StatusInternalServerError {
|
||||||
|
t.Errorf("%+v", w)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
t.Run("bad body", func(t *testing.T) {
|
||||||
|
w := httptest.NewRecorder()
|
||||||
|
r := httptest.NewRequest(http.MethodPost, "/v1/vpntor", strings.NewReader(`{}`))
|
||||||
|
h.ServeHTTP(w, r)
|
||||||
|
t.Logf("%s", w.Body.Bytes())
|
||||||
|
if w.Code != http.StatusInternalServerError {
|
||||||
|
t.Errorf("%+v", w.Code)
|
||||||
|
}
|
||||||
|
if !strings.Contains(string(w.Body.Bytes()), `.Magnet, .Dir, .URL`) {
|
||||||
|
t.Errorf("%+v", string(w.Body.Bytes()))
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
t.Run("doit", func(t *testing.T) {
|
||||||
|
calls := 0
|
||||||
|
s := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
b, _ := io.ReadAll(r.Body)
|
||||||
|
t.Logf("%s | %s", r.Header.Get("X-Transmission-Session-Id"), b)
|
||||||
|
calls += 1
|
||||||
|
w.Header().Set("X-Transmission-Session-Id", "session")
|
||||||
|
w.Write([]byte(`success`))
|
||||||
|
}))
|
||||||
|
defer s.Close()
|
||||||
|
|
||||||
|
w := httptest.NewRecorder()
|
||||||
|
r := httptest.NewRequest(http.MethodPost, "/v1/vpntor", strings.NewReader(fmt.Sprintf(`{
|
||||||
|
"Magnet": %q,
|
||||||
|
"Dir": %q,
|
||||||
|
"URL": %q
|
||||||
|
}`, "magnet", "dir", s.URL)))
|
||||||
|
h.ServeHTTP(w, r)
|
||||||
|
if w.Code != http.StatusOK {
|
||||||
|
t.Errorf("%+v", w.Code)
|
||||||
|
}
|
||||||
|
|
||||||
|
if calls != 2 {
|
||||||
|
t.Errorf("expected 2 but got %d calls", calls)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
@ -1,77 +0,0 @@
|
||||||
package server_test
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"fmt"
|
|
||||||
"io"
|
|
||||||
"net/http"
|
|
||||||
"net/http/httptest"
|
|
||||||
"show-rss/src/cmd/server"
|
|
||||||
"show-rss/src/db"
|
|
||||||
"strings"
|
|
||||||
"testing"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestHandler(t *testing.T) {
|
|
||||||
h := server.NewHandler(db.Test(t, context.Background()))
|
|
||||||
|
|
||||||
t.Run("404", func(t *testing.T) {
|
|
||||||
w := httptest.NewRecorder()
|
|
||||||
r := httptest.NewRequest(http.MethodGet, "/not-found", nil)
|
|
||||||
h.ServeHTTP(w, r)
|
|
||||||
if w.Code != http.StatusNotFound {
|
|
||||||
t.Errorf("%+v", w)
|
|
||||||
}
|
|
||||||
})
|
|
||||||
|
|
||||||
t.Run("vpntor", func(t *testing.T) {
|
|
||||||
t.Run("no body", func(t *testing.T) {
|
|
||||||
w := httptest.NewRecorder()
|
|
||||||
r := httptest.NewRequest(http.MethodGet, "/v1/vpntor", nil)
|
|
||||||
h.ServeHTTP(w, r)
|
|
||||||
if w.Code != http.StatusInternalServerError {
|
|
||||||
t.Errorf("%+v", w)
|
|
||||||
}
|
|
||||||
})
|
|
||||||
|
|
||||||
t.Run("bad body", func(t *testing.T) {
|
|
||||||
w := httptest.NewRecorder()
|
|
||||||
r := httptest.NewRequest(http.MethodPost, "/v1/vpntor", strings.NewReader(`{}`))
|
|
||||||
h.ServeHTTP(w, r)
|
|
||||||
t.Logf("%s", w.Body.Bytes())
|
|
||||||
if w.Code != http.StatusInternalServerError {
|
|
||||||
t.Errorf("%+v", w.Code)
|
|
||||||
}
|
|
||||||
if !strings.Contains(string(w.Body.Bytes()), `.Magnet, .Dir, .URL`) {
|
|
||||||
t.Errorf("%+v", string(w.Body.Bytes()))
|
|
||||||
}
|
|
||||||
})
|
|
||||||
|
|
||||||
t.Run("doit", func(t *testing.T) {
|
|
||||||
calls := 0
|
|
||||||
s := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
||||||
b, _ := io.ReadAll(r.Body)
|
|
||||||
t.Logf("%s | %s", r.Header.Get("X-Transmission-Session-Id"), b)
|
|
||||||
calls += 1
|
|
||||||
w.Header().Set("X-Transmission-Session-Id", "session")
|
|
||||||
w.Write([]byte(`success`))
|
|
||||||
}))
|
|
||||||
defer s.Close()
|
|
||||||
|
|
||||||
w := httptest.NewRecorder()
|
|
||||||
r := httptest.NewRequest(http.MethodPost, "/v1/vpntor", strings.NewReader(fmt.Sprintf(`{
|
|
||||||
"Magnet": %q,
|
|
||||||
"Dir": %q,
|
|
||||||
"URL": %q
|
|
||||||
}`, "magnet", "dir", s.URL)))
|
|
||||||
h.ServeHTTP(w, r)
|
|
||||||
if w.Code != http.StatusOK {
|
|
||||||
t.Errorf("%+v", w.Code)
|
|
||||||
}
|
|
||||||
|
|
||||||
if calls != 2 {
|
|
||||||
t.Errorf("expected 2 but got %d calls", calls)
|
|
||||||
}
|
|
||||||
})
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
@ -6,6 +6,7 @@ import (
|
||||||
"net"
|
"net"
|
||||||
"net/http"
|
"net/http"
|
||||||
"os"
|
"os"
|
||||||
|
"show-rss/src/cmd/server/handler"
|
||||||
"strconv"
|
"strconv"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
@ -24,7 +25,7 @@ func Run(ctx context.Context, listen string) error {
|
||||||
|
|
||||||
s := http.Server{
|
s := http.Server{
|
||||||
Addr: listen,
|
Addr: listen,
|
||||||
Handler: NewHandler(ctx),
|
Handler: handler.New(ctx),
|
||||||
BaseContext: func(net.Listener) context.Context { return ctx },
|
BaseContext: func(net.Listener) context.Context { return ctx },
|
||||||
}
|
}
|
||||||
defer s.Close()
|
defer s.Close()
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue