Sanket SudakeandGitHub
caffed92f4
* errors.Wrap* removal Signed-off-by: Sanket Sudake <sanketsudake@gmail.com> * remove errors.Errorf Signed-off-by: Sanket Sudake <sanketsudake@gmail.com> * Remove remaining calls Signed-off-by: Sanket Sudake <sanketsudake@gmail.com> * Few more errors Signed-off-by: Sanket Sudake <sanketsudake@gmail.com> * Fix golint errors Signed-off-by: Sanket Sudake <sanketsudake@gmail.com> --------- Signed-off-by: Sanket Sudake <sanketsudake@gmail.com>
58 lines
1.6 KiB
Go
58 lines
1.6 KiB
Go
package publisher
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/fission/fission/pkg/utils/loggerfactory"
|
|
otelUtils "github.com/fission/fission/pkg/utils/otel"
|
|
)
|
|
|
|
func TestPublisher(t *testing.T) {
|
|
fnName := "test-fn"
|
|
s := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
assert.Equal(t, "/"+fnName, r.URL.Path)
|
|
assert.Equal(t, "aaa", r.Header.Get("X-Fission-Test"))
|
|
assert.Contains(t, r.Header, "Traceparent")
|
|
}))
|
|
|
|
ctx := context.Background()
|
|
logger := loggerfactory.GetLogger()
|
|
shutdown, err := otelUtils.InitProvider(ctx, logger, fnName)
|
|
assert.NoError(t, err)
|
|
if shutdown != nil {
|
|
defer shutdown(ctx)
|
|
}
|
|
|
|
wp := MakeWebhookPublisher(logger, s.URL)
|
|
wp.Publish(ctx, "", map[string]string{"X-Fission-Test": "aaa"}, http.MethodPost, fnName)
|
|
time.Sleep(time.Second * 1)
|
|
}
|
|
|
|
func TestPublisherSubpath(t *testing.T) {
|
|
subpath := "/api/v1/read"
|
|
fnName := "test-fn-subpath"
|
|
s := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
assert.Equal(t, "/"+fnName+subpath, r.URL.Path)
|
|
assert.Equal(t, "aaa", r.Header.Get("X-Fission-Test"))
|
|
assert.Contains(t, r.Header, "Traceparent")
|
|
}))
|
|
|
|
ctx := context.Background()
|
|
logger := loggerfactory.GetLogger()
|
|
shutdown, err := otelUtils.InitProvider(ctx, logger, fnName)
|
|
assert.NoError(t, err)
|
|
if shutdown != nil {
|
|
defer shutdown(ctx)
|
|
}
|
|
|
|
wp := MakeWebhookPublisher(logger, s.URL)
|
|
wp.Publish(ctx, "", map[string]string{"X-Fission-Test": "aaa"}, http.MethodGet, fnName+subpath)
|
|
time.Sleep(time.Second * 1)
|
|
}
|