conftest.py

# -*- coding: utf-8 -*-
importmock
importpytest
fromappimportappas_app, SQLAlchemyDB
fromapp.databaseimportmodel
fromapp.utilsimportRequestUtil
importapp.utils.SshUtilasssh
from .importsuccess, load_data

# =====================================
# base fixtures
# =====================================


@pytest.fixture
defapp():
SQLAlchemyDB.create_all()
yield_app
SQLAlchemyDB.session.close()
SQLAlchemyDB.drop_all()


@pytest.fixture
defclient(app):
withapp.test_client() asc:
yieldc


@pytest.fixture
defsql(app):
returnSQLAlchemyDB.session


@pytest.fixture
deftester(app, sql):
"""模拟已登录用户"""
withapp.test_client() asc:
user=create_user(sql)
mock_func=mock.MagicMock(return_value=user.dict())
withmock.patch.object(RequestUtil, 'get_login_user', new=mock_func):
yieldc


defcreate_user(sql):
user_id='tester'
user=model.User(
id=user_id,
name=user_id,
location='',
avatar=''
 )
sql.add(user)
sql.commit()
returnuser


# =====================================
# server fixtures
# =====================================
SERVER_DATA= {
'ip': '127.0.0.1',
'name': 'dev',
'port': '22',
'account': 'root',
'pkey': 'asdfghjkl',
}


defmock_do_ssh_cmd(*args, **kwargs):
returnTrue, "OK"


@pytest.fixture
defcreate_server(tester):
deffunc(**kwargs):
data=SERVER_DATA.copy()
data.update(kwargs)
withmock.patch.object(ssh, 'do_ssh_cmd', new=mock_do_ssh_cmd):
resp=tester.post('/api/server/new', data=data)
assertsuccess(resp)
returnload_data(resp)
returnfunc


# =====================================
# webhook fixtures
# =====================================
WEBHOOK_DATA= {
'repo': 'git-webhook',
'branch': 'master',
'shell': 'echo hello',
}


@pytest.fixture
defcreate_webhook(tester):
deffunc(**kwargs):
data=WEBHOOK_DATA.copy()
data.update(kwargs)
resp=tester.post('/api/webhook/new', data=data)
assertsuccess(resp)
returnload_data(resp)
returnfunc