mirror of
https://git.mirrors.martin98.com/https://github.com/SigNoz/signoz
synced 2025-06-04 11:25:52 +08:00
7f39d8282c
* feat(keys): add support for multiple ingestion keys * ci(git): remove vendor/ from git * feat(gateway): create a proxy for sending requests to the gateway * fix(sqlite): remove keys schema * fix(api): replace with constant * fix(server): remove redundant options * fix(server): remove redundant options * test(gateway): add unit tests for gateway proxy * ci(docker): update gateway url * refactor(gateway): move gateway to api layer * fix(manager): fix declared error in manager * feat(testing): add a new testing docker-compose * fix(license): revert to nil license since select will never return a norows error * feat(gateway): add feature flags * chore(server): add a logger --------- Co-authored-by: Srikanth Chekuri <srikanth.chekuri92@gmail.com>
62 lines
1.8 KiB
Go
62 lines
1.8 KiB
Go
package gateway
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"net/http/httputil"
|
|
"net/url"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestProxyRewrite(t *testing.T) {
|
|
testCases := []struct {
|
|
name string
|
|
url *url.URL
|
|
stripPath string
|
|
in *url.URL
|
|
expected *url.URL
|
|
}{
|
|
{
|
|
name: "SamePathAdded",
|
|
url: &url.URL{Scheme: "http", Host: "backend", Path: "/path1"},
|
|
stripPath: "/strip",
|
|
in: &url.URL{Scheme: "http", Host: "localhost", Path: "/strip/path1"},
|
|
expected: &url.URL{Scheme: "http", Host: "backend", Path: "/path1/path1"},
|
|
},
|
|
{
|
|
name: "NoStripPathInput",
|
|
url: &url.URL{Scheme: "http", Host: "backend"},
|
|
stripPath: "",
|
|
in: &url.URL{Scheme: "http", Host: "localhost", Path: "/strip/path1"},
|
|
expected: &url.URL{Scheme: "http", Host: "backend", Path: "/strip/path1"},
|
|
},
|
|
{
|
|
name: "NoStripPathPresentInReq",
|
|
url: &url.URL{Scheme: "http", Host: "backend"},
|
|
stripPath: "/not-found",
|
|
in: &url.URL{Scheme: "http", Host: "localhost", Path: "/strip/path1"},
|
|
expected: &url.URL{Scheme: "http", Host: "backend", Path: "/strip/path1"},
|
|
},
|
|
}
|
|
|
|
for _, tc := range testCases {
|
|
proxy, err := NewProxy(tc.url.String(), tc.stripPath)
|
|
require.NoError(t, err)
|
|
inReq, err := http.NewRequest(http.MethodGet, tc.in.String(), nil)
|
|
require.NoError(t, err)
|
|
proxyReq := &httputil.ProxyRequest{
|
|
In: inReq,
|
|
Out: inReq.Clone(context.Background()),
|
|
}
|
|
proxy.Rewrite(proxyReq)
|
|
|
|
assert.Equal(t, tc.expected.Host, proxyReq.Out.URL.Host)
|
|
assert.Equal(t, tc.expected.Scheme, proxyReq.Out.URL.Scheme)
|
|
assert.Equal(t, tc.expected.Path, proxyReq.Out.URL.Path)
|
|
assert.Equal(t, tc.expected.Query(), proxyReq.Out.URL.Query())
|
|
}
|
|
}
|