Add more tests

Author: ElusiveSpirit

.travis.yml

@@ -1,5 +1,12 @@
 language: python
 python:
-  - "3.5"
-install: "pip install -r requirements.conf"
+- '3.5'
+install: pip install -r requirements.conf
 script: python manage.py test
+deploy:
+  provider: pypi
+  user: Solleks
+  password:
+    secure: irXWJnMvz8xoOo7N9p8nac2wyHB8x5J0mAo9NvfX+Jv6T3+KfmAOOLMYoMBhmtxlc60tQV9rkUIG2+GDQoWK0gQXIumPfChBKELWfz4d5D73q/NHgnsiwGP4T4SM3ow3Z3uKJFLe20R4jxILwCbBQQmZ2vZoeGOJpEsGYj+bhsHNGRx5mb/Gtk2X0+tFA5cZsd1FMvnOige89E5SQds8QI//k+eErpgk0KORBzhOGDbgrdAJWcNOA2wFZIrqgMZyjZ+NhFXAyxS7BzWM8zwfFUWr5890GphH8igP/PQIvbtjCMu+x8kLMQr6S8RmeDFka6POHCYaLQOoGOVo4/r9+aVAXFd+Js+caJcmo+TWSqb1gy+mNzch1OHHGVz8WRbpvHSwzYCd35x5ScVQJqNVB2Cs4w3W/7yuoCRNvXXU/bZptVwQnukFUvlBA/2KidsXHCVPu+psFpN6r5FVKRtcyZgPW86dAqEqUqPezVYizIEXVEFpMv1EXz8zpb3rafow75sSSKz8qADu24Q4LCBkQaDyStrAw0ldyLBf3rRVZGtgeFtp3EWkhY3NCzsNPfd6AWXN3xqIcNo3n9LmzX2CTKUzJT1Z5NzaQcIDBHDTSWQrIKKMZ1aEWf4t3OZO9ehEGnYIHxesuPCggT1lfysFylHVJo+GwgId1otF53Ptg4M=
+  on:
+    branch: production

README.md

@@ -41,7 +41,7 @@ By default server is unblocked
 
 Usage
 -----
-You save your django SECRET_KEY from settings
+You need to save your django SECRET_KEY from settings first
 ```
 SECRET_KEY = '...
 ```

django_dev_protector/tests.py

@@ -1,8 +1,11 @@
 import json
+import os
+
 from django.test import TestCase, Client
 from django.conf import settings
 
 import django_dev_protector.settings
+from .settings import PROTECT_STATUS_VARIABLE
 from django_dev_protector.setup import get_status
 
 
@@ -18,6 +21,15 @@ def test_block_the_server(self):
         }), content_type='application/json')
         self.assertEqual(get_status(), 'True')
 
+    def test_is_server_blocked(self):
+        response = self.client.post('/django_dev_protector/', json.dumps({
+            'key': settings.SECRET_KEY,
+            'status': True
+        }), content_type='application/json')
+        response = self.client.get('/')
+        self.assertEqual(get_status(), 'True')
+        self.assertContains(response, 'The work was not paid.')
+
     def test_unblock_the_server(self):
         """
         Unblocks server by request
@@ -27,3 +39,13 @@ def test_unblock_the_server(self):
             'status': False
         }), content_type='application/json')
         self.assertEqual(get_status(), 'False')
+        self.assertEqual(os.environ[PROTECT_STATUS_VARIABLE], 'False')
+
+    def test_is_server_unblocked(self):
+        response = self.client.post('/django_dev_protector/', json.dumps({
+            'key': settings.SECRET_KEY,
+            'status': False
+        }), content_type='application/json')
+        response = self.client.get('/')
+        self.assertEqual(get_status(), 'False')
+        self.assertContains(response, 'works!')

manage.py

@@ -3,7 +3,7 @@
 import sys
 
 if __name__ == "__main__":
-    os.environ.setdefault("DJANGO_SETTINGS_MODULE", "tests.settings")
+    os.environ.setdefault("DJANGO_SETTINGS_MODULE", "test_app.settings")
     try:
         from django.core.management import execute_from_command_line
     except ImportError:

tests/__init__.py renamed to test_app/__init__.py

@@ -36,7 +36,7 @@
     'django.middleware.clickjacking.XFrameOptionsMiddleware',
 ]
 
-ROOT_URLCONF = 'tests.urls'
+ROOT_URLCONF = 'test_app.urls'
 
 TEMPLATES = [
     {
@@ -54,7 +54,7 @@
     },
 ]
 
-WSGI_APPLICATION = 'tests.wsgi.application'
+WSGI_APPLICATION = 'test_app.wsgi.application'
 
 
 DATABASES = {

tests/settings.py renamed to test_app/settings.py

@@ -0,0 +1,13 @@
+from django.http import JsonResponse
+from django.conf.urls import url
+from django.contrib import admin
+
+
+def index(request):
+    return JsonResponse({'site': 'works!'})
+
+
+urlpatterns = [
+    url(r'^admin/', admin.site.urls),
+    url(r'^$', index)
+]

test_app/urls.py

@@ -11,6 +11,6 @@
 
 from django.core.wsgi import get_wsgi_application
 
-os.environ.setdefault("DJANGO_SETTINGS_MODULE", "tests.settings")
+os.environ.setdefault("DJANGO_SETTINGS_MODULE", "test_app.settings")
 
 application = get_wsgi_application()