# -*- coding: utf-8 -*-
from phshare.extensions import db
from phshare.models import Comment, Role, User, Photo, Tag
from tests.base import BaseTestCase


class CLITestCase(BaseTestCase):

    def setUp(self):
        super(CLITestCase, self).setUp()
        db.drop_all()

    def test_initdb_command(self):
        result = self.runner.invoke(args=['initdb'])
        self.assertIn('Initialized database.', result.output)

    def test_initdb_command_with_drop(self):
        result = self.runner.invoke(args=['initdb', '--drop'], input='y\n')
        self.assertIn('This operation will delete the database, do you want to continue?', result.output)
        self.assertIn('Drop tables.', result.output)

    def test_init_command(self):
        result = self.runner.invoke(args=['init'])
        self.assertIn('Initializing the database...', result.output)
        self.assertIn('Initializing the roles and permissions...', result.output)
        self.assertIn('Done.', result.output)
        self.assertEqual(Role.query.count(), 4)

    def test_forge_command(self):
        pass  # it will take too long time

    def test_forge_command_with_count(self):
        result = self.runner.invoke(args=['forge', '--user', '5', '--follow', '10',
                                          '--photo', '10', '--tag', '10', '--collect', '10',
                                          '--comment', '10'])

        self.assertIn('Initializing the roles and permissions...', result.output)

        self.assertEqual(User.query.count(), 6)
        self.assertIn('Generating the administrator...', result.output)

        self.assertIn('Generating 10 follows...', result.output)

        self.assertEqual(Photo.query.count(), 10)
        self.assertIn('Generating 10 photos...', result.output)

        self.assertEqual(Tag.query.count(), 10)
        self.assertIn('Generating 10 tags...', result.output)

        self.assertIn('Generating 10 collects...', result.output)

        self.assertEqual(Comment.query.count(), 10)
        self.assertIn('Generating 10 comments...', result.output)

        self.assertIn('Done.', result.output)
