コード例 #1
0
 @Test
 public void closeStream() throws IOException {
   final ConnectedStreamChannelMock channelMock = new ConnectedStreamChannelMock();
   final ChannelOutputStream delegateStream = new ChannelOutputStream(channelMock);
   final LimitedOutputStream stream = new LimitedOutputStream(delegateStream, 5);
   stream.write('a');
   stream.write('b');
   stream.write('c');
   // flush
   assertFalse(channelMock.isFlushed());
   stream.flush();
   assertTrue(channelMock.isFlushed());
   // close
   stream.close();
   assertTrue(channelMock.isShutdownWrites());
   IOException expected = null;
   try {
     stream.write('d');
   } catch (IOException e) {
     expected = e;
   }
   assertNotNull(expected);
   expected = null;
   try {
     stream.write("efg".getBytes("UTF-8"));
   } catch (IOException e) {
     expected = e;
   }
   assertNotNull(expected);
   stream.flush();
   assertWrittenMessage(channelMock, "abc");
   // idempotent
   stream.close();
   expected = null;
   try {
     stream.write('h');
   } catch (IOException e) {
     expected = e;
   }
   assertNotNull(expected);
   expected = null;
   try {
     stream.write("ijk".getBytes("UTF-8"));
   } catch (IOException e) {
     expected = e;
   }
   assertNotNull(expected);
   stream.flush();
   assertWrittenMessage(channelMock, "abc");
 }
コード例 #2
0
 @Test
 public void closeEmptyStream() throws IOException {
   final ConnectedStreamChannelMock channelMock = new ConnectedStreamChannelMock();
   final ChannelOutputStream delegateStream = new ChannelOutputStream(channelMock);
   final LimitedOutputStream stream = new LimitedOutputStream(delegateStream, 5);
   stream.close();
   assertTrue(channelMock.isShutdownWrites());
   IOException expected = null;
   try {
     stream.write('a');
   } catch (IOException e) {
     expected = e;
   }
   assertNotNull(expected);
   expected = null;
   try {
     stream.write("bcd".getBytes("UTF-8"));
   } catch (IOException e) {
     expected = e;
   }
   assertNotNull(expected);
   stream.flush();
   assertWrittenMessage(channelMock);
   // idempotent
   stream.close();
   expected = null;
   try {
     stream.write('e');
   } catch (IOException e) {
     expected = e;
   }
   assertNotNull(expected);
   expected = null;
   try {
     stream.write("fgh".getBytes("UTF-8"));
   } catch (IOException e) {
     expected = e;
   }
   assertNotNull(expected);
   stream.flush();
   assertWrittenMessage(channelMock);
 }