MediaDescriptionFieldTest.java

/*
 * TeleStax, Open Source Cloud Communications
 * Copyright 2011-2014, Telestax Inc and individual contributors
 * by the @authors tag.
 *
 * This program is free software: you can redistribute it and/or modify
 * under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation; either version 3 of
 * the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>
 *
 */

package org.restcomm.media.sdp.fields;

import junit.framework.Assert;

import org.junit.Test;
import org.restcomm.media.sdp.MediaProfile;
import org.restcomm.media.sdp.SessionDescription;
import org.restcomm.media.sdp.dtls.attributes.FingerprintAttribute;
import org.restcomm.media.sdp.fields.MediaDescriptionField;
import org.restcomm.media.sdp.ice.attributes.IcePwdAttribute;
import org.restcomm.media.sdp.ice.attributes.IceUfragAttribute;

/**
 * 
 * @author Henrique Rosa (henrique.rosa@telestax.com)
 *
 */
public class MediaDescriptionFieldTest {
	
	@Test
	public void testCustomMediaDescription() {
		// given
		String media = "audio";
		int port = 65535;
		String protocol = "RTP/AVP";
		int[] formats = new int[] { 0, 101 };
		
		// when
		MediaDescriptionField md = new MediaDescriptionField();
		md.setMedia(media);
		md.setPort(port);
		md.setProtocol(MediaProfile.RTP_AVP.getProfile());
		md.setPayloadTypes(formats);
		
		// then
		Assert.assertEquals(media, md.getMedia());
		Assert.assertEquals(port, md.getPort());
		Assert.assertEquals(protocol, md.getProtocol());
		Assert.assertTrue(md.containsPayloadType((short) 0));
		Assert.assertTrue(md.containsPayloadType((short) 101));
		Assert.assertFalse(md.containsPayloadType((short) 126));
	}

	@Test
	public void testSessionLevelAccessor() {
		// given
		SessionDescription session = new SessionDescription();
		IcePwdAttribute sessionIcePwd = new IcePwdAttribute("password");
		session.setIcePwd(sessionIcePwd);
		IceUfragAttribute sessionIceUfrag = new IceUfragAttribute("ufrag");
		session.setIceUfrag(sessionIceUfrag);
		FingerprintAttribute sessionFingerprint = new FingerprintAttribute("sha-256", "35:A6:57");
		session.setFingerprint(sessionFingerprint);
		
		// when
		MediaDescriptionField media = new MediaDescriptionField(session);
		IcePwdAttribute mediaIcePwd = new IcePwdAttribute("password2");
		media.setIcePwd(mediaIcePwd);
		IceUfragAttribute mediaIceUfrag = new IceUfragAttribute("ufrag2");
		media.setIceUfrag(mediaIceUfrag);
		
		// then
		Assert.assertEquals(mediaIcePwd, media.getIcePwd());
		Assert.assertEquals(mediaIceUfrag, media.getIceUfrag());
		Assert.assertEquals(sessionFingerprint, media.getFingerprint());
	}
}