Skip to content
This repository was archived by the owner on Apr 24, 2025. It is now read-only.

Commit dbbb010

Browse files
authored
feat: proxytest getters for request and response headers (#87)
1 parent 487bbd5 commit dbbb010

File tree

2 files changed

+22
-0
lines changed

2 files changed

+22
-0
lines changed

proxytest/http.go

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -313,6 +313,26 @@ func (h *httpHostEmulator) HttpFilterPutRequestHeaders(contextID uint32, headers
313313
h.HttpFilterPutRequestHeadersEndOfStream(contextID, headers, false)
314314
}
315315

316+
// impl HostEmulator
317+
func (h *httpHostEmulator) HttpFilterGetRequestHeaders(contextID uint32) (headers [][2]string) {
318+
cs, ok := h.httpStreams[contextID]
319+
if !ok {
320+
log.Fatalf("invalid context id: %d", contextID)
321+
}
322+
323+
return cs.requestHeaders
324+
}
325+
326+
// impl HostEmulator
327+
func (h *httpHostEmulator) HttpFilterGetResponseHeaders(contextID uint32) (headers [][2]string) {
328+
cs, ok := h.httpStreams[contextID]
329+
if !ok {
330+
log.Fatalf("invalid context id: %d", contextID)
331+
}
332+
333+
return cs.responseHeaders
334+
}
335+
316336
// impl HostEmulator
317337
func (h *httpHostEmulator) HttpFilterPutRequestHeadersEndOfStream(contextID uint32, headers [][2]string, endOfStream bool) {
318338
cs, ok := h.httpStreams[contextID]

proxytest/proxytest.go

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -37,8 +37,10 @@ type HostEmulator interface {
3737
// http
3838
HttpFilterInitContext() (contextID uint32)
3939
HttpFilterPutRequestHeaders(contextID uint32, headers [][2]string)
40+
HttpFilterGetRequestHeaders(contextID uint32) (headers [][2]string)
4041
HttpFilterPutRequestHeadersEndOfStream(contextID uint32, headers [][2]string, endOfStream bool)
4142
HttpFilterPutResponseHeaders(contextID uint32, headers [][2]string)
43+
HttpFilterGetResponseHeaders(contextID uint32) (headers [][2]string)
4244
HttpFilterPutResponseHeadersEndOfStream(contextID uint32, headers [][2]string, endOfStream bool)
4345
HttpFilterPutRequestTrailers(contextID uint32, headers [][2]string)
4446
HttpFilterPutResponseTrailers(contextID uint32, headers [][2]string)

0 commit comments

Comments
 (0)