def save(self, commit=True):
if self.instance.pk:
post = super(PostForm, self).save(commit=False)
if self.user:
post.user = self.user
if post.topic.head == post:
post.topic.name = self.cleaned_data['name']
post.topic.updated = datetime.now()
post.topic.save()
post.save()
return post
allow_post = True
if defaults.PYBB_PREMODERATION:
allow_post = defaults.PYBB_PREMODERATION(self.user, self.cleaned_data['body'])
if self.forum:
topic = Topic(forum=self.forum,
user=self.user,
name=self.cleaned_data['name'])
if not allow_post:
topic.on_moderation = True
topic.save()
else:
topic = self.topic
post = Post(topic=topic, user=self.user, user_ip=self.ip,
body=self.cleaned_data['body'])
if not allow_post:
post.on_moderation = True
post.save()
return post
def get_or_create_topic(self, allow_post):
if self._forum:
topic = Topic(
forum=self._forum,
user=self.user,
name=self.cleaned_data['name'],
)
if not allow_post:
topic.on_moderation = topic.MODERATION_IS_IN_MODERATION
topic.save()
if not defaults.PYBB_DISABLE_POLLS:
if 'poll_type' in self.cleaned_data and self.cleaned_data['poll_type'] != Poll.TYPE_NONE:
poll = Poll(
type=self.cleaned_data['poll_type'],
question=self.cleaned_data['poll_question']
)
poll.save()
topic.poll = poll
else:
topic = self._topic
return topic
def save(self, commit=True):
if self.instance.pk:
post = super(PostForm, self).save(commit=False)
if self.user:
post.user = self.user
if post.topic.head == post:
post.topic.name = self.cleaned_data['name']
post.topic.updated = datetime.now()
post.topic.save()
post.save()
return post
allow_post = True
if defaults.PYBB_PREMODERATION:
allow_post = defaults.PYBB_PREMODERATION(self.user,
self.cleaned_data['body'])
if self.forum:
topic = Topic(forum=self.forum,
user=self.user,
name=self.cleaned_data['name'])
if not allow_post:
topic.on_moderation = True
topic.save()
else:
topic = self.topic
post = Post(topic=topic,
user=self.user,
user_ip=self.ip,
body=self.cleaned_data['body'])
if not allow_post:
post.on_moderation = True
post.save()
return post
def test_topic_move_complete(self):
topic = Topic(name='xtopic', forum=self.forum, user=self.user)
topic.save()
post = Post(topic=topic, user=self.user, body='one')
post.save()
topic_merge_url = reverse('pybb:topic_move')
self.login_as(self.staff)
name = 'new name for a topic'
response = self.client.post(topic_merge_url, data={
'topic_ids': [topic.pk],
'form-TOTAL_FORMS': 1,
'form-INITIAL_FORMS': 0,
'form-0-name': name,
'form-0-forum': self.forum.pk,
'form-0-redirection_type': TopicRedirection.TYPE_PERMANENT_REDIRECT,
'submit': 1
})
self.assertRedirects(response, reverse('pybb:index'))
topic = Topic.objects.get(pk=topic.pk)
self.assertTrue(topic.redirect)
self.failUnless(topic.redirection is not None)
redirection = topic.redirection
new_topic = redirection.to_topic
self.assertEqual(new_topic.name, name)
self.assertEqual(new_topic.post_count, 1)
def test_read_tracking_multi_forum(self):
self.post
topic_1 = self.topic
topic_2 = Topic(name='topic_2', forum=self.forum, user=self.user)
topic_2.save()
Post(topic=topic_2, user=self.user, body='one').save()
forum_2 = Forum(name='forum_2', description='bar', forum=self.parent_forum)
forum_2.save()
Topic(name='garbage', forum=forum_2, user=self.user).save()
self.login_as(self.user)
# everything starts unread
self.assertEqual(ForumReadTracker.objects.count(), 0)
self.assertEqual(TopicReadTracker.objects.count(), 0)
# user reads topic_1, they should get one topic read tracker, there should be no forum read trackers
self.client.get(topic_1.get_absolute_url())
self.assertEqual(TopicReadTracker.objects.count(), 1)
self.assertEqual(TopicReadTracker.objects.filter(user=self.user).count(), 1)
self.assertEqual(TopicReadTracker.objects.filter(user=self.user, topic=topic_1).count(), 1)
# user reads topic_2, they should get a forum read tracker,
# there should be no topic read trackers for the user
self.client.get(topic_2.get_absolute_url())
self.assertEqual(TopicReadTracker.objects.count(), 0)
self.assertEqual(ForumReadTracker.objects.count(), 1)
self.assertEqual(ForumReadTracker.objects.filter(user=self.user).count(), 1)
self.assertEqual(ForumReadTracker.objects.filter(user=self.user, forum=self.forum).count(), 1)
def test_forum_updated(self):
time.sleep(1)
topic = Topic(name='xtopic', forum=self.forum, user=self.user)
topic.save()
post = Post(topic=topic, user=self.user, body='one')
post.save()
post = Post.objects.get(id=post.id)
self.assertTrue(self.forum.updated == post.created)
def test_forum_updated(self):
time.sleep(1)
topic = Topic(name='xtopic', forum=self.forum, user=self.user)
topic.save()
post = Post(topic=topic, user=self.user, body='one')
post.save()
post = Post.objects.get(id=post.id)
self.assertTrue(self.forum.updated == post.created)
def test_hidden(self):
client = Client()
parent = Forum(name='hcat', hidden=True)
parent.save()
forum_in_hidden = Forum(name='in_hidden', forum=parent)
forum_in_hidden.save()
topic_in_hidden = Topic(forum=forum_in_hidden, name='in_hidden', user=self.user)
topic_in_hidden.save()
forum_hidden = Forum(name='hidden', forum=self.parent_forum, hidden=True)
forum_hidden.save()
topic_hidden = Topic(forum=forum_hidden, name='hidden', user=self.user)
topic_hidden.save()
post_hidden = Post(topic=topic_hidden, user=self.user, body='hidden')
post_hidden.save()
post_in_hidden = Post(topic=topic_in_hidden, user=self.user, body='hidden')
post_in_hidden.save()
self.assertFalse(parent.id in [c.id for c in client.get(reverse('pybb:index')).context['forums']])
self.assertEqual(client.get(parent.get_absolute_url()).status_code, 302)
self.assertEqual(client.get(forum_in_hidden.get_absolute_url()).status_code, 302)
self.assertEqual(client.get(topic_in_hidden.get_absolute_url()).status_code, 302)
self.assertNotContains(client.get(reverse('pybb:index')), forum_hidden.get_absolute_url())
self.assertNotContains(client.get(reverse('pybb:feed_topics')), topic_hidden.get_absolute_url())
self.assertNotContains(client.get(reverse('pybb:feed_topics')), topic_in_hidden.get_absolute_url())
self.assertNotContains(client.get(reverse('pybb:feed_posts')), post_hidden.get_absolute_url())
self.assertNotContains(client.get(reverse('pybb:feed_posts')), post_in_hidden.get_absolute_url())
self.assertEqual(client.get(forum_hidden.get_absolute_url()).status_code, 302)
self.assertEqual(client.get(topic_hidden.get_absolute_url()).status_code, 302)
client.login(username='******', password='******')
self.assertTrue(parent.id in [c.id for c in client.get(reverse('pybb:index')).context['forums']])
self.assertContains(client.get(reverse('pybb:index')), forum_hidden.get_absolute_url())
self.assertEqual(client.get(parent.get_absolute_url()).status_code, 200)
self.assertEqual(client.get(forum_in_hidden.get_absolute_url()).status_code, 200)
self.assertEqual(client.get(topic_in_hidden.get_absolute_url()).status_code, 200)
self.assertEqual(client.get(forum_hidden.get_absolute_url()).status_code, 200)
self.assertEqual(client.get(topic_hidden.get_absolute_url()).status_code, 200)
self.user.is_staff = True
self.user.save()
self.assertTrue(parent.id in [c.id for c in client.get(reverse('pybb:index')).context['forums']])
self.assertContains(client.get(reverse('pybb:index')), forum_hidden.get_absolute_url())
self.assertEqual(client.get(parent.get_absolute_url()).status_code, 200)
self.assertEqual(client.get(forum_in_hidden.get_absolute_url()).status_code, 200)
self.assertEqual(client.get(topic_in_hidden.get_absolute_url()).status_code, 200)
self.assertEqual(client.get(forum_hidden.get_absolute_url()).status_code, 200)
self.assertEqual(client.get(topic_hidden.get_absolute_url()).status_code, 200)
def test_compute_active_members(self):
# Initials
self.assertEquals(self.parent_forum.member_count, 0)
self.assertEquals(self.forum.member_count, 0)
self.assertEquals(self.topic.member_count, 0)
# Add first post by self.user (count += 1)
user_post_1 = self.post
self.assertEquals(self.parent_forum.member_count, 1)
self.assertEquals(self.forum.member_count, 1)
self.assertEquals(self.topic.member_count, 1)
# Add first post by self.staff (count += 1)
staff_post_1 = Post(topic=self.topic,
user=self.staff,
body='my new post')
staff_post_1.save()
self.assertEquals(self.parent_forum.member_count, 2)
self.assertEquals(self.forum.member_count, 2)
self.assertEquals(self.topic.member_count, 2)
# Add second post by self.user (count += 0)
user_post_2 = Post(topic=self.topic,
user=self.user,
body='my new post')
user_post_2.save()
self.assertEquals(self.parent_forum.member_count, 2)
self.assertEquals(self.forum.member_count, 2)
self.assertEquals(self.topic.member_count, 2)
# Delete staff_post_1 (count -= 1)
staff_post_1.mark_as_deleted(commit=True)
self.assertEquals(self.parent_forum.member_count, 1)
self.assertEquals(self.forum.member_count, 1)
self.assertEquals(self.topic.member_count, 1)
# Restore staff_post_1 (count += 1)
staff_post_1.mark_as_undeleted(commit=True)
self.assertEquals(self.parent_forum.member_count, 2)
self.assertEquals(self.forum.member_count, 2)
self.assertEquals(self.topic.member_count, 2)
# Delete user_post_2 (count -= 0)
user_post_2.mark_as_deleted(commit=True)
self.assertEquals(self.parent_forum.member_count, 2)
self.assertEquals(self.forum.member_count, 2)
self.assertEquals(self.topic.member_count, 2)
# Add third post by self.user in parent_forum (count += 0)
new_topic = Topic(name='foo2', forum=self.parent_forum, user=self.user)
new_topic.save()
user_post_3 = Post(topic=new_topic, user=self.user, body='my new post')
user_post_3.save()
self.assertEquals(self.parent_forum.member_count, 2)
self.assertEquals(new_topic.member_count, 1)
self.assertEquals(self.forum.member_count, 2)
self.assertEquals(self.topic.member_count, 2)
def test_pagination_and_topic_addition(self):
for i in range(0, defaults.PYBB_FORUM_PAGE_SIZE + 3):
topic = Topic(name='topic_%s_' % i, forum=self.forum, user=self.user)
topic.save()
response = self.client.get(self.forum.get_absolute_url())
self.assertEqual(len(response.context['topic_list']), defaults.PYBB_FORUM_PAGE_SIZE)
self.assertTrue(response.context['is_paginated'])
self.assertEqual(response.context['paginator'].num_pages,
(int((defaults.PYBB_FORUM_PAGE_SIZE + 3) / defaults.PYBB_FORUM_PAGE_SIZE) + 1))
def test_compute_active_members(self):
# Initials
self.assertEquals(self.parent_forum.member_count, 0)
self.assertEquals(self.forum.member_count, 0)
self.assertEquals(self.topic.member_count, 0)
# Add first post by self.user (count += 1)
user_post_1 = self.post
self.assertEquals(self.parent_forum.member_count, 1)
self.assertEquals(self.forum.member_count, 1)
self.assertEquals(self.topic.member_count, 1)
# Add first post by self.staff (count += 1)
staff_post_1 = Post(topic=self.topic, user=self.staff, body='my new post')
staff_post_1.save()
self.assertEquals(self.parent_forum.member_count, 2)
self.assertEquals(self.forum.member_count, 2)
self.assertEquals(self.topic.member_count, 2)
# Add second post by self.user (count += 0)
user_post_2 = Post(topic=self.topic, user=self.user, body='my new post')
user_post_2.save()
self.assertEquals(self.parent_forum.member_count, 2)
self.assertEquals(self.forum.member_count, 2)
self.assertEquals(self.topic.member_count, 2)
# Delete staff_post_1 (count -= 1)
staff_post_1.mark_as_deleted(commit=True)
self.assertEquals(self.parent_forum.member_count, 1)
self.assertEquals(self.forum.member_count, 1)
self.assertEquals(self.topic.member_count, 1)
# Restore staff_post_1 (count += 1)
staff_post_1.mark_as_undeleted(commit=True)
self.assertEquals(self.parent_forum.member_count, 2)
self.assertEquals(self.forum.member_count, 2)
self.assertEquals(self.topic.member_count, 2)
# Delete user_post_2 (count -= 0)
user_post_2.mark_as_deleted(commit=True)
self.assertEquals(self.parent_forum.member_count, 2)
self.assertEquals(self.forum.member_count, 2)
self.assertEquals(self.topic.member_count, 2)
# Add third post by self.user in parent_forum (count += 0)
new_topic = Topic(name='foo2', forum=self.parent_forum, user=self.user)
new_topic.save()
user_post_3 = Post(topic=new_topic, user=self.user, body='my new post')
user_post_3.save()
self.assertEquals(self.parent_forum.member_count, 2)
self.assertEquals(new_topic.member_count, 1)
self.assertEquals(self.forum.member_count, 2)
self.assertEquals(self.topic.member_count, 2)
def test_topic_deletion(self):
topic = Topic(name='xtopic', forum=self.forum, user=self.user)
topic.save()
post = Post(topic=topic, user=self.user, body='one')
post.save()
post = Post(topic=topic, user=self.user, body='two')
post.save()
post.delete()
Topic.objects.get(id=topic.id)
Forum.objects.get(id=self.forum.id)
topic.delete()
Forum.objects.get(id=self.forum.id)
def test_topic_deletion(self):
topic = Topic(name='xtopic', forum=self.forum, user=self.user)
topic.save()
post = Post(topic=topic, user=self.user, body='one')
post.save()
post = Post(topic=topic, user=self.user, body='two')
post.save()
post.delete()
Topic.objects.get(id=topic.id)
Forum.objects.get(id=self.forum.id)
topic.delete()
Forum.objects.get(id=self.forum.id)
def test_staff(self):
client = Client()
parent = Forum(name='hcat', staff=True)
parent.save()
forum_in_staff = Forum(name='in_staff', forum=parent)
forum_in_staff.save()
topic_in_staff = Topic(forum=forum_in_staff, name='in_staff', user=self.user)
topic_in_staff.save()
forum_staff = Forum(name='staff', forum=self.parent_forum, staff=True)
forum_staff.save()
topic_staff = Topic(forum=forum_staff, name='staff', user=self.user)
topic_staff.save()
post_staff = Post(topic=topic_staff, user=self.user, body='staff')
post_staff.save()
post_in_staff = Post(topic=topic_in_staff, user=self.user, body='staff')
post_in_staff.save()
self.assertFalse(parent.id in [c.id for c in client.get(reverse('pybb:index')).context['forums']])
self.assertEqual(client.get(parent.get_absolute_url()).status_code, 404)
self.assertEqual(client.get(forum_in_staff.get_absolute_url()).status_code, 404)
self.assertEqual(client.get(topic_in_staff.get_absolute_url()).status_code, 404)
self.assertEqual(client.get(forum_staff.get_absolute_url()).status_code, 404)
self.assertEqual(client.get(topic_staff.get_absolute_url()).status_code, 404)
client.login(username='******', password='******')
self.assertFalse(parent.id in [c.id for c in client.get(reverse('pybb:index')).context['forums']])
self.assertNotContains(client.get(reverse('pybb:index')), forum_staff.get_absolute_url())
self.assertEqual(client.get(parent.get_absolute_url()).status_code, 404)
self.assertEqual(client.get(forum_in_staff.get_absolute_url()).status_code, 404)
self.assertEqual(client.get(topic_in_staff.get_absolute_url()).status_code, 404)
self.assertEqual(client.get(forum_staff.get_absolute_url()).status_code, 404)
self.assertEqual(client.get(topic_staff.get_absolute_url()).status_code, 404)
self.user.is_staff = True
self.user.save()
self.assertTrue(parent.id in [c.id for c in client.get(reverse('pybb:index')).context['forums']])
self.assertEqual(client.get(parent.get_absolute_url()).status_code, 200)
self.assertEqual(client.get(forum_in_staff.get_absolute_url()).status_code, 200)
self.assertEqual(client.get(topic_in_staff.get_absolute_url()).status_code, 200)
self.assertEqual(client.get(forum_staff.get_absolute_url()).status_code, 200)
self.assertEqual(client.get(topic_staff.get_absolute_url()).status_code, 200)
def test_compute(self):
# initials
self.assertEqual(self.forum.forum_count, 0)
self.assertEqual(self.parent_forum.forum_count, 1)
self.topic
self.post
parent_forum = Forum.objects.get(pk=self.parent_forum.pk)
self.assertEqual(parent_forum.topic_count, 1)
self.assertEqual(parent_forum.post_count, 1)
self.assertEqual(self.forum.topic_count, 1)
self.assertEqual(self.forum.post_count, 1)
forum = Forum.objects.create(name='bar', description='bar', forum=self.forum)
topic = Topic(name='bar', forum=forum, user=self.user)
topic.save()
post = Post(topic=topic, user=self.user, body='my new post')
post.save()
self.assertEqual(topic.post_count, 1)
self.assertEqual(forum.topic_count, 1)
self.assertEqual(Forum.objects.get(pk=self.forum.pk).topic_count, 2)
self.assertEqual(Forum.objects.get(pk=self.parent_forum.pk).topic_count, 2)
new_topic = Topic(name='foo', forum=self.forum, user=self.user)
new_topic.save()
new_post = Post(topic=topic, user=self.user, body='my new post')
new_post.save()
self.assertEqual(Forum.objects.get(pk=self.forum.pk).topic_count, 3)
self.assertEqual(Forum.objects.get(pk=self.parent_forum.pk).topic_count, 3)
post.mark_as_deleted(commit=True)
self.assertEqual(Forum.objects.get(pk=self.forum.pk).topic_count, 2)
self.assertEqual(Forum.objects.get(pk=self.parent_forum.pk).topic_count, 2)
self.assertTrue(Topic.objects.get(pk=topic.pk).deleted)
self.assertEqual(Forum.objects.get(pk=forum.pk).topic_count, 0)
self.assertEqual(Forum.objects.get(pk=forum.pk).post_count, 0)
new_topic.mark_as_deleted()
self.assertEqual(Forum.objects.get(pk=self.forum.pk).topic_count, 1)
self.assertEqual(Forum.objects.get(pk=self.parent_forum.pk).topic_count, 1)
def save(self, commit=True):
if self.instance.pk:
post = super(PostForm, self).save(commit=False)
if self.user:
post.user = self.user
if post.topic.head == post:
post.topic.name = self.cleaned_data['name']
if self.may_create_poll:
post.topic.poll_type = self.cleaned_data['poll_type']
post.topic.poll_question = self.cleaned_data[
'poll_question']
post.topic.updated = tznow()
if commit:
post.topic.save()
post.updated = tznow()
if commit:
post.save()
# Subscribe to topic
if self.cleaned_data['subscribe']:
post.topic.subscribers.add(self.user)
else:
post.topic.subscribers.remove(self.user)
return post, post.topic
allow_post = True
if defaults.PYBB_PREMODERATION:
allow_post = defaults.PYBB_PREMODERATION(self.user,
self.cleaned_data['body'])
if self.forum:
topic = Topic(
forum=self.forum,
user=self.user,
name=self.cleaned_data['name'],
poll_type=self.cleaned_data.get('poll_type',
Topic.POLL_TYPE_NONE),
poll_question=self.cleaned_data.get('poll_question', None),
slug=self.cleaned_data.get('slug', None),
)
if not allow_post:
topic.on_moderation = True
else:
topic = self.topic
post = Post(user=self.user,
user_ip=self.ip,
body=self.cleaned_data['body'])
if not allow_post:
post.on_moderation = True
if commit:
topic.save()
post.topic = topic
post.save()
return post, topic
def test_pagination_and_topic_addition(self):
for i in range(0, defaults.PYBB_FORUM_PAGE_SIZE + 3):
topic = Topic(name='topic_%s_' % i,
forum=self.forum,
user=self.user)
topic.save()
response = self.client.get(self.forum.get_absolute_url())
self.assertEqual(len(response.context['topic_list']),
defaults.PYBB_FORUM_PAGE_SIZE)
self.assertTrue(response.context['is_paginated'])
self.assertEqual(response.context['paginator'].num_pages,
(int((defaults.PYBB_FORUM_PAGE_SIZE + 3) /
defaults.PYBB_FORUM_PAGE_SIZE) + 1))
def test_topic_redirect_one_page(self):
topic = Topic(name='to-etopic', forum=self.forum, user=self.user)
topic.save()
response = self.client.get(
reverse('pybb:topic_detail',
kwargs={
'forum_slug': self.forum.slug,
'pk': topic.pk,
'slug': topic.slug,
'page': 1
}))
self.assertRedirects(response, topic.get_absolute_url(), 301)
def test_topic_redirect_one_page(self):
topic = Topic(name='to-etopic', forum=self.forum, user=self.user)
topic.save()
response = self.client.get(reverse('pybb:topic_detail', kwargs={
'forum_slug': self.forum.slug,
'pk': topic.pk,
'slug': topic.slug,
'page': 1
}))
self.assertRedirects(response,
topic.get_absolute_url(),
301)
def test_topic_updated(self):
topic = Topic(name='etopic', forum=self.forum, user=self.user)
topic.save()
time.sleep(1)
post = Post(topic=topic, user=self.user, body='bbcode [b]test[b]')
post.save()
client = Client()
response = client.get(self.forum.get_absolute_url())
self.assertEqual(response.context['topic_list'][0], topic)
time.sleep(1)
post = Post(topic=self.topic, user=self.user, body='bbcode [b]test[b]')
post.save()
client = Client()
response = client.get(self.forum.get_absolute_url())
self.assertEqual(response.context['topic_list'][0], self.topic)
def create(cls, primary_claim, linked_claim, link_type, user):
print("ClaimLink:create")
my_topic = Topic(name = linked_claim.title + " " +
link_type.reverse_name.upper() + " " +
primary_claim.title,
forum = primary_claim.forum,
user = user)
my_topic.save()
link = cls(primary_claim = primary_claim,
linked_claim = linked_claim,
link_type = link_type,
topic = my_topic,
user = user
)
return link
def test_topic_updated(self):
topic = Topic(name='etopic', forum=self.forum, user=self.user)
topic.save()
time.sleep(1)
post = Post(topic=topic, user=self.user, body='bbcode [b]test[b]')
post.save()
client = Client()
response = client.get(self.forum.get_absolute_url())
self.assertEqual(response.context['topic_list'][0], topic)
time.sleep(1)
post = Post(topic=self.topic, user=self.user, body='bbcode [b]test[b]')
post.save()
client = Client()
response = client.get(self.forum.get_absolute_url())
self.assertEqual(response.context['topic_list'][0], self.topic)
def test_topic_merge_view(self):
topic = Topic(name='xtopic', forum=self.forum, user=self.user)
topic.save()
post = Post(topic=topic, user=self.user, body='one')
post.save()
topic_merge_url = reverse('pybb:topic_merge')
self.login_as(self.staff)
response = self.client.post(topic_merge_url,
data={'topic_ids': [topic.pk]})
self.assertEqual(response.status_code, 200)
self.assertEqual(len(response.context['form'].forms), 1)
self.assertTemplateUsed(response, 'pybb/topic/merge.html')
def test_topic_merge_view(self):
topic = Topic(name='xtopic', forum=self.forum, user=self.user)
topic.save()
post = Post(topic=topic, user=self.user, body='one')
post.save()
topic_merge_url = reverse('pybb:topic_merge')
self.login_as(self.staff)
response = self.client.post(topic_merge_url, data={
'topic_ids': [topic.pk]
})
self.assertEqual(response.status_code, 200)
self.assertEqual(len(response.context['form'].forms), 1)
self.assertTemplateUsed(response, 'pybb/topic/merge.html')
def get_new_topic(self):
if hasattr(self, '_new_topic'):
return self._new_topic
if self.post.is_topic_head:
topic = self.topic
else:
topic = Topic(user=self.post.user)
if topic.name != self.cleaned_data['name']:
topic.name = self.cleaned_data['name']
# force slug auto-rebuild if slug is not speficied and topic is renamed
topic.slug = self.cleaned_data.get('slug', None)
elif self.cleaned_data.get('slug', None):
topic.slug = self.cleaned_data['slug']
topic.forum = Forum.objects.get(pk=self.cleaned_data['move_to'])
topic.slug = create_or_check_slug(topic, Topic, forum=topic.forum)
topic.save()
return topic
def get_new_topic(self):
if hasattr(self, '_new_topic'):
return self._new_topic
if self.post.is_topic_head:
topic = self.topic
else:
topic = Topic(user=self.post.user)
if topic.name != self.cleaned_data['name']:
topic.name = self.cleaned_data['name']
# force slug auto-rebuild if slug is not speficied and topic is renamed
topic.slug = self.cleaned_data.get('slug', None)
elif self.cleaned_data.get('slug', None):
topic.slug = self.cleaned_data['slug']
topic.forum = Forum.objects.get(pk=self.cleaned_data['move_to'])
topic.slug = create_or_check_slug(topic, Topic, forum=topic.forum)
topic.save()
return topic
def test_topic_merge_complete(self):
topic = Topic(name='xtopic', forum=self.forum, user=self.user)
topic.save()
post = Post(topic=topic, user=self.user, body='one')
post.save()
topic_merge_url = reverse('pybb:topic_merge')
self.login_as(self.staff)
response = self.client.post(topic_merge_url, data={
'topic_ids': [topic.pk],
'form-TOTAL_FORMS': 1,
'form-INITIAL_FORMS': 0,
'form-0-topic': self.topic.pk,
'submit': 1
})
self.assertRedirects(response, reverse('pybb:index'))
def save(self, commit=True):
if self.instance.pk:
post = super(PostForm, self).save(commit=False)
if self.user:
post.user = self.user
if post.topic.head == post:
post.topic.name = self.cleaned_data['name']
if self.may_create_poll:
post.topic.poll_type = self.cleaned_data['poll_type']
post.topic.poll_question = self.cleaned_data['poll_question']
post.topic.updated = tznow()
if commit:
post.topic.save()
post.updated = tznow()
if commit:
post.save()
return post, post.topic
allow_post = True
if defaults.PYBB_PREMODERATION:
allow_post = defaults.PYBB_PREMODERATION(self.user, self.cleaned_data['body'])
if self.forum:
topic = Topic(
forum=self.forum,
user=self.user,
name=self.cleaned_data['name'],
poll_type=self.cleaned_data.get('poll_type', Topic.POLL_TYPE_NONE),
poll_question=self.cleaned_data.get('poll_question', None),
slug=self.cleaned_data.get('slug', None),
)
if not allow_post:
topic.on_moderation = True
else:
topic = self.topic
post = Post(user=self.user, user_ip=self.ip, body=self.cleaned_data['body'])
if not allow_post:
post.on_moderation = True
if commit:
topic.save()
post.topic = topic
post.save()
return post, topic
def save(self, commit=True):
topic = super(MoveTopicForm, self).save(commit=False)
if 'target_forums' in self.changed_data:
forum_id = int(self.cleaned_data['target_forums'])
target_forum = Forum.objects.get(pk=forum_id)
if not pybb_can_administer_forum(self.user, target_forum):
raise PermissionDenied
topic.forum = target_forum
topic.save()
# Unsubscribe all users that aren't allowed to read the target
# forum
list_subscribers = list(topic.subscribers.all())
for subscriber in list_subscribers:
if not pybb_can_view_forum(subscriber, target_forum):
topic.subscribers.remove(subscriber)
if self.cleaned_data['leave_placeholder']:
t = template.loader.get_template(
'pybb/messages/topic_moved.html')
post_body = mark_safe(
t.render(Context({
'topic': topic,
'domain': self.domain
})))
placeholder_topic = Topic(forum=self.original_forum,
user=self.user,
name=topic.name,
status=2,
topic_after_move=topic,
hidden=topic.hidden)
placeholder_topic.save()
post = Post(topic=placeholder_topic,
user=self.user,
user_ip=self.ip,
body=post_body)
post.save()
return self.instance
def test_topic_merge_complete(self):
topic = Topic(name='xtopic', forum=self.forum, user=self.user)
topic.save()
post = Post(topic=topic, user=self.user, body='one')
post.save()
topic_merge_url = reverse('pybb:topic_merge')
self.login_as(self.staff)
response = self.client.post(topic_merge_url,
data={
'topic_ids': [topic.pk],
'form-TOTAL_FORMS': 1,
'form-INITIAL_FORMS': 0,
'form-0-topic': self.topic.pk,
'submit': 1
})
self.assertRedirects(response, reverse('pybb:index'))
def save(self, *args, **kwargs):
if self.forum:
topic_is_new = True
topic = Topic(forum=self.forum,
user=self.user,
name=self.cleaned_data['name'])
topic.save(*args, **kwargs)
else:
topic_is_new = False
topic = self.topic
post = Post(topic=topic,
user=self.user,
markup=self.cleaned_data['markup'],
body=self.cleaned_data['body'])
post.save(*args, **kwargs)
if pybb_settings.ATTACHMENT_ENABLE:
self.save_attachment(post, self.cleaned_data['attachment'])
return post
def test_topic_redirect(self):
response = self.client.get(self.topic.get_absolute_url())
self.assertEqual(response.status_code, 200)
topic = Topic(name='to-etopic', forum=self.forum, user=self.user)
topic.save()
self.topic.redirect = True
self.topic.save()
redirection = TopicRedirection.objects.create(from_topic=self.topic,
to_topic=topic)
response = self.client.get(self.topic.get_absolute_url())
self.assertRedirects(response,
topic.get_absolute_url(),
status_code=301)
redirection.type = TopicRedirection.TYPE_NO_REDIRECT
redirection.save()
response = self.client.get(self.topic.get_absolute_url())
self.assertEqual(response.status_code, 404)
redirection.type = TopicRedirection.TYPE_EXPIRING_REDIRECT
redirection.expired = date.today() + timedelta(days=1)
redirection.save()
response = self.client.get(self.topic.get_absolute_url())
self.assertRedirects(response, topic.get_absolute_url())
redirection.expired = date.today() - timedelta(days=1)
redirection.save()
response = self.client.get(self.topic.get_absolute_url())
self.assertEqual(response.status_code, 404)
def save(self):
if self.forum:
topic = Topic(forum=self.forum,
user=self.user,
name=self.cleaned_data['name'])
topic.save()
else:
topic = self.topic
post = Post(topic=topic,
user=self.user,
user_ip=self.ip,
markup=self.user.pybb_profile.markup,
body=self.cleaned_data['body'])
post.save()
if settings.PYBB_ATTACHMENT_ENABLE:
for f in self.files:
self.save_attachment(post, self.files[f])
return post
def save(self):
if self.forum:
topic = Topic(forum=self.forum, user=self.user, name=self.cleaned_data["name"])
topic.save()
else:
topic = self.topic
post = Post(
topic=topic,
user=self.user,
user_ip=self.ip,
markup=self.user.pybb_profile.markup,
body=self.cleaned_data["body"],
)
post.save()
if settings.PYBB_ATTACHMENT_ENABLE:
for f in self.files:
self.save_attachment(post, self.files[f])
return post
def test_topic_move_complete(self):
topic = Topic(name='xtopic', forum=self.forum, user=self.user)
topic.save()
post = Post(topic=topic, user=self.user, body='one')
post.save()
topic_merge_url = reverse('pybb:topic_move')
self.login_as(self.staff)
name = 'new name for a topic'
response = self.client.post(
topic_merge_url,
data={
'topic_ids': [topic.pk],
'form-TOTAL_FORMS': 1,
'form-INITIAL_FORMS': 0,
'form-0-name': name,
'form-0-forum': self.forum.pk,
'form-0-redirection_type':
TopicRedirection.TYPE_PERMANENT_REDIRECT,
'submit': 1
})
self.assertRedirects(response, reverse('pybb:index'))
topic = Topic.objects.get(pk=topic.pk)
self.assertTrue(topic.redirect)
self.failUnless(topic.redirection is not None)
redirection = topic.redirection
new_topic = redirection.to_topic
self.assertEqual(new_topic.name, name)
self.assertEqual(new_topic.post_count, 1)
def test_topic_redirect(self):
response = self.client.get(self.topic.get_absolute_url())
self.assertEqual(response.status_code, 200)
topic = Topic(name='to-etopic', forum=self.forum, user=self.user)
topic.save()
self.topic.redirect = True
self.topic.save()
redirection = TopicRedirection.objects.create(from_topic=self.topic, to_topic=topic)
response = self.client.get(self.topic.get_absolute_url())
self.assertRedirects(response, topic.get_absolute_url(), status_code=301)
redirection.type = TopicRedirection.TYPE_NO_REDIRECT
redirection.save()
response = self.client.get(self.topic.get_absolute_url())
self.assertEqual(response.status_code, 404)
redirection.type = TopicRedirection.TYPE_EXPIRING_REDIRECT
redirection.expired = date.today() + timedelta(days=1)
redirection.save()
response = self.client.get(self.topic.get_absolute_url())
self.assertRedirects(response, topic.get_absolute_url())
redirection.expired = date.today() - timedelta(days=1)
redirection.save()
response = self.client.get(self.topic.get_absolute_url())
self.assertEqual(response.status_code, 404)
def get_or_create_topic(self, allow_post):
if self._forum:
topic = Topic(
forum=self._forum,
user=self.user,
name=self.cleaned_data['name'],
)
if not allow_post:
topic.on_moderation = topic.MODERATION_IS_IN_MODERATION
topic.save()
if not defaults.PYBB_DISABLE_POLLS:
if 'poll_type' in self.cleaned_data and self.cleaned_data[
'poll_type'] != Poll.TYPE_NONE:
poll = Poll(type=self.cleaned_data['poll_type'],
question=self.cleaned_data['poll_question'])
poll.save()
topic.poll = poll
else:
topic = self._topic
return topic
def save(self, commit=True):
if self.instance.pk:
post = super(PostForm, self).save(commit=False)
if self.user:
post.user = self.user
if self.actor and post.user_id == self.actor.pk:
if post.is_updatable():
post.updated = tznow()
if post.topic.head == post:
topic = post.topic
topic.name = self.cleaned_data['name']
topic.updated = tznow()
topic.save()
if not defaults.PYBB_DISABLE_POLLS:
if self.cleaned_data['poll_type'] != Poll.TYPE_NONE:
poll = topic.poll or Poll()
poll.type = self.cleaned_data['poll_type']
poll.question = self.cleaned_data['poll_question']
is_new = poll.pk is None
poll.save()
if is_new:
topic.poll = poll
topic.save()
else:
if topic.poll:
topic.poll.answers.all().delete()
topic.poll = None
topic.save()
post.save()
return post
allow_post = True
if defaults.PYBB_PREMODERATION:
allow_post = defaults.PYBB_PREMODERATION(self.user, self.cleaned_data['body'])
if 'forum' in self.cleaned_data and not self._forum:
self._forum = self.cleaned_data['forum']
if self._forum:
topic = Topic(
forum=self._forum,
user=self.user,
name=self.cleaned_data['name'],
)
if not allow_post:
topic.on_moderation = True
topic.save()
if not defaults.PYBB_DISABLE_POLLS:
if 'poll_type' in self.cleaned_data and self.cleaned_data['poll_type'] != Poll.TYPE_NONE:
poll = Poll(
type=self.cleaned_data['poll_type'],
question=self.cleaned_data['poll_question']
)
poll.save()
topic.poll = poll
else:
topic = self._topic
post = Post(topic=topic, user=self.user, user_ip=self.ip,
body=self.cleaned_data['body'], hash=self.cleaned_data['hash'])
if not allow_post:
post.on_moderation = True
post.save()
return post
def test_staff(self):
client = Client()
parent = Forum(name='hcat', staff=True)
parent.save()
forum_in_staff = Forum(name='in_staff', forum=parent)
forum_in_staff.save()
topic_in_staff = Topic(forum=forum_in_staff,
name='in_staff',
user=self.user)
topic_in_staff.save()
forum_staff = Forum(name='staff', forum=self.parent_forum, staff=True)
forum_staff.save()
topic_staff = Topic(forum=forum_staff, name='staff', user=self.user)
topic_staff.save()
post_staff = Post(topic=topic_staff, user=self.user, body='staff')
post_staff.save()
post_in_staff = Post(topic=topic_in_staff,
user=self.user,
body='staff')
post_in_staff.save()
self.assertFalse(parent.id in [
c.id for c in client.get(reverse('pybb:index')).context['forums']
])
self.assertEqual(
client.get(parent.get_absolute_url()).status_code, 404)
self.assertEqual(
client.get(forum_in_staff.get_absolute_url()).status_code, 404)
self.assertEqual(
client.get(topic_in_staff.get_absolute_url()).status_code, 404)
self.assertEqual(
client.get(forum_staff.get_absolute_url()).status_code, 404)
self.assertEqual(
client.get(topic_staff.get_absolute_url()).status_code, 404)
client.login(username='******', password='******')
self.assertFalse(parent.id in [
c.id for c in client.get(reverse('pybb:index')).context['forums']
])
self.assertNotContains(client.get(reverse('pybb:index')),
forum_staff.get_absolute_url())
self.assertEqual(
client.get(parent.get_absolute_url()).status_code, 404)
self.assertEqual(
client.get(forum_in_staff.get_absolute_url()).status_code, 404)
self.assertEqual(
client.get(topic_in_staff.get_absolute_url()).status_code, 404)
self.assertEqual(
client.get(forum_staff.get_absolute_url()).status_code, 404)
self.assertEqual(
client.get(topic_staff.get_absolute_url()).status_code, 404)
self.user.is_staff = True
self.user.save()
self.assertTrue(parent.id in [
c.id for c in client.get(reverse('pybb:index')).context['forums']
])
self.assertEqual(
client.get(parent.get_absolute_url()).status_code, 200)
self.assertEqual(
client.get(forum_in_staff.get_absolute_url()).status_code, 200)
self.assertEqual(
client.get(topic_in_staff.get_absolute_url()).status_code, 200)
self.assertEqual(
client.get(forum_staff.get_absolute_url()).status_code, 200)
self.assertEqual(
client.get(topic_staff.get_absolute_url()).status_code, 200)
def test_compute(self):
# initials
self.assertEqual(self.forum.forum_count, 0)
self.assertEqual(self.parent_forum.forum_count, 1)
self.topic
self.post
parent_forum = Forum.objects.get(pk=self.parent_forum.pk)
self.assertEqual(parent_forum.topic_count, 1)
self.assertEqual(parent_forum.post_count, 1)
self.assertEqual(self.forum.topic_count, 1)
self.assertEqual(self.forum.post_count, 1)
# Add (forum --> topic --> post) in self.forum
forum = Forum.objects.create(name='bar', description='bar', forum=self.forum)
topic = Topic(name='bar', forum=forum, user=self.user)
topic.save()
post = Post(topic=topic, user=self.user, body='my new post')
post.save()
self.assertEquals(Topic.objects.get(pk=topic.pk).first_post, post)
self.assertEqual(topic.post_count, 1)
self.assertEqual(forum.topic_count, 1)
self.assertEqual(Forum.objects.get(pk=self.forum.pk).post_count, 2)
self.assertEqual(Forum.objects.get(pk=self.forum.pk).topic_count, 2)
self.assertEqual(Forum.objects.get(pk=self.parent_forum.pk).post_count, 2)
self.assertEqual(Forum.objects.get(pk=self.parent_forum.pk).topic_count, 2)
# Add (topic --> post) in self.forum
new_topic = Topic(name='foo', forum=self.forum, user=self.user)
new_topic.save()
new_post = Post(topic=new_topic, user=self.user, body='my new post')
new_post.save()
self.assertEquals(Topic.objects.get(pk=new_topic.pk).first_post, new_post)
self.assertEqual(Forum.objects.get(pk=self.forum.pk).post_count, 3)
self.assertEqual(Forum.objects.get(pk=self.forum.pk).topic_count, 3)
self.assertEqual(Forum.objects.get(pk=self.parent_forum.pk).post_count, 3)
self.assertEqual(Forum.objects.get(pk=self.parent_forum.pk).topic_count, 3)
# delete post in (self.parent_forum --> self.forum --> forum --> topic)
post.mark_as_deleted(commit=True)
self.assertTrue(Topic.objects.get(pk=topic.pk).deleted)
self.assertEquals(Topic.objects.get(pk=topic.pk).first_post, post)
self.assertTrue(Topic.objects.get(pk=topic.pk).first_post.deleted)
self.assertEqual(Forum.objects.get(pk=forum.pk).topic_count, 0)
self.assertEqual(Forum.objects.get(pk=forum.pk).post_count, 0)
self.assertEqual(Forum.objects.get(pk=self.forum.pk).topic_count, 2)
self.assertEqual(Forum.objects.get(pk=self.parent_forum.pk).topic_count, 2)
# delete new_topic in (self.parent_forum --> self.forum)
new_topic.mark_as_deleted()
self.assertTrue(Topic.objects.get(pk=new_topic.pk).deleted)
self.assertEquals(Topic.objects.get(pk=new_topic.pk).first_post, new_post)
self.assertTrue(Topic.objects.get(pk=new_topic.pk).first_post.deleted)
self.assertEqual(Forum.objects.get(pk=self.forum.pk).topic_count, 1)
self.assertEqual(Forum.objects.get(pk=self.parent_forum.pk).topic_count, 1)
# undelete new_topic in (self.parent_forum --> self.forum)
new_topic.mark_as_undeleted()
self.assertFalse(Topic.objects.get(pk=new_topic.pk).deleted)
self.assertEquals(Topic.objects.get(pk=new_topic.pk).first_post, new_post)
self.assertFalse(Topic.objects.get(pk=new_topic.pk).first_post.deleted)
self.assertEqual(Forum.objects.get(pk=self.forum.pk).topic_count, 2)
self.assertEqual(Forum.objects.get(pk=self.parent_forum.pk).topic_count, 2)
# undelete post in (self.parent_forum --> self.forum --> forum --> topic)
post.mark_as_undeleted(commit=True)
self.assertFalse(Topic.objects.get(pk=topic.pk).deleted)
self.assertEquals(Topic.objects.get(pk=topic.pk).first_post, post)
self.assertFalse(Topic.objects.get(pk=topic.pk).first_post.deleted)
self.assertEqual(Forum.objects.get(pk=forum.pk).post_count, 1)
self.assertEqual(Forum.objects.get(pk=forum.pk).topic_count, 1)
self.assertEqual(Forum.objects.get(pk=self.forum.pk).post_count, 3)
self.assertEqual(Forum.objects.get(pk=self.forum.pk).topic_count, 3)
self.assertEqual(Forum.objects.get(pk=self.parent_forum.pk).post_count, 3)
self.assertEqual(Forum.objects.get(pk=self.parent_forum.pk).topic_count, 3)
def test_compute(self):
# initials
self.assertEqual(self.forum.forum_count, 0)
self.assertEqual(self.parent_forum.forum_count, 1)
self.topic
self.post
parent_forum = Forum.objects.get(pk=self.parent_forum.pk)
self.assertEqual(parent_forum.topic_count, 1)
self.assertEqual(parent_forum.post_count, 1)
self.assertEqual(self.forum.topic_count, 1)
self.assertEqual(self.forum.post_count, 1)
# Add (forum --> topic --> post) in self.forum
forum = Forum.objects.create(name='bar',
description='bar',
forum=self.forum)
topic = Topic(name='bar', forum=forum, user=self.user)
topic.save()
post = Post(topic=topic, user=self.user, body='my new post')
post.save()
self.assertEquals(Topic.objects.get(pk=topic.pk).first_post, post)
self.assertEqual(topic.post_count, 1)
self.assertEqual(forum.topic_count, 1)
self.assertEqual(Forum.objects.get(pk=self.forum.pk).post_count, 2)
self.assertEqual(Forum.objects.get(pk=self.forum.pk).topic_count, 2)
self.assertEqual(
Forum.objects.get(pk=self.parent_forum.pk).post_count, 2)
self.assertEqual(
Forum.objects.get(pk=self.parent_forum.pk).topic_count, 2)
# Add (topic --> post) in self.forum
new_topic = Topic(name='foo', forum=self.forum, user=self.user)
new_topic.save()
new_post = Post(topic=new_topic, user=self.user, body='my new post')
new_post.save()
self.assertEquals(
Topic.objects.get(pk=new_topic.pk).first_post, new_post)
self.assertEqual(Forum.objects.get(pk=self.forum.pk).post_count, 3)
self.assertEqual(Forum.objects.get(pk=self.forum.pk).topic_count, 3)
self.assertEqual(
Forum.objects.get(pk=self.parent_forum.pk).post_count, 3)
self.assertEqual(
Forum.objects.get(pk=self.parent_forum.pk).topic_count, 3)
# delete post in (self.parent_forum --> self.forum --> forum --> topic)
post.mark_as_deleted(commit=True)
self.assertTrue(Topic.objects.get(pk=topic.pk).deleted)
self.assertEquals(Topic.objects.get(pk=topic.pk).first_post, post)
self.assertTrue(Topic.objects.get(pk=topic.pk).first_post.deleted)
self.assertEqual(Forum.objects.get(pk=forum.pk).topic_count, 0)
self.assertEqual(Forum.objects.get(pk=forum.pk).post_count, 0)
self.assertEqual(Forum.objects.get(pk=self.forum.pk).topic_count, 2)
self.assertEqual(
Forum.objects.get(pk=self.parent_forum.pk).topic_count, 2)
# delete new_topic in (self.parent_forum --> self.forum)
new_topic.mark_as_deleted()
self.assertTrue(Topic.objects.get(pk=new_topic.pk).deleted)
self.assertEquals(
Topic.objects.get(pk=new_topic.pk).first_post, new_post)
self.assertTrue(Topic.objects.get(pk=new_topic.pk).first_post.deleted)
self.assertEqual(Forum.objects.get(pk=self.forum.pk).topic_count, 1)
self.assertEqual(
Forum.objects.get(pk=self.parent_forum.pk).topic_count, 1)
# undelete new_topic in (self.parent_forum --> self.forum)
new_topic.mark_as_undeleted()
self.assertFalse(Topic.objects.get(pk=new_topic.pk).deleted)
self.assertEquals(
Topic.objects.get(pk=new_topic.pk).first_post, new_post)
self.assertFalse(Topic.objects.get(pk=new_topic.pk).first_post.deleted)
self.assertEqual(Forum.objects.get(pk=self.forum.pk).topic_count, 2)
self.assertEqual(
Forum.objects.get(pk=self.parent_forum.pk).topic_count, 2)
# undelete post in (self.parent_forum --> self.forum --> forum --> topic)
post.mark_as_undeleted(commit=True)
self.assertFalse(Topic.objects.get(pk=topic.pk).deleted)
self.assertEquals(Topic.objects.get(pk=topic.pk).first_post, post)
self.assertFalse(Topic.objects.get(pk=topic.pk).first_post.deleted)
self.assertEqual(Forum.objects.get(pk=forum.pk).post_count, 1)
self.assertEqual(Forum.objects.get(pk=forum.pk).topic_count, 1)
self.assertEqual(Forum.objects.get(pk=self.forum.pk).post_count, 3)
self.assertEqual(Forum.objects.get(pk=self.forum.pk).topic_count, 3)
self.assertEqual(
Forum.objects.get(pk=self.parent_forum.pk).post_count, 3)
self.assertEqual(
Forum.objects.get(pk=self.parent_forum.pk).topic_count, 3)
def test_hidden(self):
client = Client()
parent = Forum(name='hcat', hidden=True)
parent.save()
forum_in_hidden = Forum(name='in_hidden', forum=parent)
forum_in_hidden.save()
topic_in_hidden = Topic(forum=forum_in_hidden,
name='in_hidden',
user=self.user)
topic_in_hidden.save()
forum_hidden = Forum(name='hidden',
forum=self.parent_forum,
hidden=True)
forum_hidden.save()
topic_hidden = Topic(forum=forum_hidden, name='hidden', user=self.user)
topic_hidden.save()
post_hidden = Post(topic=topic_hidden, user=self.user, body='hidden')
post_hidden.save()
post_in_hidden = Post(topic=topic_in_hidden,
user=self.user,
body='hidden')
post_in_hidden.save()
self.assertFalse(parent.id in [
c.id for c in client.get(reverse('pybb:index')).context['forums']
])
self.assertEqual(
client.get(parent.get_absolute_url()).status_code, 302)
self.assertEqual(
client.get(forum_in_hidden.get_absolute_url()).status_code, 302)
self.assertEqual(
client.get(topic_in_hidden.get_absolute_url()).status_code, 302)
self.assertNotContains(client.get(reverse('pybb:index')),
forum_hidden.get_absolute_url())
self.assertNotContains(client.get(reverse('pybb:feed_topics')),
topic_hidden.get_absolute_url())
self.assertNotContains(client.get(reverse('pybb:feed_topics')),
topic_in_hidden.get_absolute_url())
self.assertNotContains(client.get(reverse('pybb:feed_posts')),
post_hidden.get_absolute_url())
self.assertNotContains(client.get(reverse('pybb:feed_posts')),
post_in_hidden.get_absolute_url())
self.assertEqual(
client.get(forum_hidden.get_absolute_url()).status_code, 302)
self.assertEqual(
client.get(topic_hidden.get_absolute_url()).status_code, 302)
client.login(username='******', password='******')
self.assertTrue(parent.id in [
c.id for c in client.get(reverse('pybb:index')).context['forums']
])
self.assertContains(client.get(reverse('pybb:index')),
forum_hidden.get_absolute_url())
self.assertEqual(
client.get(parent.get_absolute_url()).status_code, 200)
self.assertEqual(
client.get(forum_in_hidden.get_absolute_url()).status_code, 200)
self.assertEqual(
client.get(topic_in_hidden.get_absolute_url()).status_code, 200)
self.assertEqual(
client.get(forum_hidden.get_absolute_url()).status_code, 200)
self.assertEqual(
client.get(topic_hidden.get_absolute_url()).status_code, 200)
self.user.is_staff = True
self.user.save()
self.assertTrue(parent.id in [
c.id for c in client.get(reverse('pybb:index')).context['forums']
])
self.assertContains(client.get(reverse('pybb:index')),
forum_hidden.get_absolute_url())
self.assertEqual(
client.get(parent.get_absolute_url()).status_code, 200)
self.assertEqual(
client.get(forum_in_hidden.get_absolute_url()).status_code, 200)
self.assertEqual(
client.get(topic_in_hidden.get_absolute_url()).status_code, 200)
self.assertEqual(
client.get(forum_hidden.get_absolute_url()).status_code, 200)
self.assertEqual(
client.get(topic_hidden.get_absolute_url()).status_code, 200)
def test_read_tracking_multi_user(self):
self.post
topic_1 = self.topic
topic_2 = Topic(name='topic_2', forum=self.forum, user=self.user)
topic_2.save()
Post(topic=topic_2, user=self.user, body='one').save()
user_ann = get_user_model().objects.create_user('ann', 'ann@localhost', 'ann')
client_ann = Client()
client_ann.login(username='******', password='******')
user_bob = get_user_model().objects.create_user('bob', 'bob@localhost', 'bob')
client_bob = Client()
client_bob.login(username='******', password='******')
# Two topics, each with one post. everything is unread, so the db should reflect that:
self.assertEqual(TopicReadTracker.objects.all().count(), 0)
self.assertEqual(ForumReadTracker.objects.all().count(), 0)
# user_ann reads topic_1, she should get one topic read tracker, there should be no forum read trackers
client_ann.get(topic_1.get_absolute_url())
self.assertEqual(TopicReadTracker.objects.all().count(), 1)
self.assertEqual(TopicReadTracker.objects.filter(user=user_ann).count(), 1)
self.assertEqual(TopicReadTracker.objects.filter(user=user_ann, topic=topic_1).count(), 1)
self.assertEqual(ForumReadTracker.objects.all().count(), 0)
# user_bob reads topic_1, he should get one topic read tracker, there should be no forum read trackers
client_bob.get(topic_1.get_absolute_url())
self.assertEqual(TopicReadTracker.objects.all().count(), 2)
self.assertEqual(TopicReadTracker.objects.filter(user=user_bob).count(), 1)
self.assertEqual(TopicReadTracker.objects.filter(user=user_bob, topic=topic_1).count(), 1)
self.assertEqual(ForumReadTracker.objects.all().count(), 0)
# user_bob reads topic_2, he should get a forum read tracker,
# there should be no topic read trackers for user_bob
client_bob.get(topic_2.get_absolute_url())
self.assertEqual(TopicReadTracker.objects.all().count(), 1)
self.assertEqual(ForumReadTracker.objects.all().count(), 1)
self.assertEqual(ForumReadTracker.objects.filter(user=user_bob).count(), 1)
self.assertEqual(ForumReadTracker.objects.filter(user=user_bob, forum=self.forum).count(), 1)
# user_ann creates topic_3, there should be a new topic read tracker in the db
topic_create_url = reverse('pybb:topic_create', kwargs={'forum_id': self.forum.id})
response = client_ann.get(topic_create_url)
values = self.get_form_values(response)
values['body'] = 'topic_3'
values['name'] = 'topic_3'
values['poll_type'] = 0
response = client_ann.post(topic_create_url, data=values, follow=True)
self.assertEqual(TopicReadTracker.objects.all().count(), 2)
self.assertEqual(TopicReadTracker.objects.filter(user=user_ann).count(), 2)
self.assertEqual(ForumReadTracker.objects.all().count(), 1)
topic_3 = Topic.objects.order_by('-updated')[0]
self.assertEqual(topic_3.name, 'topic_3')
# user_ann posts to topic_1, a topic they've already read, no new trackers should be created (existing one is updated)
post_create_url = reverse('pybb:post_create', kwargs={'topic_id': topic_1.id})
response = client_ann.get(post_create_url)
values = self.get_form_values(response)
values['body'] = 'test tracking'
response = client_ann.post(post_create_url, values, follow=True)
self.assertEqual(TopicReadTracker.objects.all().count(), 2)
self.assertEqual(TopicReadTracker.objects.filter(user=user_ann).count(), 2)
self.assertEqual(ForumReadTracker.objects.all().count(), 1)
self.assertEqual(ForumReadTracker.objects.all().count(), 1)
previous_time = ForumReadTracker.objects.all()[0].time_stamp
# user bob reads topic 1 which he already read, topic tracker recreated, forum tracker untouched (topic 3 still unread)
client_bob.get(topic_1.get_absolute_url())
self.assertEqual(ForumReadTracker.objects.all().count(), 1)
self.assertEqual(ForumReadTracker.objects.all()[0].time_stamp, previous_time)
self.assertEqual(TopicReadTracker.objects.all().count(), 3)
self.assertEqual(TopicReadTracker.objects.filter(user=user_bob).count(), 1)
self.assertEqual(ForumReadTracker.objects.all().count(), 1)
previous_time = ForumReadTracker.objects.all()[0].time_stamp
# user bob reads topic 3, topic tracker purged, forum tracker updated
client_bob.get(topic_3.get_absolute_url())
self.assertEqual(ForumReadTracker.objects.all().count(), 1)
self.assertGreater(ForumReadTracker.objects.all()[0].time_stamp, previous_time)
self.assertEqual(TopicReadTracker.objects.all().count(), 2)
self.assertEqual(TopicReadTracker.objects.filter(user=user_bob).count(), 0)
def save(self, commit=True):
related_object = None
if self.cleaned_data.get('related_object', None):
try:
related_object_pk = int(self.cleaned_data['related_object'].pk)
except ValueError:
pass
else:
related_object = self.related_object_type.get_object_for_this_type(
pk=related_object_pk)
if self.instance.pk:
now = datetime.now()
topic = super(TopicForm, self).save(commit=False)
topic.related_object = related_object
if 'name' in self.fields.keys():
topic.name = self.cleaned_data['name']
if 'importance' in self.fields.keys():
topic.importance = self.cleaned_data['importance']
topic.updated = now
if self.cleaned_data.get('subscribe', False):
topic.subscribers.add(self.user)
else:
topic.subscribers.remove(self.user)
if commit:
topic.save()
post = topic.head
post.body = self.cleaned_data['body']
post.updated = now
if commit:
if self.post_form:
# Post form has some additional save logic (add edit user,
# topic subscription)
post = self.post_form.save(commit=True)
else:
post.save()
return post
topic = Topic(forum=self.forum,
user=self.user,
name=self.cleaned_data['name'])
if self.cleaned_data.has_key('importance'):
topic.importance = self.cleaned_data['importance']
topic.related_object = related_object
if commit:
topic.save()
if self.cleaned_data['subscribe']:
topic.subscribers.add(self.user)
topic.save()
post = Post(topic=topic,
user=self.user,
user_ip=self.ip,
body=self.cleaned_data['body'])
if commit:
post.save()
return post