Examples with HTTPAnnounceResponseMessage com.turn.ttorrent.common.protocol.http.HTTPAnnounceResponseMessage used on opensource projects

Example 1 with HTTPAnnounceResponseMessage

use of com.turn.ttorrent.common.protocol.http.HTTPAnnounceResponseMessage in project teamcity-torrent-plugin by JetBrains.

the class TorrentTrackerConfiguratorTest method test_announce_interval.

public void test_announce_interval() throws IOException, TrackerMessage.MessageValidationException {
    System.setProperty(TorrentConfiguration.ANNOUNCE_INTERVAL, "10");
    myConfigurator.getConfigurationWatcher().checkForModifications();
    assertEquals(10, myConfigurator.getAnnounceIntervalSec());
    assertEquals(10, myTrackerManager.getTrackerService().getAnnounceInterval());
    final String uri = "http://localhost:8111/trackerAnnounce.html" + "?info_hash=12345678901234567890" + "&peer_id=ABCDEFGHIJKLMNOPQRST" + "&ip=127.0.0.0" + "&port=6881" + "&downloaded=1234" + "&left=98765" + "&event=stopped";
    final AtomicReference<String> response = new AtomicReference<String>();
    myTrackerManager.getTrackerService().process(uri, "http://localhost:8111/", new TrackerRequestProcessor.RequestHandler() {

        public void serveResponse(int code, String description, ByteBuffer responseData) {
            response.set(new String(responseData.array()));
        }

        public ConcurrentMap<String, TrackedTorrent> getTorrentsMap() {
            return new ConcurrentHashMap<String, TrackedTorrent>();
        }
    });
    final HTTPAnnounceResponseMessage parse = (HTTPAnnounceResponseMessage) HTTPTrackerMessage.parse(new ByteArrayInputStream(response.get().getBytes()));
    assertEquals(10, parse.getInterval());
}

Example 2 with HTTPAnnounceResponseMessage

use of com.turn.ttorrent.common.protocol.http.HTTPAnnounceResponseMessage in project teamcity-torrent-plugin by JetBrains.

the class TorrentTrackerConfiguratorTest method test_torrent_expire_timeout.

public void test_torrent_expire_timeout() throws IOException, TrackerMessage.MessageValidationException, InterruptedException {
    System.setProperty(TorrentConfiguration.TRACKER_TORRENT_EXPIRE_TIMEOUT, "5");
    myConfigurator.getConfigurationWatcher().checkForModifications();
    assertEquals(5, myConfigurator.getTrackerTorrentExpireTimeoutSec());
    final String uriCompleted = "http://localhost:8111/trackerAnnounce.html" + "?info_hash=12345678901234567890" + "&peer_id=ABCDEFGHIJKLMNOPQRST" + "&ip=172.20.240.249" + "&port=6884" + "&downloaded=1234" + "&left=0" + "&event=" + "completed";
    final String uriCompleted2 = "http://localhost:8111/trackerAnnounce.html" + "?info_hash=12345678901234567890" + "&peer_id=BBCDEFGHIJKLMNOPQRST" + "&ip=172.20.240.249" + "&port=6881" + "&downloaded=1234" + "&left=0" + "&event=" + "completed";
    // assertEquals(uriCompleted, uriCompleted2);
    final AtomicReference<byte[]> response = new AtomicReference<byte[]>();
    final TrackerRequestProcessor.RequestHandler requestHandler = new TrackerRequestProcessor.RequestHandler() {

        public void serveResponse(int code, String description, ByteBuffer responseData) {
            response.set(responseData.array());
        }
    };
    myTrackerManager.getTrackerService().process(uriCompleted, "http://localhost:8111/", requestHandler);
    assertEquals(1, myTrackerManager.getTorrents().size());
    final AtomicInteger complete = new AtomicInteger(100);
    final AtomicInteger peersSize = new AtomicInteger(100);
    final String torrentHash = "3132333435363738393031323334353637383930";
    new WaitFor(15 * 1000) {

        @Override
        protected boolean condition() {
            try {
                myTrackerManager.getTrackerService().process(uriCompleted2, "http://localhost:8111/", requestHandler);
                final HTTPAnnounceResponseMessage parse = (HTTPAnnounceResponseMessage) HTTPTrackerMessage.parse(new ByteArrayInputStream(response.get()));
                complete.set(parse.getComplete());
                peersSize.set(parse.getPeers().size());
                final TrackedTorrent trackedTorrent = myTrackerManager.getTorrents().get(torrentHash);
                return parse.getComplete() == 1 && trackedTorrent.getPeers().size() == 1;
            } catch (IOException e) {
                e.printStackTrace();
            } catch (TrackerMessage.MessageValidationException e) {
                e.printStackTrace();
            }
            return false;
        }
    };
    final TrackedTorrent trackedTorrent = myTrackerManager.getTorrents().get(torrentHash);
    for (PeerUID peerUID : trackedTorrent.getPeers().keySet()) {
        if (peerUID.getTorrentHash().equals("4142434445464748494A4B4C4D4E4F5051525354"))
            fail();
    }
    new WaitFor(15 * 1000) {

        @Override
        protected boolean condition() {
            return !myTrackerManager.getTorrents().containsKey(torrentHash);
        }
    };
    assertNotContains(myTrackerManager.getTorrents().keySet(), torrentHash);
}