X-Git-Url: https://git.mdrn.pl/fnpdjango.git/blobdiff_plain/e4707d78bf213a6634de0a8ad6ec0c2ebdd6de82..28b1d7e9f4a7008712736adce363ca2770d48aea:/tests/tests/test_middleware.py

diff --git a/tests/tests/test_middleware.py b/tests/tests/test_middleware.py
new file mode 100644
index 0000000..0ba5966
--- /dev/null
+++ b/tests/tests/test_middleware.py
@@ -0,0 +1,16 @@
+from django.test import TestCase
+
+
+class MiddlewareTests(TestCase):
+    def test_realip(self):
+        self.assertEqual(
+            self.client.get('/ip/').content.decode('latin1'),
+            '127.0.0.1'
+        )
+        self.assertEqual(
+            self.client.get(
+                '/ip/',
+                HTTP_X_REAL_IP='1.2.3.4'
+            ).content.decode('latin1'),
+            '1.2.3.4'
+        )