// SPDX-FileCopyrightText: 2023 The Pion community <https://pion.ly>
// SPDX-License-Identifier: MIT

package sdp

import (
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestNewJSEPSessionDescription(t *testing.T) {
	t.Run("Without Identity", func(t *testing.T) {
		sd, err := NewJSEPSessionDescription(false)
		assert.NoError(t, err)
		assert.NotNil(t, sd)
		assert.Zero(t, sd.Version)
		assert.Equal(t, "-", sd.Origin.Username)
		assert.Equal(t, "IN", sd.Origin.NetworkType)
		assert.Equal(t, "IP4", sd.Origin.AddressType)
		assert.Equal(t, "0.0.0.0", sd.Origin.UnicastAddress)
		assert.Equal(t, SessionName("-"), sd.SessionName)
		assert.Len(t, sd.TimeDescriptions, 1)
		assert.Zero(t, sd.TimeDescriptions[0].Timing.StartTime)
		assert.Zero(t, sd.TimeDescriptions[0].Timing.StopTime)
		assert.Empty(t, sd.Attributes)
	})

	t.Run("With Identity", func(t *testing.T) {
		sd, err := NewJSEPSessionDescription(true)
		assert.NoError(t, err)
		assert.NotNil(t, sd)
		assert.Len(t, sd.Attributes, 1)
		assert.Equal(t, AttrKeyIdentity, sd.Attributes[0].Key)
	})
}

func TestSessionDescriptionAttributes(t *testing.T) {
	sd, err := NewJSEPSessionDescription(false)
	assert.NoError(t, err)

	t.Run("WithPropertyAttribute", func(t *testing.T) {
		sd = sd.WithPropertyAttribute(AttrKeyRTCPMux)
		assert.Len(t, sd.Attributes, 1)
		assert.Equal(t, AttrKeyRTCPMux, sd.Attributes[0].Key)
	})

	t.Run("WithValueAttribute", func(t *testing.T) {
		sd = sd.WithValueAttribute(AttrKeyMID, "video")
		assert.Len(t, sd.Attributes, 2)
		assert.Equal(t, AttrKeyMID, sd.Attributes[1].Key)
		assert.Equal(t, "video", sd.Attributes[1].Value)
	})

	t.Run("WithICETrickleAdvertised", func(t *testing.T) {
		sd = sd.WithICETrickleAdvertised()
		assert.Len(t, sd.Attributes, 3)
		assert.Equal(t, AttrKeyICEOptions, sd.Attributes[2].Key)
		assert.Equal(t, "trickle", sd.Attributes[2].Value)
	})

	t.Run("WithFingerprint", func(t *testing.T) {
		sd = sd.WithFingerprint("sha-256", "test-fingerprint")
		assert.Len(t, sd.Attributes, 4)
		assert.Equal(t, "fingerprint", sd.Attributes[3].Key)
		assert.Equal(t, "sha-256 test-fingerprint", sd.Attributes[3].Value)
	})
}

func TestNewJSEPMediaDescription(t *testing.T) {
	md := NewJSEPMediaDescription("video", []string{"96", "97"})
	assert.NotNil(t, md)
	assert.Equal(t, "video", md.MediaName.Media)
	assert.Equal(t, int(9), md.MediaName.Port.Value)
	assert.Equal(t, []string{"UDP", "TLS", "RTP", "SAVPF"}, md.MediaName.Protos)
	assert.Equal(t, "IN", md.ConnectionInformation.NetworkType)
	assert.Equal(t, "IP4", md.ConnectionInformation.AddressType)
	assert.Equal(t, "0.0.0.0", md.ConnectionInformation.Address.Address)
}

func TestMediaDescriptionAttributes(t *testing.T) {
	md := NewJSEPMediaDescription("audio", nil)

	t.Run("WithPropertyAttribute", func(t *testing.T) {
		md = md.WithPropertyAttribute(AttrKeyRTCPMux)
		assert.Len(t, md.Attributes, 1)
		assert.Equal(t, AttrKeyRTCPMux, md.Attributes[0].Key)
	})

	t.Run("WithValueAttribute", func(t *testing.T) {
		md = md.WithValueAttribute(AttrKeyMID, "audio")
		assert.Len(t, md.Attributes, 2)
		assert.Equal(t, AttrKeyMID, md.Attributes[1].Key)
		assert.Equal(t, "audio", md.Attributes[1].Value)
	})

	t.Run("WithFingerprint", func(t *testing.T) {
		md = md.WithFingerprint("sha-256", "test-fingerprint")
		assert.Len(t, md.Attributes, 3)
		assert.Equal(t, "fingerprint", md.Attributes[2].Key)
		assert.Equal(t, "sha-256 test-fingerprint", md.Attributes[2].Value)
	})

	t.Run("WithICECredentials", func(t *testing.T) {
		md = md.WithICECredentials("test-ufrag", "test-pwd")
		assert.Len(t, md.Attributes, 5)
		assert.Equal(t, "ice-ufrag", md.Attributes[3].Key)
		assert.Equal(t, "test-ufrag", md.Attributes[3].Value)
		assert.Equal(t, "ice-pwd", md.Attributes[4].Key)
		assert.Equal(t, "test-pwd", md.Attributes[4].Value)
	})
}

func TestMediaDescriptionCodec(t *testing.T) {
	md := NewJSEPMediaDescription("audio", nil)

	t.Run("WithCodec", func(t *testing.T) {
		md = md.WithCodec(111, "opus", 48000, 2, "minptime=10;useinbandfec=1")
		assert.Len(t, md.MediaName.Formats, 1)
		assert.Equal(t, "111", md.MediaName.Formats[0])
		assert.Len(t, md.Attributes, 2)
		assert.Equal(t, "rtpmap", md.Attributes[0].Key)
		assert.Equal(t, "111 opus/48000/2", md.Attributes[0].Value)
		assert.Equal(t, "fmtp", md.Attributes[1].Key)
		assert.Equal(t, "111 minptime=10;useinbandfec=1", md.Attributes[1].Value)
	})

	t.Run("WithMediaSource", func(t *testing.T) {
		md = md.WithMediaSource(1234567890, "test-cname", "test-stream", "test-label")
		assert.Len(t, md.Attributes, 6)
		assert.Equal(t, "ssrc", md.Attributes[2].Key)
		assert.Equal(t, "1234567890 cname:test-cname", md.Attributes[2].Value)
		assert.Equal(t, "ssrc", md.Attributes[3].Key)
		assert.Equal(t, "1234567890 msid:test-stream test-label", md.Attributes[3].Value)
		assert.Equal(t, "ssrc", md.Attributes[4].Key)
		assert.Equal(t, "1234567890 mslabel:test-stream", md.Attributes[4].Value)
		assert.Equal(t, "ssrc", md.Attributes[5].Key)
		assert.Equal(t, "1234567890 label:test-label", md.Attributes[5].Value)
	})
}