Welcome to part 3 of the tutorial! This week we'll finally get into writing our own web pages, rather than using the Django Admin site.
Let's pick up our FT where we left off - we now have the admin site set up to add Polls, including Choices. We now want to flesh out what the user sees.
Let's start by writing out our FT as human-readable comments, which describe the user's actions, and the expected behaviour of the site
python
from functional_tests import FunctionalTest, ROOT from selenium.webdriver.common.keys import Keys
class TestPolls(FunctionalTest):
- def test_voting_on_a_new_poll(self):
# First, Gertrude the administrator logs into the admin site and # creates a couple of new Polls, and their response choices
# Now, Herbert the regular user goes to the homepage of the site. He # sees a list of polls.
# He clicks on the link to the first Poll, which is called # 'How awesome is test-driven development?'
# He is taken to a poll 'results' page, which says # "no-one has voted on this poll yet"
# He also sees a form, which offers him several choices. # He decided to select "very awesome"
# He clicks 'submit'
# The page refreshes, and he sees that his choice # has updated the results. they now say # "100 %: very awesome".
# The page also says "1 votes"
# Satisfied, he goes back to sleep
A nice little test, but that very first comment rather glosses over a lot. Let's split out the Gertrude bit into its own method, for tidiness, and copy and paste in some code from the admin test.
python
# First, Gertrude the administrator logs into the admin site and # creates a couple of new Polls, and their response choices self._setup_polls_via_admin()
And add the new method
#TODO: modify this to use workshop loop method to create many polls
python
class TestPolls(FunctionalTest):
- def _setup_polls_via_admin(self):
# Gertrude logs into the admin site self.browser.get(ROOT + '/admin/') username_field = self.browser.find_element_by_name('username') username_field.send_keys('admin') password_field = self.browser.find_element_by_name('password') password_field.send_keys('adm1n') password_field.send_keys(Keys.RETURN)
# She follows the link to the Polls app, and adds a new Poll self.browser.find_elements_by_link_text('Polls')[1].click() self.browser.find_element_by_link_text('Add poll').click()
# She enters its name, and uses the 'today' and 'now' buttons to set # the publish date question_field = self.browser.find_element_by_name('question') question_field.send_keys("How awesome is Test-Driven Development?") self.browser.find_element_by_link_text('Today').click() self.browser.find_element_by_link_text('Now').click()
# She sees she can enter choices for the Poll. She adds three choice_1 = self.browser.find_element_by_name('choice_set-0-choice') choice_1.send_keys('Very awesome') choice_2 = self.browser.find_element_by_name('choice_set-1-choice') choice_2.send_keys('Quite awesome') choice_3 = self.browser.find_element_by_name('choice_set-2-choice') choice_3.send_keys('Moderately awesome')
# She saves her new poll save_button = self.browser.find_element_by_css_selector("input[value='Save']") save_button.click()
# She is returned to the "Polls" listing, where she can see her # new poll, listed as a clickable link new_poll_links = self.browser.find_elements_by_link_text( "How awesome is Test-Driven Development?" ) self.assertEquals(len(new_poll_links), 1)
# She logs out of the admin site self.browser.find_element_by_link_text('Log out').click()
So, the next thing we want to do is write some of the test of what non-admin users will see of our polls application - viewing existing polls, responding to polls by submitting a choice, and viewing poll results.
Last time we wrote the code to get Gertrude the admin to log in and create a poll. Let's write the next bit, where Herbert the normal user opens up our website, sees some polls and votes on them.
So, let's write a functional test that does all three of those things. We'll create a new file for it called fts/test_polls.py
. We'll be re-using some of the code from test_admin.py
, so you might want to do a "save as" based on that file.
python
- def test_voting_on_a_new_poll(self):
# First, Gertrude the administrator logs into the admin site and # creates a couple of new Polls, and their response choices self._setup_polls_via_admin()
# Now, Herbert the regular user goes to the homepage of the site. He # sees a list of polls. self.browser.get(ROOT) heading = self.browser.find_element_by_tag_name('h1') self.assertEquals(heading.text, 'Polls')
# He clicks on the link to the first Poll, which is called # 'How awesome is test-driven development?' self.browser.find_element_by_link_text('How awesome is Test-Driven Development?').click()
# He is taken to a poll 'results' page, which says # "no-one has voted on this poll yet" heading = self.browser.find_element_by_tag_name('h1') self.assertEquals(heading.text, 'Poll Results') body = self.browser.find_element_by_tag_name('body') self.assertIn('No-one has voted on this poll yet', body.text)
Let's run that, and see where we get:
======================================================================
FAIL: test_voting_on_a_new_poll (test_polls.TestPolls)
----------------------------------------------------------------------
Traceback (most recent call last):
File "/home/harry/workspace/Test-Driven-Django-Tutorial/mysite/fts/test_polls.py", line 57, in test_voting_on_a_new_poll
self.assertEquals(heading.text, 'Polls')
AssertionError: u'Page not found (404)' != 'Polls'
----------------------------------------------------------------------
Ran 2 tests in 19.772s
The FT is telling us that going to the ROOT url (/) produces a 404. We need to tell Django what kind of web page to return for the root of our site - the home page if you like.
Django uses a file called urls.py
, to route visitors to the python function that will deal with producing a response for them. These functions are called views in Django terminology, and they live in views.py
. (This is essentially an MVC pattern, there's some discussion of it here: https://docs.djangoproject.com/en/dev/faq/general/#django-appears-to-be-a-mvc-framework-but-you-call-the-controller-the-view-and-the-view-the-template-how-come-you-don-t-use-the-standard-names)
Let's add a new test to tests.py
. I'm going to use the Django Test Client, which has some helpful features for testing views. More info here:
https://docs.djangoproject.com/en/1.3/topics/testing/
python
from django.test.client import Client [...]
- def test_root_url_shows_all_polls(self):
# set up some polls poll1 = Poll(question='6 times 7', pub_date='2001-01-01') poll1.save() poll2 = Poll(question='life, the universe and everything', pub_date='2001-01-01') poll2.save()
client = Client() response = client.get('/')
self.assertIn(poll1.question, response.content) self.assertIn(poll2.question, response.content)
Don't forget the import at the top! Now, our first run of the tests will probably complain of a with TemplateDoesNotExist: 404.html
. Django wants us to create a template for our "404 error" page. We'll come back to that later. For now, let's make the /
url return a real HTTP response.
First we'll create a dummy view in views.py
:
python
- def polls(request):
pass
Now let's hook up this view inside urls.py
:
python
from mysite.polls import views as polls_views
- urlpatterns = patterns('',
(r'^$', polls_views.polls), (r'^admin/', include(admin.site.urls)),
)
You may notice the slightly unorthodox import of polls.views
- the alternative is you can feed in views as strings to lines in urlpatterns
, without importing anything, like this:
python
(r'^$', 'mysite.polls_views.polls'),
I like my way because it uses the 'real' view - it requires that we actually have a view defined in views.py, and that it imports properly... But it's a personal preference!
Re-running our tests should show us a different error:
======================================================================
ERROR: test_root_url_shows_all_polls (polls.tests.TestAllPollsView)
----------------------------------------------------------------------
Traceback (most recent call last):
File "/home/harry/workspace/tddjango_site/source/mysite/polls/tests.py", line 92, in test_root_url_shows_all_polls
respoviense = client.get('/')
File "/usr/lib/pymodules/python2.7/django/test/client.py", line 445, in get
response = super(Client, self).get(path, data=data, **extra)
File "/usr/lib/pymodules/python2.7/django/test/client.py", line 229, in get
return self.request(**r)
File "/usr/lib/pymodules/python2.7/django/core/handlers/base.py", line 129, in get_response
raise ValueError("The view %s.%s didn't return an HttpResponse object." % (callback.__module__, view_name))
ValueError: The view mysite.polls.views.polls didn't return an HttpResponse object.
----------------------------------------------------------------------
Let's get the view to return an HttpResponse:
python
from django.http import HttpResponse
- def polls(request):
return HttpResponse()
The tests are now more instructive:
======================================================================
FAIL: test_root_url_shows_all_polls (polls.tests.TestAllPollsView)
----------------------------------------------------------------------
Traceback (most recent call last):
File "/home/harry/workspace/tddjango_site/source/mysite/polls/tests.py", line 96, in test_root_url_shows_all_polls
self.assertIn(poll1.question, response.content)
AssertionError: '6 times 7' not found in ''
----------------------------------------------------------------------
So far, we're returning a blank page. Now, to get the tests to pass, it would be simple enough to just return a response that contained the questions of our two polls as raw text - like this:
python
from django.http import HttpResponse from polls.models import Poll
- def polls(request):
content = '' for poll in Poll.objects.all(): content += poll.question
return HttpResponse(content)
Sure enough, that gets our limited unit tests passing:
23:06 ~/workspace/tddjango_site/source/mysite (master)$ python manage.py test polls
Creating test database for alias 'default'...
......
----------------------------------------------------------------------
Ran 6 tests in 0.009s
OK
Destroying test database for alias 'default'...
Now, this probably seems like a slightly artificial situation - for starters, the two poll's names will just be concatenated together, without even a space or a carriage return. We can't possibly leave the situation like this. But the point of TDD is to be driven by the tests. At each stage, we only write the code that our tests require, because that makes absolutely sure that we have tests for all of our code.
So, rather than anticipate what we might want to put in our HttpResponse, let's go to the FT now to see what to do next.:
python functional_tests.py
======================================================================
ERROR: test_voting_on_a_new_poll (test_polls.TestPolls)
----------------------------------------------------------------------
Traceback (most recent call last):
File "/home/harry/workspace/tddjango_site/source/mysite/fts/test_polls.py", line 57, in test_voting_on_a_new_poll
heading = self.browser.find_element_by_tag_name('h1')
File "/usr/local/lib/python2.7/dist-packages/selenium/webdriver/remote/webdriver.py", line 306, in find_element_by_tag_name
return self.find_element(by=By.TAG_NAME, value=name)
File "/usr/local/lib/python2.7/dist-packages/selenium/webdriver/remote/webdriver.py", line 637, in find_element
{'using': by, 'value': value})['value']
File "/usr/local/lib/python2.7/dist-packages/selenium/webdriver/remote/webdriver.py", line 153, in execute
self.error_handler.check_response(response)
File "/usr/local/lib/python2.7/dist-packages/selenium/webdriver/remote/errorhandler.py", line 123, in check_response
raise exception_class(message, screen, stacktrace)
NoSuchElementException: Message: u'Unable to locate element: {"method":"tag name","selector":"h1"}'
----------------------------------------------------------------------
Ran 2 tests in 29.119s
The FT wants an h1
heading tag on the page. Now, again, we could hard-code this into view (maybe starting with content = <h1>Polls</h1>
before the for
loop), but at this point it seems sensible to start to use Django's template system.
The Django Test Client lets us check whether a response was rendered using a template, by using a special attribute of the response called templates
, so let's use that. In tests.py
:
python
template_names_used = [t.name for t in response.templates] self.assertIn('polls.html', template_names_used)
self.assertIn(poll1.question, response.content) self.assertIn(poll2.question, response.content)
Testing python manage.py test polls
:
======================================================================
FAIL: test_root_url_shows_all_polls (polls.tests.TestAllPollsView)
----------------------------------------------------------------------
Traceback (most recent call last):
File "/home/harry/workspace/tddjango_site/source/mysite/polls/tests.py", line 94, in test_root_url_shows_all_polls
self.assertIn('polls.html', response.templates)
AssertionError: 'polls.html' not found in []
----------------------------------------------------------------------
Ran 6 tests in 0.009s
So let's now create our template:
mkdir mysite/polls/templates
touch mysite/polls/templates/polls.html
Edit polls.html
with your favourite editor,
html+django
- <html>
- <body>
<h1>Polls</h1> {% for poll in polls %} {{ poll.question }} {% endfor %}
</body>
</html>
You'll probably recognise this as being essentially standard HTML, intermixed with some special django control codes. These are either surrounded with {%
- %}
, for flow control - like a for loop in this case, and {{
- }}
for printing variables. You can find out more about the Django template language here:
https://docs.djangoproject.com/en/1.3/topics/templates/
Let's rewrite our code to use this template. For this we can use the Django
render
function, which takes the request and the name of the template:
python
from django.shortcuts import render from polls.models import Poll
- def polls(request):
content = '' for poll in Poll.objects.all(): content += poll.question
return render(request, 'polls.html')
Our last unit test error was that we weren't using a template - let's see if this fixes it:
======================================================================
FAIL: test_root_url_shows_all_polls (polls.tests.TestAllPollsView)
----------------------------------------------------------------------
Traceback (most recent call last):
File "/home/harry/workspace/tddjango_site/source/mysite/polls/tests.py", line 97, in test_root_url_shows_all_polls
self.assertIn(poll1.question, response.content)
AssertionError: '6 times 7' not found in '<html>\n <body>\n <h1>Polls</h1>\n \n </body>\n</html>\n'
----------------------------------------------------------------------
Sure does! Unfortunately, we've lost our Poll questions from the response content...
Looking at the template code, you can see that we want to iterate through a variable called polls
. The way we pass this into a template is via a special kind of dictionary called a context. The Django test client also lets us check on what context objects were used in rendering a response, so we can write a test for that too:
python
client = Client() response = client.get('/')
template_names_used = [t.name for t in response.templates] self.assertIn('polls.html', template_names_used)
polls_in_context = response.context['polls'] self.assertEquals(list(polls_in_context), [poll1, poll2])
self.assertIn(poll1.question, response.content) self.assertIn(poll2.question, response.content)
Now, re-running the tests gives us:
======================================================================
ERROR: test_root_url_shows_all_polls (polls.tests.TestAllPollsView)
----------------------------------------------------------------------
Traceback (most recent call last):
File "/home/harry/workspace/tddjango_site/source/mysite/polls/tests.py", line 97, in test_root_url_shows_all_polls
polls_in_context = response.context['polls']
File "/usr/lib/pymodules/python2.7/django/template/context.py", line 60, in __getitem__
raise KeyError(key)
KeyError: 'polls'
----------------------------------------------------------------------
Ran 6
Essentially, we never passed any 'polls' to our template. Let's add them, but make them empty - again, the idea is to make the minimal change to move the test forwards:
python
- def polls(request):
content = '' for poll in Poll.objects.all(): content += poll.question
context = {'polls': []} return render(request, 'polls.html', context)
Notice the way we've had to call list
on polls_in_context
- that's because Django queries return special QuerySet
objects, which, although they behave like lists, don't quite compare equal like them.
Now the unit tests say:
======================================================================
FAIL: test_root_url_shows_all_polls (polls.tests.TestAllPollsView)
----------------------------------------------------------------------
Traceback (most recent call last):
File "/home/harry/workspace/tddjango_site/source/mysite/polls/tests.py", line 98, in test_root_url_shows_all_polls
self.assertEquals(list(polls_in_context), [poll1, poll2])
AssertionError: Lists differ: [] != [<Poll: 6 times 7>, <Poll: lif...
Second list contains 2 additional elements.
First extra element 0:
6 times 7
- []
+ [<Poll: 6 times 7>, <Poll: life, the universe and everything>]
----------------------------------------------------------------------
Let's fix our code so the tests pass:
python
from django.shortcuts import render from polls.models import Poll
- def polls(request):
context = {'polls': Poll.objects.all()} return render(request, 'polls.html', context)
Ta-da!:
......
----------------------------------------------------------------------
Ran 6 tests in 0.011s
OK
What do the FTs say now?:
python functional_tests.py
======================================================================
ERROR: test_voting_on_a_new_poll (test_polls.TestPolls)
----------------------------------------------------------------------
Traceback (most recent call last):
File "/home/harry/workspace/tddjango_site/source/mysite/fts/test_polls.py", line 62, in test_voting_on_a_new_poll
self.browser.find_element_by_link_text('How awesome is Test-Driven Development?').click()
File "/usr/local/lib/python2.7/dist-packages/selenium/webdriver/remote/webdriver.py", line 234, in find_element_by_link_text
return self.find_element(by=By.LINK_TEXT, value=link_text)
File "/usr/local/lib/python2.7/dist-packages/selenium/webdriver/remote/webdriver.py", line 637, in find_element
{'using': by, 'value': value})['value']
File "/usr/local/lib/python2.7/dist-packages/selenium/webdriver/remote/webdriver.py", line 153, in execute
self.error_handler.check_response(response)
File "/usr/local/lib/python2.7/dist-packages/selenium/webdriver/remote/errorhandler.py", line 123, in check_response
raise exception_class(message, screen, stacktrace)
NoSuchElementException: Message: u'Unable to locate element: {"method":"link text","selector":"How awesome is Test-Driven Development?"}'
----------------------------------------------------------------------
Ah - although our page may contain the name of our Poll, it's not yet a link we can click.
The way we'd fix this is in the polls.html
template, by adding an <a href=
.
So is this something we write a unit test for as well? Some people would tend to say that this is one unit test too many... Since this is a guide to rigorous TDD, I'm going to say we probably should in this case.
On the other hand, if we write a unit test for every single last bit of html that we want to write, every last presentational detail, then making tiny tweaks to the UI is going to be really burdensome.
At this point, a couple of rules of thumb are useful:
- In unit tests, Don't test constants
- In functional tests, Test functionality, not presentation
The first rule works out like this - if we have some code that says:
wibble = 3
There's no point in writing a test that says:
self.assertEquals(wibble, 3)
Tests are meant to check how our code behaves, not just to repeat every line of it.
The second rule is a related rule, but it's more about how users interact with your software. We want our functional tests to check that the software allows the user to accomplish certain tasks. So, we need to check that each screen contains elements that can guide the user towards the choices they need to make (the link text), and also that they function in a way that moves the user towards their goal (our link, when clicked, will take the user to the right page).
What we definitely don't need to test in our FTs are things like - what specific colour are the links (although the fact that they are a different colour to something else may be relevant). We don't need to check the particular font they use. We don't need to check whether they are displayed in a ul
or in a table
- although we may want to check that they are displayed in the correct order.
So, where does that leave us? The FT currently checks the functionality of the site - it checks the link has the correct text, and later it checks that clicking the link takes us to the right place.
So, what about unit testing the templates? Well, most of what's in a template is just a constant - we don't want to have to rewrite our unit tests just because we want to correct a typo in a bit of blurb... The parts of a template that aren't "just a constant" are the bits inside {{ }}
or {% %}
- bits that manipulate some of the context
variables we pass into the render
call.
So, in our unit tests, we need to check that the variables we pass in end up being used - that's why we have the assertIn
checks on the response.content
as well as the assertEqual
test on the response.context
.
So, what about checking that our template contains the correct hyperlinks, <a href="/poll/01/
, or whatever they may be? Well, if we were to hard-code them into the template, then that would be a bit like testing a constant. But we're not going to hard-code them, because that would violate the programming DRY principle - "Don't Repeat Yourself".
If we were to hard-code the URLs for links to individual polls, it would be really tedious if we wanted to come back and change them later - say from /poll/1/
to /poll_detail/01/
or whatever it may be. Django provides a single place to define urls, in urls.py
, and it then provides helper tools for retrieving them in other places - a function called reverse
, and a template tag called {% url %}
. So we'll use the template tag, which avoids hard-coding the URL in the template, but it also means that the hyperlink is no longer a constant, so we need to test it.
Phew, that was long winded! Anyway, the upshot is, more tests - but also, we get to learn about Django url helper functions, so it's win-win-win :-)
Let's use the reverse
function in our tests. Its first argument is the name of the view that handles the url, and we can also specify some arguments. We'll be making a view for seeing an individual Poll object, so we'll probably find the poll using its id
. Here's what that translates to in tests.py
:
python
from django.core.urlresolvers import reverse
class TestAllPollsView(TestCase):
- def test_root_url_shows_links_to_all_polls(self):
# set up some polls poll1 = Poll(question='6 times 7', pub_date='2001-01-01') poll1.save() poll2 = Poll(question='life, the universe and everything', pub_date='2001-01-01') poll2.save()
client = Client() response = client.get('/')
template_names_used = [t.name for t in response.templates] self.assertIn('polls.html', template_names_used)
# check we've passed the polls to the template polls_in_context = response.context['polls'] self.assertEquals(list(polls_in_context), [poll1, poll2])
# check the poll names appear on the page self.assertIn(poll1.question, response.content) self.assertIn(poll2.question, response.content)
# check the page also contains the urls to individual polls pages poll1_url = reverse('mysite.polls.views.poll', args=[poll1.id,]) self.assertIn(poll1_url, response.content) poll2_url = reverse('mysite.polls.views.poll', args=[poll2.id,]) self.assertIn(poll2_url, response.content)
Running this (python manage.py test polls
) gives:
======================================================================
ERROR: test_root_url_shows_links_to_all_polls (polls.tests.TestAllPollsView)
----------------------------------------------------------------------
Traceback (most recent call last):
File "/home/harry/workspace/tddjango_site/source/mysite/polls/tests.py", line 107, in test_root_url_shows_links_to_all_polls
poll1_url = reverse('mysite.polls.views.poll', kwargs=dict(poll_id=poll1.id))
File "/usr/lib/pymodules/python2.7/django/core/urlresolvers.py", line 391, in reverse
*args, **kwargs)))
File "/usr/lib/pymodules/python2.7/django/core/urlresolvers.py", line 337, in reverse
"arguments '%s' not found." % (lookup_view_s, args, kwargs))
NoReverseMatch: Reverse for 'mysite.polls.views.poll' with arguments '()' and keyword arguments '{'poll_id': 1}' not found.
----------------------------------------------------------------------
So, the reverse
function can't find a url or a view to match our request - let's add one!
In urls.py
:
python
- urlpatterns = patterns('',
(r'^$', polls_views.polls), (r'^poll/(d+)/$', polls_views.poll), (r'^admin/', include(admin.site.urls)),
)
The new line will match any url which starts with poll/, then a number made up of one or more digits - the matching group (\d+)
, which will be captured and passed as the first argument to our view - which is reflected in the reverse function's args
parameter.
Now our unit tests give a different error:
======================================================================
FAIL: test_root_url_shows_links_to_all_polls (polls.tests.TestAllPollsView)
----------------------------------------------------------------------
Traceback (most recent call last):
File "/home/harry/workspace/tddjango_site/source/mysite/polls/tests.py", line 108, in test_root_url_shows_links_to_all_polls
self.assertIn(poll1_url, response.content)
AssertionError: '/poll/1/' not found in '<html>\n <body>\n <h1>Polls</h1>\n \n 6 times 7\n \n life, the universe and everything\n \n </body>\n</html>\n'
----------------------------------------------------------------------
We'll also need to add at least a dummy view in views.py
python
- def polls(request):
context = {'polls': Poll.objects.all()} return render(request, 'polls.html', context)
- def poll():
pass
The templates don't include the urls yet. Let's add them:
html+django
- <html>
- <body>
<h1>Polls</h1> {% for poll in polls %} <a href="{% url mysite.polls.views.poll poll.id %}">{{ poll.question }}</a> {% endfor %}
</body>
</html>
Notice the call to {% url %}
, whose signature is very similar to the call to reverse
. Now our unit tests are a lot happier!:
21:08 ~/workspace/tddjango_site/source/mysite (master)$ python manage.py test polls
Creating test database for alias 'default'...
......
----------------------------------------------------------------------
Ran 6 tests in 0.012s
OK
What about the functional tests?:
======================================================================
FAIL: test_voting_on_a_new_poll (test_polls.TestPolls)
----------------------------------------------------------------------
Traceback (most recent call last):
File "/home/harry/workspace/tddjango_site/source/mysite/fts/test_polls.py", line 67, in test_voting_on_a_new_poll
self.assertEquals(heading.text, 'Poll Results')
AssertionError: u'TypeError at /poll/1/' != 'Poll Results'
----------------------------------------------------------------------
Ran 2 tests in 25.927s
Looks like it's time to start implementing our poll view, which aims to show information about an individual poll... But for this, you'll have to tune in next week!