package sinks import ( "context" "math/rand" "net/http" "net/http/httptest" "testing" "time" "github.com/stretchr/testify/assert" "github.com/stretchr/testify/require" "dynatron.me/x/stillbox/internal/common" "dynatron.me/x/stillbox/internal/forms" "dynatron.me/x/stillbox/pkg/auth" "dynatron.me/x/stillbox/pkg/calls" "dynatron.me/x/stillbox/pkg/config" "dynatron.me/x/stillbox/pkg/sources" "github.com/google/uuid" ) type hand func(w http.ResponseWriter, r *http.Request) func (h hand) ServeHTTP(w http.ResponseWriter, r *http.Request) { h(w, r) } func TestRelay(t *testing.T) { uuid.SetRand(rand.New(rand.NewSource(1))) tests := []struct { name string submitter auth.UserID apiKey string call calls.Call }{ { name: "base", submitter: auth.UserID(1), call: calls.Call{ ID: uuid.UUID([16]byte{0x52, 0xfd, 0xfc, 0x07, 0x21, 0x82, 0x45, 0x4f, 0x96, 0x3f, 0x5f, 0x0f, 0x9a, 0x62, 0x1d, 0x72}), Submitter: common.PtrTo(auth.UserID(1)), System: 197, Talkgroup: 10101, DateTime: time.Date(2024, 11, 10, 23, 33, 02, 0, time.Local), AudioName: "smallest.mp3", // the smallest MP3 there is Audio: []byte{0xFF, 0xF3, 0x14, 0xC4, 0x00, 0x00, 0x00, 0x03, 0x48, 0x01, 0x40, 0x00, 0x00, 0x4C, 0x41, 0x4D, 0x45, 0x33, 0x2E, 0x39, 0x36, 0x2E, 0x31, 0x55}, AudioType: "audio/mpeg", Duration: calls.CallDuration(24000000), TalkgroupLabel: common.PtrTo("Some TG"), }, }, } for _, tc := range tests { t.Run(tc.name, func(t *testing.T) { var serr error var called bool h := hand(func(w http.ResponseWriter, r *http.Request) { called = true assert.Equal(t, "/api/call-upload", r.URL.Path) serr = r.ParseMultipartForm(1024 * 1024 * 2) if serr != nil { t.Log("parsemultipart", serr) return } cur := new(sources.CallUploadRequest) serr = forms.Unmarshal(r, cur, forms.WithAcceptBlank()) cur.DontStore = true if serr != nil { t.Log("unmarshal", serr) return } assert.Equal(t, tc.apiKey, cur.Key) toC, tcerr := cur.ToCall(tc.submitter) require.NoError(t, tcerr) assert.Equal(t, &tc.call, toC) }) svr := httptest.NewServer(h) cfg := config.Relay{ URL: svr.URL, APIKey: tc.apiKey, } ns := &nullSinks{} rm, err := NewRelayManager(ns, []config.Relay{cfg}) require.NoError(t, err) err = rm.relays[0].Call(context.Background(), &tc.call) assert.True(t, called) assert.NoError(t, err) assert.NoError(t, serr) }) } } type nullSinks struct{} func (*nullSinks) Register(name string, toAdd Sink, required bool) {} func (*nullSinks) Unregister(name string) {} func (*nullSinks) Shutdown() {} func (*nullSinks) EmitCall(ctx context.Context, call *calls.Call) error { return nil }