1 /* 2 * MIT License 3 * 4 * Copyright (c) 2010-2020 The Waffle Project Contributors: https://github.com/Waffle/waffle/graphs/contributors 5 * 6 * Permission is hereby granted, free of charge, to any person obtaining a copy 7 * of this software and associated documentation files (the "Software"), to deal 8 * in the Software without restriction, including without limitation the rights 9 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell 10 * copies of the Software, and to permit persons to whom the Software is 11 * furnished to do so, subject to the following conditions: 12 * 13 * The above copyright notice and this permission notice shall be included in all 14 * copies or substantial portions of the Software. 15 * 16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 17 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 18 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE 19 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 20 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, 21 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE 22 * SOFTWARE. 23 */ 24 package waffle.shiro.negotiate; 25 26 import java.io.IOException; 27 import java.util.ArrayList; 28 import java.util.List; 29 import java.util.Map; 30 31 import javax.servlet.http.HttpServletResponse; 32 33 /** 34 * The Class MockServletResponse. 35 */ 36 public abstract class MockServletResponse implements HttpServletResponse { 37 38 /** The is flushed. */ 39 boolean isFlushed; 40 41 /** The error code. */ 42 int errorCode; 43 44 /** The headers. */ 45 Map<String, String> headers; 46 47 /** The headers added. */ 48 Map<String, List<String>> headersAdded; 49 50 /** The sc. */ 51 int sc; 52 53 @Override 54 public void addHeader(final String name, final String value) { 55 if (this.headersAdded.containsKey(name)) { 56 this.headersAdded.get(name).add(value); 57 return; 58 } 59 60 final List<String> values = new ArrayList<>(); 61 values.add(value); 62 this.headersAdded.put(name, values); 63 } 64 65 @Override 66 public void flushBuffer() throws IOException { 67 this.isFlushed = true; 68 } 69 70 @Override 71 public void sendError(final int sendError) throws IOException { 72 this.errorCode = sendError; 73 } 74 75 @Override 76 public void setHeader(final String name, final String value) { 77 this.headers.put(name, value); 78 } 79 80 @Override 81 public void setStatus(final int status) { 82 this.sc = status; 83 } 84 85 }