ForgeDiscussion/forgediscussion/tests/test_app.py

#       Licensed to the Apache Software Foundation (ASF) under one
#       or more contributor license agreements.  See the NOTICE file
#       distributed with this work for additional information
#       regarding copyright ownership.  The ASF licenses this file
#       to you under the Apache License, Version 2.0 (the
#       "License"); you may not use this file except in compliance
#       with the License.  You may obtain a copy of the License at
#
#         http://www.apache.org/licenses/LICENSE-2.0
#
#       Unless required by applicable law or agreed to in writing,
#       software distributed under the License is distributed on an
#       "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
#       KIND, either express or implied.  See the License for the
#       specific language governing permissions and limitations
#       under the License.

import tempfile
import json
import os
from operator import attrgetter
from cgi import FieldStorage
from io import BytesIO
from alluratest.controller import setup_basic_test, setup_unit_test

from tg import tmpl_context as c

from forgediscussion.site_stats import posts_24hr
from ming.orm import ThreadLocalORMSession

from allura import model as M
from allura.tests import decorators as td
from forgediscussion.tests.functional.test_rest import TestDiscussionApiBase
from forgediscussion.model.forum import Forum, ForumThread, ForumPost
from forgediscussion import utils


class TestApp(TestDiscussionApiBase):  # creates some sample data

    @td.with_discussion
    def test_uninstall(self):
        assert ForumPost.query.get(text='Hi boys and girls')
        # c.app.uninstall(c.project) errors out, but works ok in test_uninstall for repo tools.  So instead:
        c.project.uninstall_app('discussion')
        assert not ForumPost.query.get(text='Hi boys and girls')

    @td.with_discussion
    def test_tickets_stats_24hr(self):
        # invoked normally via entry point
        assert 2 == posts_24hr()


class TestAppSitemap:

    def setup_method(self, method):
        setup_basic_test()
        setup_unit_test()
        self.user = M.User.query.get(username='root')

    @td.with_discussion
    def test_sitemap_xml(self):
        assert [] == c.app.sitemap_xml()
        forum = utils.create_forum(c.app, dict(
            shortname='test-forum',
            name="Test Forum",
            description="Test Forum Description",
        ))
        ThreadLocalORMSession.flush_all()
        assert [] == c.app.sitemap_xml()
        thread = ForumThread(
            subject='test-thread',
        )
        thread.set_forum(forum)
        ThreadLocalORMSession.flush_all()

        assert [] == c.app.sitemap_xml()
        thread.post(
            subject='test-post',
            text='this is a test post.',
        )
        ThreadLocalORMSession.flush_all()
        assert 1 == len(c.app.sitemap_xml())


class TestBulkExport(TestDiscussionApiBase):

    def test_bulk_export(self):
        # Clear out some context vars, to properly simulate how this is run from the export task
        # Besides, it's better not to need c context vars
        c.app = c.project = None

        project = M.Project.query.get(shortname='test')
        discussion = project.app_instance('discussion')
        f = tempfile.TemporaryFile('w+')
        discussion.bulk_export(f)
        f.seek(0)
        discussion = json.loads(f.read())
        forums = sorted(discussion['forums'], key=lambda x: x['name'])

        assert forums[0]['shortname'] == 'general'
        assert (
            forums[0]['description'] == 'Forum about anything you want to talk about.')
        assert forums[0]['name'] == 'General Discussion'
        forums[0]['threads'] = sorted(forums[0]['threads'],
                                      key=lambda x: x['posts'][0]['subject'])
        assert (
            forums[0]['threads'][0]['posts'][0]['text'] == 'Hi boys and girls')
        assert (
            forums[0]['threads'][0]['posts'][0]['subject'] == 'Hi guys')
        assert forums[0]['threads'][1]['posts'][0]['text'] == '1st post'
        assert (
            forums[0]['threads'][1]['posts'][0]['subject'] == "Let's talk")
        assert forums[1]['shortname'] == 'héllo'
        assert forums[1]['description'] == 'Say héllo here'
        assert forums[1]['name'] == 'Say Héllo'

    def test_export_with_attachments(self):
        project = M.Project.query.get(shortname='test')
        discussion = project.app_instance('discussion')
        thread = sorted(Forum.query.get(shortname='general').threads,
                        key=attrgetter('last_post_date'))[-1]
        post = thread.first_post
        test_file1 = FieldStorage()
        test_file1.name = 'file_info'
        test_file1.filename = 'test_file'
        test_file1.file = BytesIO(b'test file1\n')
        post.add_attachment(test_file1)
        ThreadLocalORMSession.flush_all()

        f = tempfile.TemporaryFile('w+')
        temp_dir = tempfile.mkdtemp()
        discussion.bulk_export(f, temp_dir, True)
        f.seek(0)
        discussion = json.loads(f.read())
        forums = sorted(discussion['forums'], key=lambda x: x['name'])
        threads = sorted(forums[0]['threads'], key=lambda x: x['subject'])
        file_path = os.path.join(
            'discussion',
            str(post.discussion_id),
            str(post.thread_id),
            post.slug,
            'test_file'
        )
        assert threads[0]['posts'][0]['attachments'][0]['path'] == file_path
        os.path.exists(file_path)