mirror of
https://github.com/buger/goreplay.git
synced 2024-04-21 12:32:02 +00:00
Urban Ishimwe
6d812ceb7f
// PluginReader is an interface for input plugins
type PluginReader interface {
PluginRead() (msg *Message, err error)
}
// PluginWriter is an interface for output plugins
type PluginWriter interface {
PluginWrite(msg *Message) (n int, err error)
}
38 lines
980 B
Go
38 lines
980 B
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"compress/gzip"
|
|
"strconv"
|
|
"testing"
|
|
|
|
"github.com/buger/goreplay/proto"
|
|
)
|
|
|
|
func TestHTTPPrettifierGzip(t *testing.T) {
|
|
b := bytes.NewBufferString("")
|
|
w := gzip.NewWriter(b)
|
|
w.Write([]byte("test"))
|
|
w.Close()
|
|
|
|
size := strconv.Itoa(len(b.Bytes()))
|
|
|
|
payload := []byte("HTTP/1.1 200 OK\r\nContent-Length: " + size + "\r\nContent-Encoding: gzip\r\n\r\n")
|
|
payload = append(payload, b.Bytes()...)
|
|
|
|
newPayload := prettifyHTTP(payload)
|
|
|
|
if string(newPayload) != "HTTP/1.1 200 OK\r\nContent-Length: 4\r\n\r\ntest" {
|
|
t.Errorf("Payload not match %q", string(newPayload))
|
|
}
|
|
}
|
|
|
|
func TestHTTPPrettifierChunked(t *testing.T) {
|
|
payload := []byte("POST / HTTP/1.1\r\nHost: www.w3.org\r\nTransfer-Encoding: chunked\r\n\r\n4\r\nWiki\r\n5\r\npedia\r\ne\r\n in\r\n\r\nchunks.\r\n0\r\n\r\n")
|
|
|
|
payload = prettifyHTTP(payload)
|
|
if string(proto.Header(payload, []byte("Content-Length"))) != "23" {
|
|
t.Errorf("payload should have content length of 23")
|
|
}
|
|
}
|