Practice-7-Done

docker/.python_history

@@ -0,0 +1,5 @@
+from models.hero import Hero
+from hero.models import Hero
+Hero.objects.all()
+exit
+exit()

toh/hero/admin.py

@@ -1,3 +1,5 @@
 from django.contrib import admin
+from .models import Hero
 
 # Register your models here.
+admin.site.register(Hero)

toh/hero/tests.py

@@ -1,3 +1,28 @@
-from django.test import TestCase
+from django.test import TestCase, Client
+from .models import Hero
 
 # Create your tests here.
+class HeroTestCase(TestCase):
+    def setUp(self):
+        Hero.objects.create(name='Superman')
+        Hero.objects.create(name='Batman')
+        Hero.objects.create(name='Ironman')
+
+    def test_hero_count(self):
+        self.assertEqual(Hero.objects.all().count(),3)
+
+    def test_hero_id(self):
+        client=Client()
+        response = client.get('/hero/10/')
+
+        self.assertEqual(response.status_code,200)
+        self.assertIn('10',response.content.decode())
+
+    def test_hero_index(self):
+        client = Client()
+        response = client.get('/hero/')
+        self.assertEqual(response.status_code,200)
+        self.assertIn('1',response.content.decode())
+        response = client.get('/hero/')
+        self.assertEqual(response.status_code,200)
+        self.assertIn('2',response.content.decode())

toh/hero/urls.py

@@ -2,8 +2,10 @@
 from . import views
 
 urlpatterns = [
-    path('', views.hero_list),
+    path('',views.index),
+    path('list/', views.hero_list),
     path('<int:id>/', views.id, name='hero_id'),
+    path('token/',views.token),
     path('<str:name>/', views.name, name='hero_name'),
     path('info/<int:id>/', views.hero_info, name='hero_info'),
 ]

toh/hero/views.py

@@ -3,17 +3,28 @@
 import json
 from json.decoder import JSONDecodeError
 from .models import Hero
+from django.views.decorators.csrf import ensure_csrf_cookie
 
-# def index(request):
-#     return HttpResponse('Hello, world!')
+@ensure_csrf_cookie
+def token(request):
+    if request.method=='GET':
+        return HttpResponse(status=204)
+    else:
+        return HttpResponseNotAllowed(['GET'])
+
+def index(request):
+    if 'visit_count' not in request.session:
+        request.session['visit_count']=1
+    else:
+        request.session['visit_count']+=1
+    return HttpResponse('Hello World! You visited {} times\n'.format(request.session['visit_count']))
 
 def id(request, id):
     return HttpResponse(f'Your id is {id}!')
 
 def name(request, name):
     return HttpResponse(f'Your name is {name}!')
 
-@csrf_exempt
 def hero_list(request):
     if request.method == 'GET':
         hero_all_list = [hero for hero in Hero.objects.all().values()]