updated hws to use new hlog and ezconf
This commit is contained in:
@@ -17,6 +17,10 @@ import (
|
||||
func Test_GZIP_Compression(t *testing.T) {
|
||||
var buf bytes.Buffer
|
||||
|
||||
dbg, _ := hlog.LogLevel("debug")
|
||||
logcfg := &hlog.Config{
|
||||
LogLevel: dbg,
|
||||
}
|
||||
t.Run("GZIP enabled compresses response", func(t *testing.T) {
|
||||
server, err := hws.NewServer(&hws.Config{
|
||||
Host: "127.0.0.1",
|
||||
@@ -25,7 +29,7 @@ func Test_GZIP_Compression(t *testing.T) {
|
||||
})
|
||||
require.NoError(t, err)
|
||||
|
||||
logger, err := hlog.NewLogger(hlog.LogLevel("Debug"), &buf, nil, "")
|
||||
logger, err := hlog.NewLogger(logcfg, &buf)
|
||||
require.NoError(t, err)
|
||||
|
||||
err = server.AddLogger(logger)
|
||||
@@ -80,7 +84,7 @@ func Test_GZIP_Compression(t *testing.T) {
|
||||
})
|
||||
require.NoError(t, err)
|
||||
|
||||
logger, err := hlog.NewLogger(hlog.LogLevel("Debug"), &buf, nil, "")
|
||||
logger, err := hlog.NewLogger(logcfg, &buf)
|
||||
require.NoError(t, err)
|
||||
|
||||
err = server.AddLogger(logger)
|
||||
@@ -131,7 +135,7 @@ func Test_GZIP_Compression(t *testing.T) {
|
||||
})
|
||||
require.NoError(t, err)
|
||||
|
||||
logger, err := hlog.NewLogger(hlog.LogLevel("Debug"), &buf, nil, "")
|
||||
logger, err := hlog.NewLogger(logcfg, &buf)
|
||||
require.NoError(t, err)
|
||||
|
||||
err = server.AddLogger(logger)
|
||||
@@ -179,20 +183,20 @@ func Test_GzipResponseWriter(t *testing.T) {
|
||||
t.Run("Can write through gzip writer", func(t *testing.T) {
|
||||
var buf bytes.Buffer
|
||||
gzWriter := gzip.NewWriter(&buf)
|
||||
|
||||
|
||||
testData := []byte("Test data to compress")
|
||||
n, err := gzWriter.Write(testData)
|
||||
require.NoError(t, err)
|
||||
assert.Equal(t, len(testData), n)
|
||||
|
||||
|
||||
err = gzWriter.Close()
|
||||
require.NoError(t, err)
|
||||
|
||||
|
||||
// Decompress and verify
|
||||
gzReader, err := gzip.NewReader(&buf)
|
||||
require.NoError(t, err)
|
||||
defer gzReader.Close()
|
||||
|
||||
|
||||
decompressed, err := io.ReadAll(gzReader)
|
||||
require.NoError(t, err)
|
||||
assert.Equal(t, testData, decompressed)
|
||||
@@ -215,9 +219,9 @@ func Test_GzipResponseWriter(t *testing.T) {
|
||||
req := httptest.NewRequest("GET", "/test", nil)
|
||||
req.Header.Set("Accept-Encoding", "gzip")
|
||||
rr := httptest.NewRecorder()
|
||||
|
||||
|
||||
wrapped.ServeHTTP(rr, req)
|
||||
|
||||
|
||||
// Note: This is a simplified test
|
||||
})
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user