SdpFactoryTest.java

/**
 * 
 */
package org.restcomm.media.rtp.sdp;

import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

import org.junit.Assert;
import org.junit.Test;
import org.restcomm.media.rtp.channels.AudioChannel;
import org.restcomm.media.rtp.sdp.SdpFactory;
import org.restcomm.media.sdp.SdpException;
import org.restcomm.media.sdp.SessionDescription;
import org.restcomm.media.sdp.fields.MediaDescriptionField;
import org.restcomm.media.sdp.fields.parser.MediaDescriptionFieldParser;
import org.restcomm.media.sdp.format.RTPFormat;
import org.restcomm.media.sdp.format.RTPFormats;
import org.restcomm.media.spi.format.EncodingName;
import org.restcomm.media.spi.format.Format;
import org.restcomm.media.spi.format.FormatFactory;

/**
 * @author Henrique Rosa (henrique.rosa@telestax.com)
 *
 */
public class SdpFactoryTest {

    private final String localAddress = "127.0.0.1";
    private final String externalAddress = "external.address.com";
    private final String rtpAddress = "rtp.address.com";
    private final String rtcpAddress = "rtcp.address.com";
    private Format audioFormat = FormatFactory.createAudioFormat(new EncodingName("audio"));
    private RTPFormat rtpFormat = new RTPFormat(0, audioFormat);
    private RTPFormats rtpFormats = new RTPFormats();

    public SdpFactoryTest() {
        this.rtpFormats.add(rtpFormat);
    }
	
	@Test
	public void testRejectApplication() throws SdpException {
		// given
		String applicationSdp = "m=application 9 DTLS/SCTP 5000";
		MediaDescriptionFieldParser mediaFieldParser = new MediaDescriptionFieldParser();
		MediaDescriptionField applicationField = mediaFieldParser.parse(applicationSdp);
		
		SessionDescription answer = new SessionDescription();
		
		// when
		SdpFactory.rejectMediaField(answer, applicationField);
		
		// then
		MediaDescriptionField mediaDescription = answer.getMediaDescription("application");
		Assert.assertNotNull(mediaDescription);
		Assert.assertEquals(0, mediaDescription.getPort());
		Assert.assertEquals("DTLS/SCTP", mediaDescription.getProtocol());
		int[] payloadTypes = mediaDescription.getPayloadTypes();
		Assert.assertEquals(1, payloadTypes.length);
		Assert.assertEquals(5000, payloadTypes[0]);
	}

    @Test
    public void testValidExternalAddress() {
        // given
        AudioChannel audioChannel = mock(AudioChannel.class);

        // when
        when(audioChannel.getExternalAddress()).thenReturn(this.externalAddress);
        when(audioChannel.getRtpAddress()).thenReturn(this.rtpAddress);
        when(audioChannel.getRtcpAddress()).thenReturn(this.rtcpAddress);
        when(audioChannel.isIceEnabled()).thenReturn(false);
        when(audioChannel.isRtcpMux()).thenReturn(false);
        when(audioChannel.getFormats()).thenReturn(this.rtpFormats);
        when(audioChannel.getMediaType()).thenReturn(AudioChannel.MEDIA_TYPE);

        // then test build SDP with valid externalAddress
        SessionDescription sd = SdpFactory.buildSdp(true, this.localAddress, this.externalAddress, audioChannel);
        Assert.assertTrue(sd.getConnection().getAddress().equals(this.externalAddress));
        Assert.assertTrue(sd.getMediaDescription("audio").getConnection().getAddress().equals(this.externalAddress));
        Assert.assertTrue(sd.getMediaDescription("audio").getRtcp().getAddress().equals(this.externalAddress));

    }

    @Test
    public void testValidExternalAddressWithIceRtcpMux() {
        // given
        AudioChannel audioChannel = mock(AudioChannel.class);

        // when
        when(audioChannel.getExternalAddress()).thenReturn(this.externalAddress);
        when(audioChannel.getRtpAddress()).thenReturn(this.rtpAddress);
        when(audioChannel.getRtcpAddress()).thenReturn(this.rtcpAddress);
        when(audioChannel.isIceEnabled()).thenReturn(true);
        when(audioChannel.isRtcpMux()).thenReturn(true);
        when(audioChannel.getFormats()).thenReturn(this.rtpFormats);
        when(audioChannel.getMediaType()).thenReturn(AudioChannel.MEDIA_TYPE);

        // then test build SDP with valid externalAddress, ICE and RTCP-MUX
        SessionDescription sd = SdpFactory.buildSdp(true, this.localAddress, this.externalAddress, audioChannel);
        Assert.assertTrue(sd.getConnection().getAddress().equals(this.externalAddress));
        Assert.assertTrue(sd.getMediaDescription("audio").getConnection().getAddress().equals(this.externalAddress));
        Assert.assertTrue(sd.getMediaDescription("audio").getRtcp().getAddress().equals(this.externalAddress));

    }

