lab7

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,27 @@
-from django.test import TestCase
+from django.test import TestCase, Client
+from .models import Hero
 
 # Create your tests here.
+class HeroTestCase(TestCase):
+    # Before Each
+    def setUp(self):
+        Hero.objects.create(name='Superman')
+        Hero.objects.create(name='Batman')
+        Hero.objects.create(name='Itronman')
+
+    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_visit_counter(self):
+        client = Client()
+        response = client.get('/hero/')
+        self.assertIn('1', response.content.decode())
+        response = client.get('/hero/')
+        self.assertIn('2', response.content.decode())

toh/hero/urls.py

@@ -1,9 +1,11 @@
+from django.contrib import admin
 from django.urls import path
 from . import views
 
 urlpatterns = [
-    path('', views.hero_list),
+    path('', views.index),
     path('<int:id>/', views.id, name='hero_id'),
     path('<str:name>/', views.name, name='hero_name'),
     path('info/<int:id>/', views.hero_info, name='hero_info'),
+    path('admin/', admin.site.urls)
 ]

toh/hero/views.py

@@ -4,8 +4,12 @@
 from json.decoder import JSONDecodeError
 from .models import Hero
 
-# def index(request):
-#     return HttpResponse('Hello, world!')
+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}!')