You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
79 lines
3.3 KiB
79 lines
3.3 KiB
# -*- coding: utf-8 -*-
|
|
"""
|
|
:author: Grey Li (李辉)
|
|
:url: http://greyli.com
|
|
:copyright: © 2018 Grey Li <withlihui@gmail.com>
|
|
:license: MIT, see LICENSE for more details.
|
|
"""
|
|
from bluelog.models import Admin, Post, Category, Comment
|
|
from bluelog.extensions import db
|
|
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', '--username', 'grey', '--password', '123'])
|
|
self.assertIn('Creating the temporary administrator account...', result.output)
|
|
self.assertIn('Creating the default category...', result.output)
|
|
self.assertIn('Done.', result.output)
|
|
self.assertEqual(Admin.query.count(), 1)
|
|
self.assertEqual(Admin.query.first().username, 'grey')
|
|
self.assertEqual(Category.query.first().name, 'Default')
|
|
|
|
def test_init_command_with_update(self):
|
|
self.runner.invoke(args=['init', '--username', 'grey', '--password', '123'])
|
|
result = self.runner.invoke(args=['init', '--username', 'new grey', '--password', '123'])
|
|
self.assertIn('The administrator already exists, updating...', result.output)
|
|
self.assertNotIn('Creating the temporary administrator account...', result.output)
|
|
self.assertEqual(Admin.query.count(), 1)
|
|
self.assertEqual(Admin.query.first().username, 'new grey')
|
|
self.assertEqual(Category.query.first().name, 'Default')
|
|
|
|
def test_forge_command(self):
|
|
result = self.runner.invoke(args=['forge'])
|
|
|
|
self.assertEqual(Admin.query.count(), 1)
|
|
self.assertIn('Generating the administrator...', result.output)
|
|
|
|
self.assertEqual(Post.query.count(), 50)
|
|
self.assertIn('Generating 50 posts...', result.output)
|
|
|
|
self.assertEqual(Category.query.count(), 10 + 1)
|
|
self.assertIn('Generating 10 categories...', result.output)
|
|
|
|
self.assertEqual(Comment.query.count(), 500 + 50 + 50 + 50)
|
|
self.assertIn('Generating 500 comments...', result.output)
|
|
|
|
self.assertIn('Generating links...', result.output)
|
|
self.assertIn('Done.', result.output)
|
|
|
|
def test_forge_command_with_count(self):
|
|
result = self.runner.invoke(args=['forge', '--category', '5', '--post', '20', '--comment', '100'])
|
|
self.assertEqual(Admin.query.count(), 1)
|
|
self.assertIn('Generating the administrator...', result.output)
|
|
|
|
self.assertEqual(Post.query.count(), 20)
|
|
self.assertIn('Generating 20 posts...', result.output)
|
|
|
|
self.assertEqual(Category.query.count(), 5 + 1)
|
|
self.assertIn('Generating 5 categories...', result.output)
|
|
|
|
self.assertEqual(Comment.query.count(), 100 + 10 + 10 + 10)
|
|
self.assertIn('Generating 100 comments...', result.output)
|
|
|
|
self.assertIn('Generating links...', result.output)
|
|
self.assertIn('Done.', result.output)
|