    @Test
    public void testEmptyExternallAddress() {
        // given
        AudioChannel audioChannel = mock(AudioChannel.class);

        // when
        when(audioChannel.getExternalAddress()).thenReturn("");
        when(audioChannel.getRtpAddress()).thenReturn(this.rtpAddress);
        when(audioChannel.getRtcpAddress()).thenReturn(this.rtcpAddress);
        when(audioChannel.isIceEnabled()).thenReturn(false);
        when(audioChannel.isRtcpMux()).thenReturn(false);
        when(audioChannel.getFormats()).thenReturn(this.rtpFormats);
        when(audioChannel.getMediaType()).thenReturn(AudioChannel.MEDIA_TYPE);

        // then test build SDP with empty externalAddress
        SessionDescription sd = SdpFactory.buildSdp(true, this.localAddress, "", audioChannel);
        Assert.assertTrue(sd.getConnection().getAddress().equals(this.localAddress));
        Assert.assertTrue(sd.getMediaDescription("audio").getConnection().getAddress().equals(this.rtpAddress));
        Assert.assertTrue(sd.getMediaDescription("audio").getRtcp().getAddress().equals(this.rtcpAddress));

    }

    @Test
    public void testEmptyExternalAddressWithIceRtcpMux() {
        // given
        AudioChannel audioChannel = mock(AudioChannel.class);

        // when
        when(audioChannel.getExternalAddress()).thenReturn("");
        when(audioChannel.getRtpAddress()).thenReturn(this.rtpAddress);
        when(audioChannel.getRtcpAddress()).thenReturn(this.rtcpAddress);
        when(audioChannel.isIceEnabled()).thenReturn(true);
        when(audioChannel.isRtcpMux()).thenReturn(true);
        when(audioChannel.getFormats()).thenReturn(this.rtpFormats);
        when(audioChannel.getMediaType()).thenReturn(AudioChannel.MEDIA_TYPE);

        // then test build SDP with empty externalAddress, ICE and RTCP-MUX
        SessionDescription sd = SdpFactory.buildSdp(true, this.localAddress, "", audioChannel);
        Assert.assertTrue(sd.getConnection().getAddress().equals(this.rtpAddress));
        Assert.assertTrue(sd.getMediaDescription("audio").getConnection().getAddress().equals(this.rtpAddress));
        Assert.assertTrue(sd.getMediaDescription("audio").getRtcp().getAddress().equals(this.rtpAddress));

    }

    @Test
    public void testNullExternalAddress() {
        // given
        AudioChannel audioChannel = mock(AudioChannel.class);

        // when
        when(audioChannel.getExternalAddress()).thenReturn(null);
        when(audioChannel.getRtpAddress()).thenReturn(this.rtpAddress);
        when(audioChannel.getRtcpAddress()).thenReturn(this.rtcpAddress);
        when(audioChannel.isIceEnabled()).thenReturn(false);
        when(audioChannel.isRtcpMux()).thenReturn(false);
        when(audioChannel.getFormats()).thenReturn(this.rtpFormats);
        when(audioChannel.getMediaType()).thenReturn(AudioChannel.MEDIA_TYPE);

        // then test build SDP with null externalAddress
        SessionDescription sd = SdpFactory.buildSdp(true, this.localAddress, null, audioChannel);
        Assert.assertTrue(sd.getConnection().getAddress().equals(this.localAddress));
        Assert.assertTrue(sd.getMediaDescription("audio").getConnection().getAddress().equals(this.rtpAddress));
        Assert.assertTrue(sd.getMediaDescription("audio").getRtcp().getAddress().equals(this.rtcpAddress));

    }

    @Test
    public void testNullExternalAddressWithIceRtcpMux() {
        // given
        AudioChannel audioChannel = mock(AudioChannel.class);

        // when
        when(audioChannel.getExternalAddress()).thenReturn(null);
        when(audioChannel.getRtpAddress()).thenReturn(this.rtpAddress);
        when(audioChannel.getRtcpAddress()).thenReturn(this.rtcpAddress);
        when(audioChannel.isIceEnabled()).thenReturn(true);
        when(audioChannel.isRtcpMux()).thenReturn(true);
        when(audioChannel.getFormats()).thenReturn(this.rtpFormats);
        when(audioChannel.getMediaType()).thenReturn(AudioChannel.MEDIA_TYPE);

        // then test build SDP with null externalAddress, ICE and RTCP-MUX
        SessionDescription sd = SdpFactory.buildSdp(true, this.localAddress, null, audioChannel);
        Assert.assertTrue(sd.getConnection().getAddress().equals(this.rtpAddress));
        Assert.assertTrue(sd.getMediaDescription("audio").getConnection().getAddress().equals(this.rtpAddress));
        Assert.assertTrue(sd.getMediaDescription("audio").getRtcp().getAddress().equals(this.rtpAddress));
    }

}