Ви не можете вибрати більше 25 тем
Теми мають розпочинатися з літери або цифри, можуть містити дефіси (-) і не повинні перевищувати 35 символів.
61 рядки
2.2 KiB
61 рядки
2.2 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 albumy.extensions import db
|
|
from albumy.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)
|