1 files changed, 33 insertions, 0 deletions
diff --git a/core/src/test/java/fi/iki/elonen/HttpSessionHeadersTest.java b/core/src/test/java/fi/iki/elonen/HttpSessionHeadersTest.java
new file mode 100644
index 0000000..75a0a49
--- /dev/null
+++ b/core/src/test/java/fi/iki/elonen/HttpSessionHeadersTest.java
@@ -0,0 +1,33 @@
+package fi.iki.elonen;
+
+import org.junit.Test;
+
+import java.io.ByteArrayInputStream;
+import java.io.ByteArrayOutputStream;
+import java.net.InetAddress;
+
+import static org.junit.Assert.assertEquals;
+
+public class HttpSessionHeadersTest extends HttpServerTest {
+    private static final String DUMMY_REQUEST_CONTENT = "dummy request content";
+    private static final TestTempFileManager TEST_TEMP_FILE_MANAGER = new TestTempFileManager();
+
+    @Override
+    public void setUp() {
+        super.setUp();
+    }
+
+    @Test
+    public void testHeadersRemoteIp() throws Exception {
+        ByteArrayInputStream inputStream = new ByteArrayInputStream(DUMMY_REQUEST_CONTENT.getBytes());
+        ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
+        String[] ipAddresses = { "127.0.0.1", "192.168.1.1", "192.30.252.129" };
+        for(String ipAddress : ipAddresses) {
+            InetAddress inetAddress = InetAddress.getByName(ipAddress);
+            NanoHTTPD.HTTPSession session = testServer.createSession(TEST_TEMP_FILE_MANAGER, inputStream, outputStream, inetAddress);
+            assertEquals(ipAddress, session.getHeaders().get("remote-addr"));
+            assertEquals(ipAddress, session.getHeaders().get("http-client-ip"));
+        }
+    }
+
+}