Move django tests out of the example app

Author: gunthercox

.travis.yml

@@ -28,8 +28,9 @@ services:
   - mongodb
 
 script:
+  - python runtests.py
   - python examples/django_app/manage.py test examples/django_app/
-  - nosetests --with-coverage --cover-package=chatterbot
+  - nosetests tests/ --with-coverage --cover-package=chatterbot
   - python tests/benchmarks.py
 
 after_success:

examples/django_app/tests/test_example.py

@@ -0,0 +1,92 @@
+import json
+from django.test import TestCase
+from django.core.urlresolvers import reverse
+from django.utils.encoding import force_text
+from chatterbot.ext.django_chatterbot.views import ChatterBotView
+
+
+class ApiTestCase(TestCase):
+
+    def setUp(self):
+        super(ApiTestCase, self).setUp()
+        self.api_url = reverse('chatterbot:chatterbot')
+
+    def test_post(self):
+        """
+        Test that a response is returned.
+        """
+        data = {
+            'text': 'How are you?'
+        }
+        response = self.client.post(
+            self.api_url,
+            data=json.dumps(data),
+            content_type='application/json',
+            format='json'
+        )
+
+        self.assertEqual(response.status_code, 200)
+        self.assertIn('text', str(response.content))
+        self.assertIn('in_response_to', str(response.content))
+
+    def test_post_extra_data(self):
+        post_data = {
+            'text': 'Good morning.',
+            'extra_data': {
+                'user': 'jen@example.com'
+            }
+        }
+        response = self.client.post(
+            self.api_url,
+            data=json.dumps(post_data),
+            content_type='application/json',
+            format='json'
+        )
+
+        self.assertEqual(response.status_code, 200)
+        self.assertIn('text', str(response.content))
+        self.assertIn('extra_data', str(response.content))
+        self.assertIn('in_response_to', str(response.content))
+
+
+class ApiIntegrationTestCase(TestCase):
+
+    def setUp(self):
+        super(ApiIntegrationTestCase, self).setUp()
+        self.api_url = reverse('chatterbot:chatterbot')
+
+        # Clear the response queue before tests
+        ChatterBotView.chatterbot.conversation_sessions.get_default().conversation.flush()
+
+    def tearDown(self):
+        super(ApiIntegrationTestCase, self).tearDown()
+
+        # Clear the response queue after tests
+        ChatterBotView.chatterbot.conversation_sessions.get_default().conversation.flush()
+
+    def _get_json(self, response):
+        return json.loads(force_text(response.content))
+
+    def test_get_conversation_empty(self):
+        response = self.client.get(self.api_url)
+        data = self._get_json(response)
+
+        self.assertIn('conversation', data)
+        self.assertEqual(len(data['conversation']), 0)
+
+    def test_get_conversation(self):
+        response = self.client.post(
+            self.api_url,
+            data=json.dumps({'text': 'How are you?'}),
+            content_type='application/json',
+            format='json'
+        )
+
+        response = self.client.get(self.api_url)
+        data = self._get_json(response)
+
+        self.assertIn('conversation', data)
+        self.assertEqual(len(data['conversation']), 1)
+        self.assertEqual(len(data['conversation'][0]), 2)
+        self.assertIn('text', data['conversation'][0][0])
+        self.assertIn('text', data['conversation'][0][1])

runtests.py

@@ -0,0 +1,20 @@
+#!/usr/bin/env python
+
+"""
+This is the test runner for the ChatterBot's Django tests.
+"""
+
+import os
+import sys
+
+import django
+from django.conf import settings
+from django.test.utils import get_runner
+
+if __name__ == '__main__':
+    os.environ['DJANGO_SETTINGS_MODULE'] = 'tests_django.test_settings'
+    django.setup()
+    TestRunner = get_runner(settings)
+    test_runner = TestRunner()
+    failures = test_runner.run_tests(['tests_django'])
+    sys.exit(bool(failures))

tests_django/__init__.py

@@ -7,7 +7,7 @@ class ApiTestCase(TestCase):
 
     def setUp(self):
         super(ApiTestCase, self).setUp()
-        self.api_url = reverse('chatterbot:chatterbot')
+        self.api_url = reverse('chatterbot')
 
     def test_post(self):
         """

examples/django_app/tests/test_api.py tests_django/test_api.pyexamples/django_app/tests/test_api.py renamed to tests_django/test_api.py

@@ -12,10 +12,10 @@ def test_modified_settings(self):
     def test_name_setting(self):
         from django.core.urlresolvers import reverse
 
-        api_url = reverse('chatterbot:chatterbot')
+        api_url = reverse('chatterbot')
         response = self.client.get(api_url)
 
         self.assertEqual(response.status_code, 405)
         self.assertIn('detail', str(response.content))
         self.assertIn('name', str(response.content))
-        self.assertIn('Django ChatterBot Example', str(response.content))
+        self.assertIn('Test Django ChatterBot', str(response.content))

…amples/django_app/tests/test_settings.py tests_django/test_chatterbot_settings.pyexamples/django_app/tests/test_settings.py renamed to tests_django/test_chatterbot_settings.py examples/django_app/tests/test_settings.py renamed to tests_django/test_chatterbot_settings.py

@@ -10,7 +10,7 @@ class ApiIntegrationTestCase(TestCase):
 
     def setUp(self):
         super(ApiIntegrationTestCase, self).setUp()
-        self.api_url = reverse('chatterbot:chatterbot')
+        self.api_url = reverse('chatterbot')
 
         # Clear the response queue before tests
         ChatterBotView.chatterbot.conversation_sessions.get_default().conversation.flush()

…amples/django_app/tests/test_commands.py tests_django/test_commands.pyexamples/django_app/tests/test_commands.py renamed to tests_django/test_commands.py examples/django_app/tests/test_commands.py renamed to tests_django/test_commands.py

@@ -0,0 +1,43 @@
+"""
+Test Django settings
+"""
+import os
+
+BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
+
+SECRET_KEY = 'fake-key'
+
+INSTALLED_APPS = [
+    'django.contrib.auth',
+    'django.contrib.contenttypes',
+    'django.contrib.sessions',
+    'chatterbot.ext.django_chatterbot',
+    'tests_django',
+]
+
+CHATTERBOT = {
+    'name': 'Test Django ChatterBot',
+    'trainer': 'chatterbot.trainers.ChatterBotCorpusTrainer',
+    'training_data': [
+        'chatterbot.corpus.english.greetings'
+    ]
+}
+
+ROOT_URLCONF = 'chatterbot.ext.django_chatterbot.urls'
+
+MIDDLEWARE_CLASSES = (
+    'django.contrib.sessions.middleware.SessionMiddleware',
+    'django.middleware.common.CommonMiddleware',
+    'django.contrib.auth.middleware.AuthenticationMiddleware',
+    'django.contrib.auth.middleware.SessionAuthenticationMiddleware',
+    'django.contrib.messages.middleware.MessageMiddleware',
+    'django.middleware.clickjacking.XFrameOptionsMiddleware',
+    'django.middleware.security.SecurityMiddleware',
+)
+
+DATABASES = {
+    'default': {
+        'ENGINE': 'django.db.backends.sqlite3',
+        'NAME': os.path.join(BASE_DIR, 'db.sqlite3'),
+    }
+}