Helper method to receive WebhookEvent (#109)

This commit is contained in:
David Zhao
2022-03-30 16:14:23 -07:00
committed by GitHub
parent 6ba1221db3
commit 54544c489c
3 changed files with 105 additions and 24 deletions

View File

@@ -36,3 +36,26 @@ func (p *FileBasedKeyProvider) GetSecret(key string) string {
func (p *FileBasedKeyProvider) NumKeys() int {
return len(p.keys)
}
type SimpleKeyProvider struct {
apiKey string
apiSecret string
}
func NewSimpleKeyProvider(apiKey, apiSecret string) *SimpleKeyProvider {
return &SimpleKeyProvider{
apiKey: apiKey,
apiSecret: apiSecret,
}
}
func (p *SimpleKeyProvider) GetSecret(key string) string {
if key == p.apiKey {
return p.apiSecret
}
return ""
}
func (p *SimpleKeyProvider) NumKeys() int {
return 1
}

View File

@@ -6,7 +6,10 @@ import (
"io/ioutil"
"net/http"
"google.golang.org/protobuf/encoding/protojson"
"github.com/livekit/protocol/auth"
"github.com/livekit/protocol/livekit"
)
// Receive reads and verifies incoming webhook is signed with key/secret pair
@@ -48,3 +51,20 @@ func Receive(r *http.Request, provider auth.KeyProvider) ([]byte, error) {
return data, nil
}
// ReceiveWebhookEvent reads and verifies incoming webhook, and returns a parsed WebhookEvent
func ReceiveWebhookEvent(r *http.Request, provider auth.KeyProvider) (*livekit.WebhookEvent, error) {
data, err := Receive(r, provider)
if err != nil {
return nil, err
}
unmarshalOpts := protojson.UnmarshalOptions{
DiscardUnknown: true,
AllowPartial: true,
}
event := livekit.WebhookEvent{}
if err = unmarshalOpts.Unmarshal(data, &event); err != nil {
return nil, err
}
return &event, nil
}

View File

@@ -5,45 +5,83 @@ import (
"encoding/json"
"net"
"net/http"
"sync"
"testing"
"github.com/stretchr/testify/require"
"github.com/livekit/protocol/auth"
"github.com/livekit/protocol/livekit"
"github.com/livekit/protocol/webhook"
)
const (
apiKey = "mykey"
apiSecret = "mysecret"
)
func TestWebHook(t *testing.T) {
s := newServer(":8765")
require.NoError(t, s.Start())
defer s.Stop()
authProvider := auth.NewFileBasedKeyProviderFromMap(map[string]string{
"mykey": "mysecret",
})
payload := map[string]interface{}{
"test": "payload",
"nested": map[string]interface{}{
"structure": true,
},
}
s.handler = func(r *http.Request) {
// receive logic
data, err := webhook.Receive(r, authProvider)
require.NoError(t, err)
var decoded map[string]interface{}
require.NoError(t, json.Unmarshal(data, &decoded))
require.EqualValues(t, decoded, payload)
}
notifier := webhook.NewNotifier("mykey", "mysecret", []string{
authProvider := auth.NewSimpleKeyProvider(
apiKey, apiSecret,
)
notifier := webhook.NewNotifier(apiKey, apiSecret, []string{
"http://localhost:8765",
})
require.NoError(t, notifier.Notify(context.Background(), payload))
t.Run("test json payload", func(t *testing.T) {
payload := map[string]interface{}{
"test": "payload",
"nested": map[string]interface{}{
"structure": true,
},
}
wg := sync.WaitGroup{}
wg.Add(1)
s.handler = func(r *http.Request) {
defer wg.Done()
// receive logic
data, err := webhook.Receive(r, authProvider)
require.NoError(t, err)
var decoded map[string]interface{}
require.NoError(t, json.Unmarshal(data, &decoded))
require.EqualValues(t, decoded, payload)
}
require.NoError(t, notifier.Notify(context.Background(), payload))
wg.Wait()
})
t.Run("test event payload", func(t *testing.T) {
event := &livekit.WebhookEvent{
Event: webhook.EventTrackPublished,
Participant: &livekit.ParticipantInfo{
Identity: "test",
},
Track: &livekit.TrackInfo{
Sid: "TR_abcde",
},
}
wg := sync.WaitGroup{}
wg.Add(1)
s.handler = func(r *http.Request) {
defer wg.Done()
decodedEvent, err := webhook.ReceiveWebhookEvent(r, authProvider)
require.NoError(t, err)
require.EqualValues(t, event, decodedEvent)
}
require.NoError(t, notifier.Notify(context.Background(), event))
wg.Wait()
})
}
type testServer struct {