blob: 628b5dd63fdbb89463e365ff718ccc1f4905f5d7 (
plain)
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
|
package org.traccar.helper;
import org.junit.Before;
import org.junit.Test;
import org.springframework.mock.web.MockHttpServletRequest;
import static org.junit.Assert.assertEquals;
public class IpRetrieverTest {
private static final String NORMAL_IP_ADDRESS = "231.23.45.65";
private static final String GATEWAY_IP_ADDRESS = "147.120.1.5";
private static final String IP_ADDRESS_BEHIND_REVERSE_PROXY = "231.23.45.65, 10.20.10.33, 10.20.20.34";
private MockHttpServletRequest mockHttpServletRequest;
@Before
public void init(){
mockHttpServletRequest = new MockHttpServletRequest();
}
@Test
public void testIpBehindReverseProxy(){
mockHttpServletRequest.setRemoteAddr(GATEWAY_IP_ADDRESS);
mockHttpServletRequest.addHeader("X-FORWARDED-FOR",IP_ADDRESS_BEHIND_REVERSE_PROXY);
assertEquals(NORMAL_IP_ADDRESS, IpRetriever.retrieveIP(mockHttpServletRequest));
}
@Test
public void testNormalIp(){
mockHttpServletRequest.setRemoteAddr(NORMAL_IP_ADDRESS);
assertEquals(NORMAL_IP_ADDRESS, IpRetriever.retrieveIP(mockHttpServletRequest));
}
}
|
