SettingsBasedContentAuthorityTest.java

package com.limegroup.gnutella.auth;


import java.util.Random;

import junit.framework.Test;

import org.limewire.core.settings.ContentSettings;
import org.limewire.io.IpPort;

import com.google.inject.Injector;
import com.limegroup.gnutella.LimeTestUtils;
import com.limegroup.gnutella.messages.Message;
import com.limegroup.gnutella.messages.PingRequestFactory;
import com.limegroup.gnutella.util.LimeTestCase;
 
public class SettingsBasedContentAuthorityTest extends LimeTestCase {
    
    private IpPortContentAuthorityFactory ipPortContentAuthorityFactory;
    private PingRequestFactory pingRequestFactory;

    public SettingsBasedContentAuthorityTest(String name) {
        super(name);
    }

    public static Test suite() {
        return buildTestSuite(SettingsBasedContentAuthorityTest.class);
    }

    @Override
    protected void setUp() throws Exception {
		Injector injector = LimeTestUtils.createInjector();
		ipPortContentAuthorityFactory = injector.getInstance(IpPortContentAuthorityFactory.class);
		pingRequestFactory = injector.getInstance(PingRequestFactory.class);
    }
    
    public void testInitialize() {
        ContentSettings.AUTHORITIES.setValue(new String[] { "yahoo.com", "google.com:82", "askjeeves.com:6346" });
        SettingsBasedContentAuthority auth = new SettingsBasedContentAuthority(ipPortContentAuthorityFactory);
        auth.initialize();
        ContentAuthority[] all = auth.getAuthorities();
        assertEquals(3, all.length);
        
        IpPort one   = ((IpPortContentAuthority)all[0]).getIpPort();
        assertEquals("yahoo.com", one.getAddress());
        assertEquals(80, one.getPort());
        
        IpPort two   = ((IpPortContentAuthority)all[1]).getIpPort();
        assertEquals("google.com", two.getAddress());
        assertEquals(82, two.getPort());
        
        IpPort three = ((IpPortContentAuthority)all[2]).getIpPort();
        assertEquals("askjeeves.com", three.getAddress());
        assertEquals(6346, three.getPort());      
    }
    
    public void testInitializeDNSFails() {
        ContentSettings.AUTHORITIES.setValue(new String[] { "notarealdnsnamesodonteventryit.com", "google.com" });
        SettingsBasedContentAuthority auth = new SettingsBasedContentAuthority(ipPortContentAuthorityFactory);
        auth.initialize();
        ContentAuthority[] all = auth.getAuthorities();
        assertEquals(1, all.length);
        
        IpPort one   = ((IpPortContentAuthority)all[0]).getIpPort();
        assertEquals("google.com", one.getAddress());
        assertEquals(80, one.getPort());           
    }
    
    public void testRandomSending() throws Exception {
        final int[] randoms = new int[] { 0, 2, 1, 0, 1, 0, 0, 1, 2, 2 };
        
        SettingsBasedContentAuthority auth = new SettingsBasedContentAuthority(ipPortContentAuthorityFactory) {
            @Override
            public Random newRandom() {
                return new StubRandom(randoms);
            }
        };
        StubContentAuthority zero = new StubContentAuthority();
        StubContentAuthority one  = new StubContentAuthority();
        StubContentAuthority two  = new StubContentAuthority();
        auth.setAuthorities(new ContentAuthority[] { zero, one, two } );

        Message[] ms = new Message[randoms.length];
        for(int i = 0; i < randoms.length; i++) {
            ms[i] = pingRequestFactory.createPingRequest((byte)1);
            auth.send(ms[i]);
        }
        
        assertSame(zero.getSent().get(0), ms[0]);
        assertSame( two.getSent().get(0), ms[1]);
        assertSame( one.getSent().get(0), ms[2]);
        assertSame(zero.getSent().get(1), ms[3]);
        assertSame( one.getSent().get(1), ms[4]);
        assertSame(zero.getSent().get(2), ms[5]);
        assertSame(zero.getSent().get(3), ms[6]);
        assertSame( one.getSent().get(2), ms[7]);
        assertSame( two.getSent().get(1), ms[8]);
        assertSame( two.getSent().get(2), ms[9]);
    }
    
    private static class StubRandom extends Random {
        private int next = 0;
        private int[] rnds;
        
        StubRandom(int[] rnds) {
            this.rnds = rnds;
        }
        
        @Override
        public int nextInt(int i) {
            return rnds[next++];
        }
    }
    
}