test_models.py

"""
Tests for the rss_proxy models
"""
from django.test import TestCase

from rss_proxy.models import WhitelistedRssUrl


class WhitelistedRssUrlTests(TestCase):
    """ Tests for the rss_proxy.WhitelistedRssUrl model """

    def setUp(self):
        super(WhitelistedRssUrlTests, self).setUp()
        self.whitelisted_rss_url = WhitelistedRssUrl.objects.create(url='http://www.example.com')

    def test_unicode(self):
        """
        Test the unicode function returns the url
        """
        self.assertEqual(unicode(self.whitelisted_rss_url), self.whitelisted_rss_url.url)