01: /*
02: * $HeadURL: https://svn.apache.org/repos/asf/httpcomponents/httpcore/tags/4.0-beta1/module-nio/src/test/java/org/apache/http/mockup/ReadableByteChannelMockup.java $
03: * $Revision: 613298 $
04: * $Date: 2008-01-18 23:09:22 +0100 (Fri, 18 Jan 2008) $
05: *
06: * ====================================================================
07: * Licensed to the Apache Software Foundation (ASF) under one
08: * or more contributor license agreements. See the NOTICE file
09: * distributed with this work for additional information
10: * regarding copyright ownership. The ASF licenses this file
11: * to you under the Apache License, Version 2.0 (the
12: * "License"); you may not use this file except in compliance
13: * with the License. You may obtain a copy of the License at
14: *
15: * http://www.apache.org/licenses/LICENSE-2.0
16: *
17: * Unless required by applicable law or agreed to in writing,
18: * software distributed under the License is distributed on an
19: * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
20: * KIND, either express or implied. See the License for the
21: * specific language governing permissions and limitations
22: * under the License.
23: * ====================================================================
24: *
25: * This software consists of voluntary contributions made by many
26: * individuals on behalf of the Apache Software Foundation. For more
27: * information on the Apache Software Foundation, please see
28: * <http://www.apache.org/>.
29: *
30: */
31:
32: package org.apache.http.mockup;
33:
34: import java.io.IOException;
35: import java.nio.ByteBuffer;
36: import java.nio.channels.ReadableByteChannel;
37:
38: import org.apache.http.util.EncodingUtils;
39:
40: public class ReadableByteChannelMockup implements ReadableByteChannel {
41:
42: private final String[] chunks;
43: private final String charset;
44:
45: private int chunkCount = 0;
46:
47: private ByteBuffer currentChunk;
48: private boolean closed = false;
49:
50: public ReadableByteChannelMockup(final String[] chunks,
51: final String charset) {
52: super ();
53: this .chunks = chunks;
54: this .charset = charset;
55: }
56:
57: private void prepareChunk() {
58: if (this .currentChunk == null
59: || !this .currentChunk.hasRemaining()) {
60: if (this .chunkCount < this .chunks.length) {
61: String s = this .chunks[this .chunkCount];
62: this .chunkCount++;
63: this .currentChunk = ByteBuffer.wrap(EncodingUtils
64: .getBytes(s, this .charset));
65: } else {
66: this .closed = true;
67: }
68: }
69: }
70:
71: public int read(final ByteBuffer dst) throws IOException {
72: prepareChunk();
73: if (this .closed) {
74: return -1;
75: }
76: int i = 0;
77: while (dst.hasRemaining() && this .currentChunk.hasRemaining()) {
78: dst.put(this .currentChunk.get());
79: i++;
80: }
81: return i;
82: }
83:
84: public void close() throws IOException {
85: this .closed = true;
86: }
87:
88: public boolean isOpen() {
89: return !this.closed;
90: }
91:
92: }
|