Compare commits
4 commits
c5ad01daee
...
0442fc1e2e
Author | SHA1 | Date | |
---|---|---|---|
0442fc1e2e | |||
3adfce18cb | |||
6e57d6818b | |||
86448ac4a7 |
18 changed files with 40 additions and 23 deletions
2
.gitignore
vendored
2
.gitignore
vendored
|
@ -5,4 +5,4 @@
|
|||
!*.md
|
||||
!*.mod
|
||||
!*.sum
|
||||
!*.txt
|
||||
!*.txt
|
|
@ -81,4 +81,4 @@ func (ctx *context) Status(status int) Context {
|
|||
func (ctx *context) String(body string) error {
|
||||
ctx.response.body = append(ctx.response.body, body...)
|
||||
return nil
|
||||
}
|
||||
}
|
|
@ -66,4 +66,4 @@ func TestRedirect(t *testing.T) {
|
|||
response := s.Request("GET", "/", nil, nil)
|
||||
assert.Equal(t, response.Status(), 301)
|
||||
assert.Equal(t, response.Header("Location"), "/target")
|
||||
}
|
||||
}
|
|
@ -1,4 +1,4 @@
|
|||
package web
|
||||
|
||||
// Handler is a function that deals with the given request/response context.
|
||||
type Handler func(Context) error
|
||||
type Handler func(Context) error
|
|
@ -4,4 +4,4 @@ package web
|
|||
type Header struct {
|
||||
Key string
|
||||
Value string
|
||||
}
|
||||
}
|
|
@ -79,4 +79,4 @@ func (req *request) addParameter(key string, value string) {
|
|||
Key: key,
|
||||
Value: value,
|
||||
})
|
||||
}
|
||||
}
|
|
@ -51,4 +51,4 @@ func TestRequestParam(t *testing.T) {
|
|||
response := s.Request("GET", "/blog/my-article", nil, nil)
|
||||
assert.Equal(t, response.Status(), 200)
|
||||
assert.Equal(t, string(response.Body()), "my-article")
|
||||
}
|
||||
}
|
|
@ -76,4 +76,4 @@ func (res *response) Write(body []byte) (int, error) {
|
|||
func (res *response) WriteString(body string) (int, error) {
|
||||
res.body = append(res.body, body...)
|
||||
return len(body), nil
|
||||
}
|
||||
}
|
|
@ -97,4 +97,4 @@ func TestResponseHeaderOverwrite(t *testing.T) {
|
|||
assert.Equal(t, response.Status(), 200)
|
||||
assert.Equal(t, response.Header("Content-Type"), "text/html")
|
||||
assert.Equal(t, string(response.Body()), "")
|
||||
}
|
||||
}
|
11
Server.go
11
Server.go
|
@ -19,6 +19,7 @@ import (
|
|||
// Server is the interface for an HTTP server.
|
||||
type Server interface {
|
||||
Get(path string, handler Handler)
|
||||
Ready() chan struct{}
|
||||
Request(method string, path string, headers []Header, body io.Reader) Response
|
||||
Router() *router.Router[Handler]
|
||||
Run(address string) error
|
||||
|
@ -31,6 +32,7 @@ type server struct {
|
|||
contextPool sync.Pool
|
||||
router *router.Router[Handler]
|
||||
errorHandler func(Context, error)
|
||||
ready chan struct{}
|
||||
}
|
||||
|
||||
// NewServer creates a new HTTP server.
|
||||
|
@ -54,6 +56,7 @@ func NewServer() Server {
|
|||
errorHandler: func(ctx Context, err error) {
|
||||
log.Println(ctx.Request().Path(), err)
|
||||
},
|
||||
ready: make(chan struct{}),
|
||||
}
|
||||
|
||||
s.contextPool.New = func() any { return s.newContext() }
|
||||
|
@ -65,6 +68,11 @@ func (s *server) Get(path string, handler Handler) {
|
|||
s.Router().Add("GET", path, handler)
|
||||
}
|
||||
|
||||
// Ready returns a channel that will be closed once the listener is ready for connection handling.
|
||||
func (s *server) Ready() chan struct{} {
|
||||
return s.ready
|
||||
}
|
||||
|
||||
// Request performs a synthetic request and returns the response.
|
||||
// This function keeps the response in memory so it's slightly slower than a real request.
|
||||
// However it is very useful inside tests where you don't want to spin up a real web server.
|
||||
|
@ -99,6 +107,7 @@ func (s *server) Run(address string) error {
|
|||
|
||||
stop := make(chan os.Signal, 1)
|
||||
signal.Notify(stop, os.Interrupt, syscall.SIGTERM)
|
||||
close(s.ready)
|
||||
<-stop
|
||||
return nil
|
||||
}
|
||||
|
@ -261,4 +270,4 @@ func (s *server) newContext() *context {
|
|||
status: 200,
|
||||
},
|
||||
}
|
||||
}
|
||||
}
|
|
@ -35,6 +35,7 @@ func TestRun(t *testing.T) {
|
|||
go func() {
|
||||
defer syscall.Kill(syscall.Getpid(), syscall.SIGTERM)
|
||||
|
||||
<-s.Ready()
|
||||
_, err := http.Get("http://127.0.0.1:8080/")
|
||||
assert.Nil(t, err)
|
||||
}()
|
||||
|
@ -48,6 +49,7 @@ func TestBadRequest(t *testing.T) {
|
|||
go func() {
|
||||
defer syscall.Kill(syscall.Getpid(), syscall.SIGTERM)
|
||||
|
||||
<-s.Ready()
|
||||
conn, err := net.Dial("tcp", ":8080")
|
||||
assert.Nil(t, err)
|
||||
defer conn.Close()
|
||||
|
@ -73,6 +75,7 @@ func TestBadRequestHeader(t *testing.T) {
|
|||
go func() {
|
||||
defer syscall.Kill(syscall.Getpid(), syscall.SIGTERM)
|
||||
|
||||
<-s.Ready()
|
||||
conn, err := net.Dial("tcp", ":8080")
|
||||
assert.Nil(t, err)
|
||||
defer conn.Close()
|
||||
|
@ -95,6 +98,7 @@ func TestBadRequestMethod(t *testing.T) {
|
|||
go func() {
|
||||
defer syscall.Kill(syscall.Getpid(), syscall.SIGTERM)
|
||||
|
||||
<-s.Ready()
|
||||
conn, err := net.Dial("tcp", ":8080")
|
||||
assert.Nil(t, err)
|
||||
defer conn.Close()
|
||||
|
@ -116,6 +120,7 @@ func TestBadRequestPath(t *testing.T) {
|
|||
go func() {
|
||||
defer syscall.Kill(syscall.Getpid(), syscall.SIGTERM)
|
||||
|
||||
<-s.Ready()
|
||||
conn, err := net.Dial("tcp", ":8080")
|
||||
assert.Nil(t, err)
|
||||
defer conn.Close()
|
||||
|
@ -141,6 +146,7 @@ func TestBadRequestProtocol(t *testing.T) {
|
|||
go func() {
|
||||
defer syscall.Kill(syscall.Getpid(), syscall.SIGTERM)
|
||||
|
||||
<-s.Ready()
|
||||
conn, err := net.Dial("tcp", ":8080")
|
||||
assert.Nil(t, err)
|
||||
defer conn.Close()
|
||||
|
@ -163,6 +169,7 @@ func TestConnectionClose(t *testing.T) {
|
|||
go func() {
|
||||
defer syscall.Kill(syscall.Getpid(), syscall.SIGTERM)
|
||||
|
||||
<-s.Ready()
|
||||
conn, err := net.Dial("tcp", ":8080")
|
||||
assert.Nil(t, err)
|
||||
defer conn.Close()
|
||||
|
@ -183,6 +190,7 @@ func TestEarlyClose(t *testing.T) {
|
|||
go func() {
|
||||
defer syscall.Kill(syscall.Getpid(), syscall.SIGTERM)
|
||||
|
||||
<-s.Ready()
|
||||
conn, err := net.Dial("tcp", ":8080")
|
||||
assert.Nil(t, err)
|
||||
|
||||
|
@ -203,4 +211,4 @@ func TestUnavailablePort(t *testing.T) {
|
|||
|
||||
s := web.NewServer()
|
||||
s.Run(":8080")
|
||||
}
|
||||
}
|
|
@ -12,4 +12,4 @@ func main() {
|
|||
})
|
||||
|
||||
s.Run(":8080")
|
||||
}
|
||||
}
|
6
go.mod
6
go.mod
|
@ -3,6 +3,6 @@ module git.urbach.dev/go/web
|
|||
go 1.24
|
||||
|
||||
require (
|
||||
git.urbach.dev/go/assert v0.0.0-20250225153414-fc1f84f19edf
|
||||
git.urbach.dev/go/router v0.0.0-20250601162231-e35d5715d1a5
|
||||
)
|
||||
git.urbach.dev/go/assert v0.0.0-20250606150337-559d3d3afcda
|
||||
git.urbach.dev/go/router v0.0.0-20250606151301-c10e38fec918
|
||||
)
|
8
go.sum
8
go.sum
|
@ -1,4 +1,4 @@
|
|||
git.urbach.dev/go/assert v0.0.0-20250225153414-fc1f84f19edf h1:BQWa5GKNUsA5CSUa/+UlFWYCEVe3IDDKRbVqBLK0mAE=
|
||||
git.urbach.dev/go/assert v0.0.0-20250225153414-fc1f84f19edf/go.mod h1:y9jGII9JFiF1HNIju0u87OyPCt82xKCtqnAFyEreCDo=
|
||||
git.urbach.dev/go/router v0.0.0-20250601162231-e35d5715d1a5 h1:3qlZjgWbrHw4LWM4uBzOTldbpqCLJdeSgvcYK6f3xpc=
|
||||
git.urbach.dev/go/router v0.0.0-20250601162231-e35d5715d1a5/go.mod h1:O+doTe0DZdT2XMsTY5pe6qUMqEEIAbwtX6Xp6EqHw34=
|
||||
git.urbach.dev/go/assert v0.0.0-20250606150337-559d3d3afcda h1:VN6ZQwtwLOm2xTms+v8IIeeNjvs55qyEBNArv3dPq9g=
|
||||
git.urbach.dev/go/assert v0.0.0-20250606150337-559d3d3afcda/go.mod h1:PNI/NSBOqvoeU58/7eBsIR09Yoq2S/qtSRiTrctkiq0=
|
||||
git.urbach.dev/go/router v0.0.0-20250606151301-c10e38fec918 h1:fdeFr4gwou9KxvQqhcrPbLSYM9xzvaF1pwp53DzU3co=
|
||||
git.urbach.dev/go/router v0.0.0-20250606151301-c10e38fec918/go.mod h1:seUQ5raGaj6fDeZP6d7JdgnWQys8oTrtFdvBhAp1IZA=
|
2
http.go
2
http.go
|
@ -38,4 +38,4 @@ func parseURL(url string) (scheme string, host string, path string, query string
|
|||
|
||||
path = url
|
||||
return
|
||||
}
|
||||
}
|
|
@ -87,4 +87,4 @@ Please see the [license documentation](https://urbach.dev/license).
|
|||
|
||||
## Copyright
|
||||
|
||||
© 2024 Eduard Urbach
|
||||
© 2024 Eduard Urbach
|
|
@ -46,4 +46,4 @@ func Text(ctx web.Context, body string) error {
|
|||
func XML(ctx web.Context, body string) error {
|
||||
ctx.Response().SetHeader("Content-Type", "text/xml")
|
||||
return ctx.String(body)
|
||||
}
|
||||
}
|
|
@ -64,4 +64,4 @@ func TestContentTypes(t *testing.T) {
|
|||
assert.Equal(t, string(response.Body()), test.Response)
|
||||
})
|
||||
}
|
||||
}
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue