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

from lms.djangoapps.rss_proxy.models import WhitelistedRssUrl


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

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

    def test_unicode(self):
        """
        Test the unicode function returns the url
        """
        assert str(self.whitelisted_rss_url) == self.whitelisted_rss_url.url