mirror of
https://github.com/livekit/livekit.git
synced 2026-04-05 03:45:44 +00:00
6751361187
In order to ensure clients are publishing with compatible codecs, the server can enforce codec restrictions.
26 lines
1.1 KiB
Go
26 lines
1.1 KiB
Go
package rtc
|
|
|
|
import (
|
|
"testing"
|
|
|
|
livekit "github.com/livekit/livekit-server/proto"
|
|
"github.com/pion/webrtc/v3"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestIsCodecEnabled(t *testing.T) {
|
|
t.Run("empty fmtp requirement should match all", func(t *testing.T) {
|
|
enabledCodecs := []*livekit.Codec{{Mime: "video/h264"}}
|
|
require.True(t, isCodecEnabled(enabledCodecs, webrtc.RTPCodecCapability{MimeType: webrtc.MimeTypeH264, SDPFmtpLine: "special"}))
|
|
require.True(t, isCodecEnabled(enabledCodecs, webrtc.RTPCodecCapability{MimeType: webrtc.MimeTypeH264}))
|
|
require.False(t, isCodecEnabled(enabledCodecs, webrtc.RTPCodecCapability{MimeType: webrtc.MimeTypeVP8}))
|
|
})
|
|
|
|
t.Run("when fmtp is provided, require match", func(t *testing.T) {
|
|
enabledCodecs := []*livekit.Codec{{Mime: "video/h264", FmtpLine: "special"}}
|
|
require.True(t, isCodecEnabled(enabledCodecs, webrtc.RTPCodecCapability{MimeType: webrtc.MimeTypeH264, SDPFmtpLine: "special"}))
|
|
require.False(t, isCodecEnabled(enabledCodecs, webrtc.RTPCodecCapability{MimeType: webrtc.MimeTypeH264}))
|
|
require.False(t, isCodecEnabled(enabledCodecs, webrtc.RTPCodecCapability{MimeType: webrtc.MimeTypeVP8}))
|
|
})
|
|
}
|