mirror of
https://github.com/dutchcoders/transfer.sh.git
synced 2024-12-11 13:20:19 +01:00
cb6e5cb0c7
* use dep for vendoring * lets encrypt * moved web to transfer.sh-web repo * single command install * added first tests
52 lines
1.4 KiB
Go
52 lines
1.4 KiB
Go
package handlers
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"testing"
|
|
)
|
|
|
|
type baseWriter struct{}
|
|
|
|
func (b *baseWriter) Write(data []byte) (int, error) { return 0, nil }
|
|
func (b *baseWriter) WriteHeader(code int) {}
|
|
func (b *baseWriter) Header() http.Header { return nil }
|
|
|
|
func TestNilWriter(t *testing.T) {
|
|
rw, ok := GetResponseWriter(nil, func(w http.ResponseWriter) bool {
|
|
return true
|
|
})
|
|
assertTrue(rw == nil, "expected nil, got non-nil", t)
|
|
assertTrue(!ok, "expected false, got true", t)
|
|
}
|
|
|
|
func TestBaseWriter(t *testing.T) {
|
|
bw := &baseWriter{}
|
|
rw, ok := GetResponseWriter(bw, func(w http.ResponseWriter) bool {
|
|
return true
|
|
})
|
|
assertTrue(rw == bw, fmt.Sprintf("expected %#v, got %#v", bw, rw), t)
|
|
assertTrue(ok, "expected true, got false", t)
|
|
}
|
|
|
|
func TestWrappedWriter(t *testing.T) {
|
|
bw := &baseWriter{}
|
|
ctx := &contextResponseWriter{bw, nil}
|
|
rw, ok := GetResponseWriter(ctx, func(w http.ResponseWriter) bool {
|
|
_, ok := w.(*baseWriter)
|
|
return ok
|
|
})
|
|
assertTrue(rw == bw, fmt.Sprintf("expected %#v, got %#v", bw, rw), t)
|
|
assertTrue(ok, "expected true, got false", t)
|
|
}
|
|
|
|
func TestWrappedNotFoundWriter(t *testing.T) {
|
|
bw := &baseWriter{}
|
|
ctx := &contextResponseWriter{bw, nil}
|
|
rw, ok := GetResponseWriter(ctx, func(w http.ResponseWriter) bool {
|
|
_, ok := w.(*statusResponseWriter)
|
|
return ok
|
|
})
|
|
assertTrue(rw == nil, fmt.Sprintf("expected nil, got %#v", rw), t)
|
|
assertTrue(!ok, "expected false, got true", t)
|
|
}
|