Add test cases

requirements.txt

@@ -3,4 +3,6 @@ Flask==0.11.1
 itsdangerous==0.24
 Jinja2==2.8
 MarkupSafe==0.23
+py==1.4.31
+pytest==3.0.3
 Werkzeug==0.11.11

test_server.py

@@ -0,0 +1,32 @@
+# -*- coding: utf-8 -*-
+
+import pytest
+import server
+
+@pytest.fixture
+def client(request):
+    server.app.config['TESTING'] = True
+    client = server.app.test_client()
+
+    def teardown():
+        pass
+
+    request.addfinalizer(teardown)
+
+    return client
+
+
+def post_webhook_from_trusted_remote_addr(client):
+    return client.post('/webhook', environ_base={'REMOTE_ADDR': '111.11.1.1'})
+
+
+def post_test_route(client):
+    return client.post('/teset')
+
+
+def test_limit_remote_addr(client):
+    rv = post_webhook_from_trusted_remote_addr(client)
+    assert rv.status_code == 403
+
+    rv = post_test_route(client)
+    assert rv.status_code != 403