Welcome to the Treehouse Community

Want to collaborate on code errors? Have bugs you need feedback on? Looking for an extra set of eyes on your latest project? Get support with fellow developers, designers, and programmers of all backgrounds and skill levels here with the Treehouse Community! While you're at it, check out some resources Treehouse students have shared here.

Looking to learn something new?

Treehouse offers a seven day free trial for new students. Get access to thousands of hours of content and join thousands of Treehouse students and alumni in the community today.

Start your free trial

Python Django Basics Test Time Test our Writer model

peterbristow
peterbristow
10,403 Points
Posted by peterbristow
peterbristow
10,403 Points

Why is this unit test not working?

Please see the code...

articles/tests.py 
from django.test import TestCase

from .models import Writer


class WriterModelTestCase(TestCase):
    '''Tests for the Writer model'''
    writer = Writer.objects.create(
        name="Peter",
        email="email@test.com",
        bio="This is the bio"
    )
    self.assertIn(writer.email, writer.mailto())
articles/models.py 
from django.db import models


class Article(models.Model):
    headline = models.CharField(max_length=255)
    publish_date = models.DateTimeField()
    content = models.TextField()
    writer = models.ForeignKey('Writer')

    def __str__(self):
        return self.headline


class Writer(models.Model):
    name = models.CharField(max_length=255)
    email = models.EmailField()
    bio = models.TextField()

    def __str__(self):
        return self.name

    def mailto(self):
        return '{} <{}>'.format(self.name, self.email)

2 Answers

Julian Garcia
Julian Garcia
18,380 Points
Julian Garcia
Julian Garcia
18,380 Points

It don`t pass because it is necessary to create methods inside de class. And more specifically it needs that the names of that methods start with test_, except for setUp and tearDown.

I rewrite your code a little.

class WriterModelTestCase(TestCase): '''Tests for the Writer model''' def setUp(self): self.writer = Writer.objects.create( name="Peter", email="email@test.com", bio="This is the bio" ) def test_email(self): self.assertIn(self.writer.email, self.writer.mailto())

Besides that in setUp method you define beforehand all the things you are going to need in your test methods.

Hope this help.

Steven Turturo
Steven Turturo
6,890 Points
Steven Turturo
Steven Turturo
6,890 Points

You just need to put your test case into a function/method. 

from django.test import TestCase

from .models import Writer


class WriterModelTestCase(TestCase):
    '''Tests for the Writer model'''
    def test_email(self):
        writer = Writer.objects.create(
            name="Peter",
            email="email@test.com",
            bio="This is the bio"
        )
        self.assertIn(writer.email, writer.mailto())