forked from scribejava/scribejava
-
Notifications
You must be signed in to change notification settings - Fork 7
/
ConnectionStub.java
86 lines (69 loc) · 1.57 KB
/
ConnectionStub.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
package org.scribe.model;
import java.io.*;
import java.net.*;
import java.util.*;
public class ConnectionStub extends HttpURLConnection
{
private Map<String, String> headers = new HashMap<String, String>();
private Map<String, List<String>> responseHeaders = new HashMap<String, List<String>>();
private int inputStreamCalled = 0;
public ConnectionStub() throws Exception
{
super(new URL("http://example.com"));
}
@Override
public void setRequestProperty(String key, String value)
{
headers.put(key, value);
}
@Override
public String getRequestProperty(String s)
{
return headers.get(s);
}
public Map<String, String> getHeaders()
{
return headers;
}
@Override
public int getResponseCode() throws IOException
{
return 200;
}
@Override
public InputStream getInputStream() throws IOException
{
inputStreamCalled++;
return new ByteArrayInputStream("contents".getBytes());
}
public int getTimesCalledInpuStream()
{
return inputStreamCalled;
}
@Override
public OutputStream getOutputStream() throws IOException
{
ByteArrayOutputStream baos = new ByteArrayOutputStream();
baos.write("contents".getBytes());
return baos;
}
@Override
public Map<String, List<String>> getHeaderFields()
{
return responseHeaders;
}
public void addResponseHeader(String key, String value)
{
responseHeaders.put(key, Arrays.asList(value));
}
public void connect() throws IOException
{
}
public void disconnect()
{
}
public boolean usingProxy()
{
return false;
}
}