1 /*
2 * Licensed to the Apache Software Foundation (ASF) under one or more
3 * contributor license agreements. See the NOTICE file distributed with
4 * this work for additional information regarding copyright ownership.
5 * The ASF licenses this file to You under the Apache License, Version 2.0
6 * (the "License"); you may not use this file except in compliance with
7 * the License. You may obtain a copy of the License at
8 *
9 * http://www.apache.org/licenses/LICENSE-2.0
10 *
11 * Unless required by applicable law or agreed to in writing, software
12 * distributed under the License is distributed on an "AS IS" BASIS,
13 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 * See the License for the specific language governing permissions and
15 * limitations under the License.
16 */
17 package org.apache.juneau.utest.utils;
18
19 import java.io.*;
20
21 import org.apache.juneau.*;
22 import org.apache.juneau.common.utils.*;
23 import org.apache.juneau.internal.*;
24 import org.apache.juneau.parser.*;
25
26 /**
27 * Utility class for creating mocked stream parser.
28 */
29 public class FakeStreamParser extends InputStreamParser {
30
31 //-------------------------------------------------------------------------------------------------------------------
32 // Static
33 //-------------------------------------------------------------------------------------------------------------------
34
35 public static Builder create() {
36 return new Builder();
37 }
38
39 //-------------------------------------------------------------------------------------------------------------------
40 // Builder
41 //-------------------------------------------------------------------------------------------------------------------
42
43 public static class Builder extends InputStreamParser.Builder {
44 Function3<InputStreamParserSession,byte[],ClassMeta<?>,Object> function;
45
46 public Builder function(Function3<InputStreamParserSession,byte[],ClassMeta<?>,Object> value) {
47 function = value;
48 return this;
49 }
50
51 @Override
52 public Builder consumes(String value) {
53 super.consumes(value);
54 return this;
55 }
56
57 @Override
58 public Builder copy() {
59 return this;
60 }
61 }
62
63 //-------------------------------------------------------------------------------------------------------------------
64 // Instance
65 //-------------------------------------------------------------------------------------------------------------------
66
67 private final Function3<InputStreamParserSession,byte[],ClassMeta<?>,Object> function;
68
69 public FakeStreamParser(Builder builder) {
70 super(builder);
71 this.function = builder.function;
72 }
73
74 @Override
75 public <T> T doParse(ParserSession session, ParserPipe pipe, ClassMeta<T> type) throws IOException, ParseException, ExecutableException {
76 if (function != null)
77 return type.cast(function.apply((InputStreamParserSession)session, IOUtils.readBytes(pipe.getInputStream()), type));
78 return null;
79 }
80 }