We now have a model where we can save data

zkanda · zkanda · commit 0e4cb8008c09 · 2017-02-24T23:53:26.000+08:00
diff --git a/PROCESS.md b/PROCESS.md
@@ -143,3 +143,15 @@ Now we are done with coding the send_sms page with form.
 Let's go to the view function for send_sms, it's time for unit test.
 
 Added the test and mocking for send_sms view
+
+
+5 Saving user input
+
+Lets create the table where the data would be viewed.
+
+Now we create a database called Messages
+
+Then create a unit test to create that model
+
+
+
diff --git a/functional_test.py b/functional_test.py
@@ -81,18 +81,21 @@ def test_can_view_the_page(self):
 
         # He still didn't see any reply.
         table_replies = self.browser.find_element_by_id("id_table_replies")
-        rows = table_replies.find_element_by_tag_name("tr")
+        rows = table_replies.find_elements_by_tag_name("tr")
         self.assertTrue(
             any(row.text == "" for row in rows)
         )
 
         # He got a notification from his phone, check and recieve the message
         # that he sent earlier. He immediately replied:
         # Elpedio Adoptante
+        self.assertEqual(self.reply_from_chikka().status_code, 200)
+
+        time.sleep(2)
 
         # He tries to refresh the page and saw the message like this.
         # Number        Name
-        # 09152087801   JayR
+        # 09152087801   Elpedio Adoptante
         self.browser.refresh()
         table_replies = self.browser.find_element_by_id("id_table_replies")
         rows = table_replies.find_element_by_tag_name("tr")
@@ -102,6 +105,15 @@ def test_can_view_the_page(self):
 
 
         # He is now happy and closes the browser.
+    
+
+    def reply_from_chikka(self):
+        import requests
+        return requests.post("http://localhost:8000/chikka_receiver/", data={
+            "mobile_number": "0945983495439",
+            "message": "Elpedio Adoptante"
+        })
+
 
 
 if __name__ == '__main__':
diff --git a/sms_survey/engagement/migrations/0001_initial.py b/sms_survey/engagement/migrations/0001_initial.py
@@ -0,0 +1,24 @@
+# -*- coding: utf-8 -*-
+# Generated by Django 1.10.5 on 2017-02-24 15:45
+from __future__ import unicode_literals
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    initial = True
+
+    dependencies = [
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='Messages',
+            fields=[
+                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('mobile_number', models.CharField(max_length=20)),
+                ('message', models.TextField()),
+            ],
+        ),
+    ]
diff --git a/sms_survey/engagement/models.py b/sms_survey/engagement/models.py
@@ -1,3 +1,7 @@
 from django.db import models
 
 # Create your models here.
+
+class Messages(models.Model):
+    mobile_number = models.CharField(max_length=20)
+    message = models.TextField()
diff --git a/sms_survey/engagement/templates/index.html b/sms_survey/engagement/templates/index.html
@@ -3,5 +3,10 @@
     <body>
         <header>Surveyor</header>
         <a href="/send_sms/" id="id_send_sms_page">Want to send SMS, click here!</a>
+
+        <table id="id_table_replies">
+            <tr></tr>
+        </table>
+
     </body>
 </html>
diff --git a/sms_survey/engagement/tests.py b/sms_survey/engagement/tests.py
@@ -3,6 +3,7 @@
 from django.test import TestCase
 from django.http import HttpRequest
 from .views import index
+from .models import Messages
 
 
 class HomePageTest(TestCase):
@@ -40,3 +41,34 @@ def test_chikka_api_is_called(self, mock_post):
 
         # send chikka
         mock_post.assert_called()
+
+class MessageModelTest(TestCase):
+    
+    def setUp(self):
+        Messages.objects.create(
+            id=1,
+            mobile_number="9847543895",
+            message="This is a message",
+        )
+
+    def test_can_save_data(self):
+        # We can save data
+        m = Messages.objects.all()
+        self.assertTrue(len(m) > 0)
+
+    def test_can_retreive_data(self):
+        # Retreive data
+        m = Messages.objects.get(pk=1)
+        self.assertEqual(m.mobile_number, "9847543895")
+
+class ChikkaReceiverTest(TestCase):
+
+    def test_should_return_200(self):
+        # it should return 200
+        response = self.client.post("/chikka_receiver/", data={
+            "mobile_number": "0945983495439",
+            "message": "Elpedio Adoptante"
+        })
+        self.assertEqual(response.status_code, 200)
+
+
diff --git a/sms_survey/engagement/views.py b/sms_survey/engagement/views.py
@@ -1,5 +1,6 @@
 import requests
 from django.shortcuts import render, redirect
+# from .models import Messages
 
 # Create your views here.
 def  index(request):
@@ -17,6 +18,11 @@ def send_sms(request):
     return render(request, 'send_sms.html')
 
 
+def chikka_receiver(request):
+    from django.http import HttpResponse
+    return HttpResponse({"message": "Success"})
+
+
 def chikka_proxy(request):
     from django.http import HttpResponse
     return HttpResponse({"message": "Success"})
diff --git a/sms_survey/sms_survey/urls.py b/sms_survey/sms_survey/urls.py
@@ -15,11 +15,12 @@
 """
 from django.conf.urls import url
 
-from engagement.views import index, send_sms, chikka_proxy
+from engagement.views import index, send_sms, chikka_proxy, chikka_receiver
 
 urlpatterns = [
     url(r'^$', index, name='index'),
     url(r'^send_sms/$', send_sms, name='send_sms'),
+    url(r'^chikka_receiver/$', chikka_receiver, name='chikka_receiver'),
     
 
     # This is only need for testing locally
