source: src/test/java/negotiator/protocol/MultiLateralProtocolAdapterTest.java

package negotiator.protocol;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertTrue;
import static org.mockito.Mockito.mock;

import java.util.ArrayList;
import java.util.Collection;
import java.util.List;

import org.junit.Before;
import org.junit.Test;
import org.mockito.Mockito;

import genius.core.parties.NegotiationParty;
import genius.core.protocol.DefaultMultilateralProtocol;
import genius.core.session.Session;

public class MultiLateralProtocolAdapterTest {

        private DefaultMultilateralProtocol protocol;

        Session session = mock(Session.class);

        @Before
        public void init() {

                protocol = mock(DefaultMultilateralProtocol.class,
                                Mockito.CALLS_REAL_METHODS);
        }

        @Test
        public void testEndNego() {
                protocol.endNegotiation();
                assertTrue(protocol.isFinished(session, null));
        }

        @Test
        public void testFilterEmptyList() {
                List<NegotiationParty> negotiationParties = new ArrayList<NegotiationParty>();
                Collection<NegotiationParty> filtered = protocol.includeOnly(
                                negotiationParties, NegotiationParty.class);
                assertTrue("filter result not empty", filtered.isEmpty());
        }

        @Test
        public void testFilterOneParty() {
                List<NegotiationParty> negotiationParties = new ArrayList<NegotiationParty>();
                NegotiationParty oneParty = mock(NegotiationParty.class);
                negotiationParties.add(oneParty);

                Collection<NegotiationParty> filtered = protocol.includeOnly(
                                negotiationParties, oneParty.getClass());
                assertFalse("filter result is empty", filtered.isEmpty());
        }

        /**
         * Filter two parties, only first one should remain.
         */
        @Test
        public void testFilterTwoParties() {
                List<NegotiationParty> negotiationParties = new ArrayList<NegotiationParty>();
                NegotiationParty party1 = mock(NegoParty1.class);
                NegotiationParty party2 = mock(NegoParty2.class);

                negotiationParties.add(party1);
                negotiationParties.add(party2);

                Collection<NegotiationParty> filtered = protocol.includeOnly(
                                negotiationParties, party1.getClass());
                assertEquals(1, filtered.size());
                assertEquals(party1, filtered.iterator().next());
        }

        /**
         * Filter two parties, only first one should remain.
         */
        @Test
        public void testFilterTwoPartiesExclude() {
                List<NegotiationParty> negotiationParties = new ArrayList<NegotiationParty>();
                NegotiationParty party1 = mock(NegoParty1.class);
                NegotiationParty party2 = mock(NegoParty2.class);

                negotiationParties.add(party1);
                negotiationParties.add(party2);

                Collection<NegotiationParty> filtered = protocol.exclude(
                                negotiationParties, party1.getClass());
                assertEquals(1, filtered.size());
                assertEquals(party2, filtered.iterator().next());
        }

        interface NegoParty1 extends NegotiationParty {
        }

        interface NegoParty2 extends NegotiationParty {

        }

}