Merge pull request #53 from tetratelabs/http-framework

http filter framework

Author: mathetake

README.md

@@ -52,7 +52,7 @@ The target Envoy version is `release/v1.15`
 
 ## setup
 
-- __git clone__ this repository to `${GOPATH}/github.com/tetratelabs/proxy-wasm-sdk-go`
+- __git clone__ this repository to `${GOPATH}/github.com/tetratelabs/proxy-wasm-go-sdk`
     - `go get` may fail because some functions do not have the function body.
 - For IDE and editors, please set `-tags=proxytest` build tag for correct completion and task runners.
 

examples/http_auth_random/main.go

@@ -22,6 +22,8 @@ import (
 	"github.com/tetratelabs/proxy-wasm-go-sdk/proxywasm/types"
 )
 
+const clusterName = "httpbin"
+
 func main() {
 	proxywasm.SetNewHttpContext(newContext)
 }
@@ -48,15 +50,17 @@ func (ctx *httpHeaders) OnHttpRequestHeaders(int, bool) types.Action {
 	}
 
 	if _, err := proxywasm.HostCallDispatchHttpCall(
-		"httpbin", hs, "", [][2]string{}, 50000); err != nil {
+		clusterName, hs, "", [][2]string{}, 50000); err != nil {
 		proxywasm.LogCritical("dipatch httpcall failed: ", err.Error())
 	}
 
+	proxywasm.LogInfo("http call dispatched to ", clusterName)
+
 	return types.ActionPause
 }
 
 // override default
-func (ctx *httpHeaders) OnHttpCallResponse(_ uint32, _ int, bodySize int, _ int) {
+func (ctx *httpHeaders) OnHttpCallResponse(_ int, bodySize int, _ int) {
 	hs, err := proxywasm.HostCallGetHttpCallResponseHeaders()
 	if err != nil {
 		proxywasm.LogCritical("failed to get response body: ", err.Error())

examples/http_auth_random/main_test.go

@@ -0,0 +1,67 @@
+package main
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+
+	"github.com/tetratelabs/proxy-wasm-go-sdk/proxytest"
+	"github.com/tetratelabs/proxy-wasm-go-sdk/proxywasm/types"
+)
+
+func TestHttpHeaders_OnHttpRequestHeaders(t *testing.T) {
+	host, done := proxytest.NewHttpFilterHost(newContext)
+	defer done()
+
+	id := host.InitContext()
+	host.PutRequestHeaders(id, [][2]string{{"key", "value"}}) // OnHttpRequestHeaders called
+
+	require.True(t, host.IsDispatchCalled(id))                     // check if http call is dispatched
+	require.Equal(t, types.ActionPause, host.GetCurrentAction(id)) // check if the current action is pause
+
+	logs := host.GetLogs(types.LogLevelInfo)
+	require.GreaterOrEqual(t, len(logs), 2)
+
+	assert.Equal(t, "http call dispatched to "+clusterName, logs[len(logs)-1])
+	assert.Equal(t, "request header: key: value", logs[len(logs)-2])
+}
+
+func TestHttpHeaders_OnHttpCallResponse(t *testing.T) {
+	host, done := proxytest.NewHttpFilterHost(newContext)
+	defer done()
+
+	// http://httpbin.org/uuid
+	headers := [][2]string{
+		{"HTTP/1.1", "200 OK"}, {"Date:", "Thu, 17 Sep 2020 02:47:07 GMT"},
+		{"Content-Type", "application/json"}, {"Content-Length", "53"},
+		{"Connection", "keep-alive"}, {"Server", "gunicorn/19.9.0"},
+		{"Access-Control-Allow-Origin", "*"}, {"Access-Control-Allow-Credentials", "true"},
+	}
+
+	// access granted body
+	id := host.InitContext()
+	body := []byte(`{"uuid": "7b10a67a-1c67-4199-835b-cbefcd4a63d4"}`)
+	host.PutCalloutResponse(id, headers, nil, body)
+	assert.Nil(t, host.GetSentLocalResponse(id))
+
+	logs := host.GetLogs(types.LogLevelInfo)
+	require.Greater(t, len(logs), 1)
+	assert.Equal(t, "access granted", logs[len(logs)-1])
+
+	// access denied body
+	id = host.InitContext()
+	body = []byte(`{"uuid": "aaaaaaaa-1c67-4199-835b-cbefcd4a63d4"}`)
+	host.PutCalloutResponse(id, headers, nil, body)
+	assert.NotNil(t, host.GetSentLocalResponse(id))
+	logs = host.GetLogs(types.LogLevelInfo)
+	assert.Equal(t, "access forbidden", logs[len(logs)-1])
+
+	localResponse := host.GetSentLocalResponse(id) // check local responses
+	require.NotNil(t, localResponse)
+	assert.Equal(t, uint32(403), localResponse.StatusCode)
+	assert.Equal(t, []byte("access forbidden"), localResponse.Data)
+	require.Len(t, localResponse.Headers, 1)
+	assert.Equal(t, "powered-by", localResponse.Headers[0][0])
+	assert.Equal(t, "proxy-wasm-go-sdk!!", localResponse.Headers[0][1])
+}

examples/http_headers/main_test.go

@@ -0,0 +1,41 @@
+package main
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+
+	"github.com/tetratelabs/proxy-wasm-go-sdk/proxytest"
+	"github.com/tetratelabs/proxy-wasm-go-sdk/proxywasm/types"
+)
+
+func TestHttpHeaders_OnHttpRequestHeaders(t *testing.T) {
+	host, done := proxytest.NewHttpFilterHost(newContext)
+	defer done()
+	id := host.InitContext()
+
+	hs := [][2]string{{"key1", "value1"}, {"key2", "value2"}}
+	host.PutRequestHeaders(id, hs) // call OnHttpRequestHeaders
+
+	logs := host.GetLogs(types.LogLevelInfo)
+	require.Greater(t, len(logs), 1)
+
+	assert.Equal(t, "request header: key2: value2", logs[len(logs)-1])
+	assert.Equal(t, "request header: key1: value1", logs[len(logs)-2])
+}
+
+func TestHttpHeaders_OnHttpResponseHeaders(t *testing.T) {
+	host, done := proxytest.NewHttpFilterHost(newContext)
+	defer done()
+	id := host.InitContext()
+
+	hs := [][2]string{{"key1", "value1"}, {"key2", "value2"}}
+	host.PutResponseHeaders(id, hs) // call OnHttpResponseHeaders
+
+	logs := host.GetLogs(types.LogLevelInfo)
+	require.Greater(t, len(logs), 1)
+
+	assert.Equal(t, "response header: key2: value2", logs[len(logs)-1])
+	assert.Equal(t, "response header: key1: value1", logs[len(logs)-2])
+}

examples/metrics/main_test.go

@@ -5,6 +5,7 @@ import (
 
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
+
 	"github.com/tetratelabs/proxy-wasm-go-sdk/proxytest"
 	"github.com/tetratelabs/proxy-wasm-go-sdk/proxywasm/types"
 )

examples/network/main.go

@@ -35,6 +35,8 @@ func (ctx network) OnNewConnection() types.Action {
 }
 
 func (ctx network) OnDownstreamData(dataSize int, _ bool) types.Action {
+	// TODO: dispatch http call
+
 	if dataSize == 0 {
 		return types.ActionContinue
 	}
@@ -54,6 +56,7 @@ func (ctx network) OnDownstreamClose(types.PeerType) {
 }
 
 func (ctx network) OnUpstreamData(dataSize int, _ bool) types.Action {
+
 	if dataSize == 0 {
 		return types.ActionContinue
 	}