from django.test import TestCase
from boards.models import Post


class ParserTest(TestCase):
    def test_preparse_quote(self):
        raw_text = '>quote\nQuote in >line\nLine\n>Quote'
        preparsed_text = Post.objects._preparse_text(raw_text)

        self.assertEqual(
            '[quote]quote[/quote]\nQuote in >line\nLine\n[quote]Quote[/quote]',
            preparsed_text, 'Quote not preparsed.')

    def test_preparse_comment(self):
        raw_text = '//comment'
        preparsed_text = Post.objects._preparse_text(raw_text)

        self.assertEqual('[comment]comment[/comment]', preparsed_text,
                         'Comment not preparsed.')

    def test_preparse_reflink(self):
        raw_text = '>>12\nText'
        preparsed_text = Post.objects._preparse_text(raw_text)

        self.assertEqual('[post]12[/post]\nText',
                         preparsed_text, 'Reflink not preparsed.')

    def preparse_user(self):
        raw_text = '@user\nuser@example.com\n@user\nuser @user'
        preparsed_text = Post.objects._preparse_text(raw_text)

        self.assertEqual('[user]user[/user]\nuser@example.com\n[user]user[/user]\nuser [user]user[/user]',
                         preparsed_text, 'User link not preparsed.')