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

Angelo Giron
Angelo Giron
1,391 Points
Posted by Angelo Giron
Angelo Giron
1,391 Points

How i should use assertIn in tests of Django?

I have the following code, but in the code challenge it says "Didn't find 'self.assertIn' "

articles/tests.py 
from django.test import TestCase
from .models import Writer


class WriterModelTestCase(TestCase):
    '''Tests for the Writer model'''
    def test_writer_creation(self):
        writer = Writer.objects.create(name='angelo', bio='no hay bio', email='angelo@noemail.com')
        self.assertIn('to angelo@noemail.com', 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 'to {}'.format(self.email)

3 Answers

Christoffer Sandberg
Christoffer Sandberg
14,334 Points
Christoffer Sandberg
Christoffer Sandberg
14,334 Points

Try changing: 

self.assertIn('to angelo@noemail.com', writer.mailto())

to 

self.assertIn(writer.email, writer.mailto())

[edit formatting -cf]

Brady Rothrock
Brady Rothrock
2,709 Points

I also found that I needed to remove the name and bio fields. So mine looked like:

class WriterModelTestCase(TestCase):
    '''Tests for the Writer model'''
    def test_writer_creation(self):
        writer = Writer.objects.create(email='brady@noemail.com')
        self.assertIn(writer.email, writer.mailto())
Martin Cornejo Saavedra
Martin Cornejo Saavedra
18,132 Points
Martin Cornejo Saavedra
Martin Cornejo Saavedra
18,132 Points

I've not been yet in django courses, but I did python testing, and I remember you had to import unittest module, so I'd change this line:

#from django.test import TestCase   #I'd comment this line

#and add this one
from unittest import TestCase

Let me know if this was helpful.

Angelo Giron
Angelo Giron
1,391 Points
Angelo Giron
Angelo Giron
1,391 Points

Thanks for your reply but i got same results. When my test is correct appears this message: Bummer! Didn't find self.assertIn used.

crispinalibasah
crispinalibasah
1,151 Points
crispinalibasah
crispinalibasah
1,151 Points

I still have the same problem as Angelo's.