Used get_current_site in comments feed class

2 files changed, 22 insertions, 0 deletions

diff --git a/tests/regressiontests/comment_tests/tests/feed_tests.py b/tests/regressiontests/comment_tests/tests/feed_tests.py
index b15ec0c7b8..1ec316eab8 100644
--- a/tests/regressiontests/comment_tests/tests/feed_tests.py
+++ b/tests/regressiontests/comment_tests/tests/feed_tests.py
@@ -1,12 +1,32 @@
 from __future__ import absolute_import
 
+from django.conf import settings
+from django.contrib.comments.models import Comment
+from django.contrib.contenttypes.models import ContentType
+from django.contrib.sites.models import Site
+
 from . import CommentTestCase
+from ..models import Article
 
 
 class CommentFeedTests(CommentTestCase):
     urls = 'regressiontests.comment_tests.urls'
     feed_url = '/rss/comments/'
 
+    def setUp(self):
+        site_2 = Site.objects.create(id=settings.SITE_ID+1,
+            domain="example2.com", name="example2.com")
+        # A comment for another site
+        c5 = Comment.objects.create(
+            content_type = ContentType.objects.get_for_model(Article),
+            object_pk = "1",
+            user_name = "Joe Somebody",
+            user_email = "jsomebody@example.com",
+            user_url = "http://example.com/~joe/",
+            comment = "A comment for the second site.",
+            site = site_2,
+        )
+
     def test_feed(self):
         response = self.client.get(self.feed_url)
         self.assertEqual(response.status_code, 200)
@@ -15,3 +35,4 @@ class CommentFeedTests(CommentTestCase):
         self.assertContains(response, '<title>example.com comments</title>')
         self.assertContains(response, '<link>http://example.com/</link>')
         self.assertContains(response, '</rss>')
+        self.assertNotContains(response, "A comment for the second site.")
diff --git a/tests/regressiontests/comment_tests/urls.py b/tests/regressiontests/comment_tests/urls.py
index b2f676786f..0a7e8b5fdf 100644
--- a/tests/regressiontests/comment_tests/urls.py
+++ b/tests/regressiontests/comment_tests/urls.py
@@ -15,6 +15,7 @@ urlpatterns = patterns('',
     url(r'^flag/(\d+)/$', views.custom_flag_comment),
     url(r'^delete/(\d+)/$', views.custom_delete_comment),
     url(r'^approve/(\d+)/$', views.custom_approve_comment),
+    url(r'^cr/(\d+)/(.+)/$', 'django.contrib.contenttypes.views.shortcut', name='comments-url-redirect'),
 )
 
 urlpatterns += patterns('',