goreplay/http_modifier_test.go

package main

import (
    "testing"
    "github.com/buger/gor/proto"
    "bytes"
)

func TestHTTPModifierWithoutConfig(t *testing.T) {
    if NewHTTPModifier(&HTTPModifierConfig{}) != nil {
        t.Error("If no config specified should not be initialized")
    }
}

func TestHTTPModifierHeaderFilters(t *testing.T) {
    filters := HTTPHeaderFilters{}
    filters.Set("Host:^www.w3.org$")

    modifier := NewHTTPModifier(&HTTPModifierConfig{
        headerFilters: filters,
    })

    payload := []byte("POST /post HTTP/1.1\r\nContent-Length: 7\r\nHost: www.w3.org\r\n\r\na=1&b=2")

    if len(modifier.Rewrite(payload)) == 0 {
        t.Error("Request should pass filters")
    }

    filters = HTTPHeaderFilters{}
    // Setting filter that not match our header
    filters.Set("Host:^www.w4.org$")

    modifier = NewHTTPModifier(&HTTPModifierConfig{
        headerFilters: filters,
    })

    if len(modifier.Rewrite(payload)) != 0 {
        t.Error("Request should not pass filters")
    }
}


func TestHTTPModifierURLRewrite(t *testing.T) {
    var url, new_url []byte

    rewrites := UrlRewriteMap{}

    payload := func(url []byte) []byte {
        return []byte("POST " + string(url) + " HTTP/1.1\r\nContent-Length: 7\r\nHost: www.w3.org\r\n\r\na=1&b=2")
    }

    err := rewrites.Set("/v1/user/([^\\/]+)/ping:/v2/user/$1/ping")
    if err != nil {
        t.Error("Should not error on /v1/user/([^\\/]+)/ping:/v2/user/$1/ping")
    }

    modifier := NewHTTPModifier(&HTTPModifierConfig{
        urlRewrite: rewrites,
    })

    url = []byte("/v1/user/joe/ping")
    if new_url = proto.Path(modifier.Rewrite(payload(url))); bytes.Equal(new_url, url) {
        t.Error("Request url should have been rewritten, wasn't", string(new_url))
    }

    url = []byte("/v1/user/ping")
    if new_url = proto.Path(modifier.Rewrite(payload(url))); !bytes.Equal(new_url, url) {
        t.Error("Request url should have been rewritten, wasn't", string(new_url))
    }
}

func TestHTTPModifierHeaderHashFilters(t *testing.T) {
    filters := HTTPHeaderHashFilters{}
    filters.Set("Header2:1/2")

    modifier := NewHTTPModifier(&HTTPModifierConfig{
        headerHashFilters: filters,
    })

    payload := func(header []byte) []byte {
        return []byte("POST / HTTP/1.1\r\n" + string(header) + "Content-Length: 7\r\nHost: www.w3.org\r\n\r\na=1&b=2")
    }

    if p := modifier.Rewrite(payload([]byte(""))); len(p) > 0 {
        t.Error("Request should not pass filters, Header2 does not exist")
    }

    if p := modifier.Rewrite(payload([]byte("Header2: 3\r\n"))); len(p) > 0 {
        t.Error("Request should not pass filters, Header2 hash too high")
    }

    if p := modifier.Rewrite(payload([]byte("Header2: 1\r\n"))); len(p) == 0 {
        t.Error("Request should pass filters")
    }
}

func TestHTTPModifierHeaders(t *testing.T) {
    headers := HTTPHeaders{}
    headers.Set("Header1:1")
    headers.Set("Host:localhost")

    modifier := NewHTTPModifier(&HTTPModifierConfig{
        headers: headers,
    })

    payload := []byte("POST /post HTTP/1.1\r\nContent-Length: 7\r\nHost: www.w3.org\r\n\r\na=1&b=2")
    new_payload := []byte("POST /post HTTP/1.1\r\nHeader1: 1\r\nContent-Length: 7\r\nHost: localhost\r\n\r\na=1&b=2")

    if payload = modifier.Rewrite(payload); !bytes.Equal(payload, new_payload) {
        t.Error("Should update request headers", string(payload))
    }
}