test case for http auth

pull/314/head
Darien Raymond 8 years ago
parent 80312627c4
commit bae0de7d95
No known key found for this signature in database
GPG Key ID: 7251FFA14BB18169

@ -41,6 +41,12 @@ type HeaderWriter struct {
header *alloc.Buffer
}
func NewHeaderWriter(header *alloc.Buffer) *HeaderWriter {
return &HeaderWriter{
header: header,
}
}
func (this *HeaderWriter) Write(writer io.Writer) error {
if this.header == nil {
return nil

@ -0,0 +1,23 @@
package http_test
import (
"testing"
"v2ray.com/core/common/alloc"
"v2ray.com/core/testing/assert"
. "v2ray.com/core/transport/internet/authenticators/http"
)
func TestReaderWriter(t *testing.T) {
assert := assert.On(t)
cache := alloc.NewBuffer()
writer := NewHeaderWriter(alloc.NewLocalBuffer(256).Clear().AppendString("abcd" + ENDING))
writer.Write(cache)
cache.Write([]byte{'e', 'f', 'g'})
reader := &HeaderReader{}
buffer, err := reader.Read(cache)
assert.Error(err).IsNil()
assert.Bytes(buffer.Value).Equals([]byte{'e', 'f', 'g'})
}
Loading…
Cancel
Save