Changeset b19866c


Ignore:
Timestamp:
Jul 27, 2015 3:33:30 AM (5 years ago)
Author:
str4d <str4d@…>
Branches:
master
Children:
5eda1e00
Parents:
48bcc031
Message:

Refactor part 2

File:
1 edited

Legend:

Unmodified
Added
Removed
  • apps/streaming/java/test/junit/net/i2p/client/streaming/impl/MessageInputStreamTest.java

    r48bcc031 rb19866c  
    55import java.util.Collections;
    66
     7import org.junit.After;
    78import org.junit.Before;
    89import org.junit.Test;
    910
    1011import junit.framework.TestCase;
    11 
    1212import net.i2p.I2PAppContext;
    1313import net.i2p.data.ByteArray;
     
    2222    private Log _log;
    2323    private ConnectionOptions _options;
     24    private MessageInputStream in;
    2425
    2526    @Before
     
    2829        _log = _context.logManager().getLog(MessageInputStreamTest.class);
    2930        _options = new ConnectionOptions();
     31
     32        in = new MessageInputStream(_context, _options.getMaxMessageSize(),
     33                _options.getMaxWindowSize(), _options.getInboundBufferSize());
     34    }
     35
     36    @After
     37    public void tearDown() {
     38        in.close();
    3039    }
    3140
     
    3544        _context.random().nextBytes(orig);
    3645
    37         MessageInputStream in = new MessageInputStream(_context, _options.getMaxMessageSize(),
    38                 _options.getMaxWindowSize(), _options.getInboundBufferSize());
    3946        for (int i = 0; i < orig.length / 1024; i++) {
    4047            byte msg[] = new byte[1024];
     
    5865        _context.random().nextBytes(orig);
    5966
    60         MessageInputStream in = new MessageInputStream(_context, _options.getMaxMessageSize(),
    61                 _options.getMaxWindowSize(), _options.getInboundBufferSize());
    6267        ArrayList<Integer> order = new ArrayList<Integer>(32);
    6368        for (int i = 0; i < orig.length / 1024; i++)
     
    8792        _context.random().nextBytes(orig);
    8893
    89         MessageInputStream in = new MessageInputStream(_context, _options.getMaxMessageSize(),
    90                 _options.getMaxWindowSize(), _options.getInboundBufferSize());
    9194        for (int n = 0; n < 3; n++) {
    9295            ArrayList<Integer> order = new ArrayList<Integer>(32);
     
    119122        _context.random().nextBytes(orig);
    120123
    121         MessageInputStream in = new MessageInputStream(_context, _options.getMaxMessageSize(),
    122                 _options.getMaxWindowSize(), _options.getInboundBufferSize());
    123124        ArrayList<Integer> order = new ArrayList<Integer>(32);
    124125        for (int i = 0; i < orig.length / 1024; i++)
     
    127128
    128129        int offset = 0;
    129         try {
    130             for (int i = 0; i < orig.length / 1024; i++) {
    131                 byte msg[] = new byte[1024];
    132                 Integer cur = (Integer)order.get(i);
    133                 System.arraycopy(orig, cur.intValue()*1024, msg, 0, 1024);
    134                 in.messageReceived(cur.intValue(), new ByteArray(msg));
    135                 _log.debug("Injecting " + cur);
     130        for (int i = 0; i < orig.length / 1024; i++) {
     131            byte msg[] = new byte[1024];
     132            Integer cur = (Integer)order.get(i);
     133            System.arraycopy(orig, cur.intValue()*1024, msg, 0, 1024);
     134            in.messageReceived(cur.intValue(), new ByteArray(msg));
     135            _log.debug("Injecting " + cur);
    136136
    137                 if (in.available() > 0) {
    138                     int curRead = in.read(read, offset, read.length-offset);
    139                     _log.debug("read " + curRead);
    140                     if (curRead == -1)
    141                         fail("EOF with offset " + offset);
    142                     else
    143                         offset += curRead;
    144                 }
     137            if (in.available() > 0) {
     138                int curRead = in.read(read, offset, read.length-offset);
     139                _log.debug("read " + curRead);
     140                if (curRead == -1)
     141                    fail("EOF with offset " + offset);
     142                else
     143                    offset += curRead;
    145144            }
    146         } finally {
    147             in.close();
    148145        }
    149146
Note: See TracChangeset for help on using the changeset viewer.