def test_project_create(func_dir, func_init_db):
args = app.create_parser().parse_args(
['--db', db.url, 'project', 'create', '-d', func_dir, '-n', 'test'])
assert args.project_dir == func_dir
assert args.project_name == 'test'
assert args.db == db.url
args.handler(args)
db.upgrade()
p = db.session.query(Project).filter_by(path_name=func_dir).first()
assert p is None
# on Windows/Python2, another session is create, need to remove
# this session externally (*)
db._session.remove()
args.handler(args)
p = db.session.query(Project).filter_by(path_name=func_dir).first()
assert p is not None
assert p.path_name == func_dir
assert p.name == 'test'
db._session.remove() # same as (*)
args.handler(args) # already registered, confirm not occur error
ps = db.session.query(Project).filter_by(path_name=func_dir).all()
assert len(ps) == 1
def test_server_debug(func_init_db):
args = app.create_parser().parse_args([
'--db', db.url, '--db-echo', 'server', '-H', 'test.domain', '-p',
'5001', '-d'
])
assert args.host == 'test.domain'
assert args.port == 5001
assert args.debug
assert args.db == db.url
assert args.db_echo
mock_app = MagicMock()
mock_app_creator = MagicMock(return_value=mock_app)
with patch('werkzeug.serving.run_simple', MagicMock()) as f, \
patch('chainerui.app.create_app', mock_app_creator):
args.handler(args)
f.assert_not_called()
db.upgrade()
args.handler(args)
f.assert_called_once()
f.assert_called_with('test.domain',
5001,
mock_app,
use_reloader=True,
use_debugger=True,
threaded=True)
def test_check_current_db_revision(func_init_db):
assert not check_current_db_revision()
_upgrade()
assert not check_current_db_revision()
db.upgrade()
assert check_current_db_revision()
def test_server_production(func_init_db):
args = app.create_parser().parse_args(
['--db', db.url, 'server', '-H', 'test.domain', '-p', '5001'])
assert args.host == 'test.domain'
assert args.port == 5001
assert not args.debug
assert args.db == db.url
assert not args.db_echo
mock_app = MagicMock()
mock_app_creator = MagicMock(return_value=mock_app)
mock_server = MagicMock()
mock_server_init = MagicMock(return_value=mock_server)
with patch('gevent.pywsgi.WSGIServer', mock_server_init), \
patch('chainerui.app.create_app', mock_app_creator):
args.handler(args)
mock_server.serve_forever.assert_not_called()
db.upgrade()
args.handler(args)
mock_server.serve_forever.assert_called_once()
def func_db(func_init_db):
db.upgrade()