def test_tagging(self): "Testing tagging." eq = self.assertEqual eq(0, Tag.objects.all().count() ) # Create an admin user and a post. title = "Hello Posts!" email = "*****@*****.**" jane = User.objects.create(email=email) html = "<b>Hello World!</b>" post = Post(title=title, author=jane, type=Post.QUESTION, content=html) post.save() post.add_tags("t1,t2, t3") eq(3, Tag.objects.all().count()) post = Post(title=title, author=jane, type=Post.QUESTION, content=html) post.save() post.add_tags("t1, t2, t3, t2, t1, t1") t1 = Tag.objects.get(name="t1") t3 = Tag.objects.get(name="t3") eq(2, t1.count) eq(2, t3.count) post.add_tags("t2 t4") t1 = Tag.objects.get(name="t1") t3 = Tag.objects.get(name="t3") eq(1, t1.count) eq(1, t3.count)
def create_a_post(self): title = "Post 1, title needs to be sufficiently long" content = ('Lorem ipsum dolor sit amet, consectetur adipisicing elit, sed do eiusmod' 'tempor incididunt ut labore et dolore magna aliqua.') post_type = Post.QUESTION post = Post(title=title, content=content, author=self.user, type=post_type) post.save() return post
def create_post(self, post_type, parent=None, days=3): # Create a post. title = "Post 1, title needs to be sufficiently long" content = ('Lorem ipsum dolor sit amet, consectetur adipisicing elit, sed do eiusmod' 'tempor incididunt ut labore et dolore magna aliqua.') post = Post(title=title, content=content, author=self.user, type=post_type, parent=parent) #post.save() post.creation_date = datetime.today() - timedelta(days=days) post.save() return post
def test_note_creation(self): "Testing notifications" eq = self.assertEqual # Create some users title = "Test" emails = ["*****@*****.**", "*****@*****.**", "*****@*****.**", "*****@*****.**", "*****@*****.**", "*****@*****.**" ] email_count = len(emails) users, posts, user = [], [], None parent = None for email in emails: # Create users. user = User.objects.create(email=email) users.append(user) # A welcome message for each user. eq(note_count(), email_count) # Create a question. first = users[0] post = Post(title=title, author=first, type=Post.QUESTION) post.save() answers = [] for user in users: # Every user adds an answer. answ = Post(author=user, type=Post.ANSWER, parent=post) answ.save() answers.append(answ) # Total number of posts eq(note_count(), 21) # Every user has one subscription to the main post eq(email_count, Subscription.objects.all().count()) # Each user has a messages for content posted after # they started following the thread. for index, user in enumerate(users): mesg_c = Message.objects.filter(user=user).count() eq (mesg_c, email_count - index )
def test_post_creation(self): "Testing post creation." eq = self.assertEqual # Create an admin user and a post. title = "Hello Posts!" email = "*****@*****.**" jane = User.objects.create(email=email) html = "<b>Hello World!</b>" post = Post(title=title, author=jane, type=Post.QUESTION, content=html) post.save() # Get the object fresh. post = Post.objects.get(pk=post.id) eq(post.type, Post.QUESTION) eq(post.root, post) eq(post.parent, post) # Subscriptions are automatically created sub = Subscription.objects.get(user=jane) eq(sub.user, jane) eq(sub.post, post) # A new post triggers a message to the author. email = "*****@*****.**" john = User.objects.create(email=email) answer = Post(author=john, parent=post, type=Post.ANSWER) answer.save() eq(answer.root, post) eq(answer.parent, post) eq(answer.type, Post.ANSWER) # Add comment. The parent will override the post type. email = "*****@*****.**" bob = User.objects.create(email=email) comment = Post(author=bob, type=Post.QUESTION, parent=answer) comment.save() eq(comment.root, post) eq(comment.parent, answer) eq(comment.type, Post.COMMENT) # Everyone posting in a thread gets a subscription to the root post of the subs = Subscription.objects.filter(post=post) eq(len(subs), 3)
def post(self, request, *args, **kwargs): user = request.user post = self.get_obj() post = post_permissions(request, post) # The default return url response = HttpResponseRedirect(post.root.get_absolute_url()) if not post.is_editable: messages.warning(request, "You may not moderate this post") return response # Initialize the form class. form = self.form_class(request.POST, pk=post.id) # Bail out on errors. if not form.is_valid(): messages.error(request, "%s" % form.errors) return response # A shortcut to the clean form data. get = form.cleaned_data.get # These will be used in updates, will bypasses signals. query = Post.objects.filter(pk=post.id) root = Post.objects.filter(pk=post.root_id) action = get('action') if action == OPEN and not user.is_moderator: messages.error(request, "Only a moderator may open a post") return response if action == MOVE_TO_ANSWER and post.type == Post.COMMENT: # This is a valid action only for comments. messages.success(request, "Moved post to answer") query.update(type=Post.ANSWER, parent=post.root) root.update(reply_count=F("reply_count") + 1) return response if action == MOVE_TO_COMMENT and post.type == Post.ANSWER: # This is a valid action only for answers. messages.success(request, "Moved post to answer") query.update(type=Post.COMMENT, parent=post.root) root.update(reply_count=F("reply_count") - 1) return response # Some actions are valid on top level posts only. if action in (CLOSE_OFFTOPIC, DUPLICATE) and not post.is_toplevel: messages.warning(request, "You can only close or open a top level post") return response if action == OPEN: query.update(status=Post.OPEN) messages.success(request, "Opened post: %s" % post.title) return response if action in CLOSE_OFFTOPIC: query.update(status=Post.CLOSED) messages.success(request, "Closed post: %s" % post.title) content = html.render(name="messages/offtopic_posts.html", user=post.author, comment=get("comment"), post=post) comment = Post(content=content, type=Post.COMMENT, parent=post, author=user) comment.save() return response if action == CROSSPOST: content = html.render(name="messages/crossposted.html", user=post.author, comment=get("comment"), post=post) comment = Post(content=content, type=Post.COMMENT, parent=post, author=user) comment.save() return response if action == DUPLICATE: query.update(status=Post.CLOSED) posts = Post.objects.filter(id__in=get("dupe")) content = html.render(name="messages/duplicate_posts.html", user=post.author, comment=get("comment"), posts=posts) comment = Post(content=content, type=Post.COMMENT, parent=post, author=user) comment.save() return response if action == DELETE: # Delete marks a post deleted but does not remove it. # Remove means to delete the post from the database with no trace. # Posts with children or older than some value can only be deleted not removed # The children of a post. children = Post.objects.filter(parent_id=post.id).exclude(pk=post.id) # The condition where post can only be deleted. delete_only = children or post.age_in_days > 7 or post.vote_count > 1 or (post.author != user) if delete_only: # Deleted posts can be undeleted by re-opening them. query.update(status=Post.DELETED) messages.success(request, "Deleted post: %s" % post.title) response = HttpResponseRedirect(post.root.get_absolute_url()) else: # This will remove the post. Redirect depends on the level of the post. url = "/" if post.is_toplevel else post.parent.get_absolute_url() post.delete() messages.success(request, "Removed post: %s" % post.title) response = HttpResponseRedirect(url) # Recompute post reply count post.update_reply_count() return response # By this time all actions should have been performed messages.warning(request, "That seems to be an invalid action for that post. \ It is probably ok! Actions may be shown even when not valid.") return response