diff --git a/virtualenv/lists/tests.py b/virtualenv/lists/tests.py index f918990fc5554ed79e7ea2c8eb6f6fbc999a1b6e..fe29807897f889fbb7c330dd9dee66d15e76b2f0 100644 --- a/virtualenv/lists/tests.py +++ b/virtualenv/lists/tests.py @@ -1,9 +1,21 @@ from django.urls import resolve from django.test import TestCase -from lists.views import home_page +from django.http import HttpRequest + +from lists.views import home_page + class HomePageTest(TestCase): def test_root_url_resolves_to_home_page_view(self): - found = resolve('/') - self.assertEqual(found.func, home_page) \ No newline at end of file + found = resolve('/') + self.assertEqual(found.func, home_page) + + + def test_home_page_returns_correct_html(self): + request = HttpRequest() + response = home_page(request) + html = response.content.decode('utf8') + self.assertTrue(html.startswith('<html>')) + self.assertIn('<title>To-Do lists</title>', html) + self.assertTrue(html.endswith('</html>')) \ No newline at end of file diff --git a/virtualenv/lists/views.py b/virtualenv/lists/views.py index 444abe40e5262a8f76393398d1a6a3287c31a17f..ec6a9bc35fd1974326e6e59363c99d430c696bd7 100644 --- a/virtualenv/lists/views.py +++ b/virtualenv/lists/views.py @@ -1,5 +1,5 @@ -from django.shortcuts import render +from django.http import HttpResponse # Create your views here. -def home_page(): - pass \ No newline at end of file +def home_page(request): + return HttpResponse('<html><title>To-Do lists</title></html>') \ No newline at end of file