mwgencode 1.4.1__py3-none-any.whl

gencode/gencode/template/test_test_base.tmp

@@ -0,0 +1,34 @@
+import unittest
+from app import create_app_swagger, db
+
+class BasicTestCase(unittest.TestCase):
+    def setUp(self):
+        self.app = create_app_swagger('testing').app
+        self.app_context = self.app.app_context()
+        self.app_context.push()
+        db.create_all()
+        self.client = self.app.test_client()
+        self.server ='http://localhost:9999/{{service_name}}/v1.0'
+
+    def url_for(self,endpoint):
+        return '%s/%s'%(self.server,endpoint)
+
+    def tearDown(self):
+        db.session.remove()
+        db.drop_all()
+        self.app_context.pop()
+
+    # 范例一：测试health
+    # def test_health(self):
+    #     response = self.client.get(self.url_for('health'))
+    #     self.assertEqual(response.status_code,200)
+
+    # def test_register(self):
+    #     response = self.client.post(url_for('main.register'), data={
+    #             'email': '879651072@qq.com',
+    #             'name': 'Hyman',
+    #             'password1': '123',
+    #             'password2': '123'})
+    #     self.assertTrue(response.status_code == 302)
+        
+

gencode/gencode/template/tests/__init__.pys

@@ -0,0 +1,5 @@
+import unittest
+def test():
+    """Run the unit tests."""
+    tests = unittest.TestLoader().discover('tests')
+    unittest.TextTestRunner().run(tests)

gencode/gencode/template/tests/init_test_data.pys

@@ -0,0 +1,80 @@
+from app import db
+import json
+# todo 如果model中没有定义user ，则在此定义
+# class Authuser(db.Model):
+#     __tablename__ = 'appuser'
+#     uid = db.Column("id",db.String(50), primary_key=True)
+#     uname = db.Column("name",db.String(50))
+#     manageuser = db.Column("ismanage",db.Boolean)
+#     manageuserid = db.Column("manageuserid",db.String(50))
+#     password = db.Column("passwordmd5",db.String(50))
+#     systemuser = db.Column("issystemuser",db.Boolean)
+#     companyid = db.Column(db.String(50))
+#     # 邮件，用于修改密码
+#     email = db.Column(db.String(200))
+#     # 微信的openid
+#     wx_openid = db.Column(db.String(50))
+#     def __repr__(self):
+#         return json.dumps(self.to_json())
+#     def to_json(self,fields:list=None):
+#         '''
+#         按资料库中的fieldname输出json
+#         :param fields:资料库栏位名，不指定fields时，输出所有的资料，否则按指定fields输出
+#         :return:
+#         '''
+#         return {key: getattr(self, key) for key in (fields or self.__table__.columns.keys())
+#                    if hasattr(self,key)
+#                }
+def add_seed(session, cls, flt, datas):
+    '''
+    增加物件
+    :param session: sQLAlchemy session
+    :param cls: boclass，如：Employee
+    :param flt: 过滤条件，如：{'id':1000,'name':'cxh'}
+    :param datas: 物件body，如：
+                  {'id':1000,'emp_name':'cxh',
+                   'departid':999}
+    :return: 企业物件，employee
+    '''
+    data = session.query(cls).filter_by(**flt).first()
+    if not data:
+        data = cls(**datas)
+        session.add(data)
+        session.commit()
+    else:
+        session.query(cls).filter_by(**flt).update(datas)
+        session.commit()
+    return data
+
+def init_companys():
+    from app.models import Company
+    comp1 = add_seed(db.session, Company, {'id': 'comp1'},
+             {'id': 'comp1', 'code': 'code1', 'name': 'name1', 'xtype': 1,'inuse':True})
+    comp2 = add_seed(db.session, Company, {'id': 'comp2','inuse':True},
+             {'id': 'comp2', 'code': 'code2', 'name': 'name2', 'xtype': 1,'companyid':'comp1','inuse':True})
+    comp3 = add_seed(db.session, Company, {'id': 'comp3'},
+             {'id': 'comp3', 'code': 'code3', 'name': 'name3', 'xtype': 1,'companyid':'comp1','inuse':True})
+    return comp1,comp2,comp3
+
+def init_test_data():
+    '''
+    初始化测试资料
+    :return:返回各种类型的用户，方便后期测试
+    '''
+    init_companys()
+    # todo import AuthUser
+    from app.models import AuthUser
+    # 超级用户
+    dev =  add_seed(db.session,AuthUser,{'uid':'2222'},
+                      {'uid':'2222','uname':'dev','systemuser':True,'manageuser':False,'manageuserid':'','password':'2222'})
+    # 管理者用户
+    manage =  add_seed(db.session,AuthUser,{'uid':'2000'},
+                      {'uid':'2000','uname':'manage','systemuser':False,'manageuser':True,'manageuserid':'','password':'2000','companyid':'comp1'})
+    # 管理者下的普通用户
+    user_mng1 = add_seed(db.session,AuthUser,{'uid':'2001'},
+                      {'uid':'2001','uname':'user_mng1','systemuser':False,'manageuser':False,'manageuserid':'2000','password':'2001','companyid':'comp1'})
+    # 普通用户
+    user = add_seed(db.session,AuthUser,{'uid':'2002'},
+                      {'uid':'2002','uname':'user','systemuser':False,'manageuser':False,'manageuserid':'','password':'2002','companyid':'comp2'})
+    # current_app.logger.info('init_test_data success')
+    return dev, manage, user_mng1, user

gencode/gencode/template/tests/run.pys

@@ -0,0 +1,4 @@
+import tests
+
+if __name__ == '__main__':
+    tests.test()

gencode/gencode/template/tests/test_base.pys

@@ -0,0 +1,87 @@
+'''
+测试范例：
+1，form 提交
+        response = self.client.post(self.url_for('empuser_login'),
+                                    # header 定义
+                                    headers={'Content-Type': 'multipart/form-data', 'Accept': 'application/json'},
+                                    # data 必须是字典
+                                    data={'user': '13712363409', 'password': 'string','valid_code':'1234','companyid':'comp1'})
+2，post body，内容是 json
+        response = self.client.post(self.url_for('empuser_update'),
+                                    # header 定义
+                                    headers=self.get_headers(self.manage_user,{'Content-Type': 'application/json'}),
+                                    # data 必须是str，json.dumps
+                                    data=json.dumps({'uid': 'string', 'uname': 'string', 'companyid': 'string', 'emp_name': 'string', 'password': 'string'}))
+3，mock 使用 注意事项
+    from unittest import mock
+        。。。
+        # check_valid_code是 import的unit，不能设定函数的单元auth_utils，只需要指定引用单元empusermng
+        @mock.patch('app.api.v1_0.empusermng.check_valid_code')
+        def test_empuser_register_post(self,mock_check_valid_code):
+            # 驗證碼檢查找 mock
+            mock_check_valid_code.side_effect = [True,True,True,True]
+            response = self.client.post(self.url_for('empuser_register?valid_code=1234'),
+                                        headers=self.get_headers(self.manage_user,{'Content-Type': 'application/json'}),
+                                        data=json.dumps({ 'uname': '13712363409', 'companyid': 'comp1', 'emp_name': 'cxh', 'password': 'string'}))
+            # mock_check_valid_code.
+            。。。
+            mock_check_valid_code.assert_called_with('1234',0,None)
+        。。。
+        # 对应修饰器的顺序需颠倒
+        @mock.patch('app.api.v1_0.auth_companymng.get_companyids')
+        @mock.patch('app.api.v1_0.auth_companymng.get_service_companyid')
+        def test_auth_companys_servicename_get(self,mock_get_service_companyid,mock_get_companyids):
+            """
+            测试获取认证的公司
+            """
+            # 博览会服务
+            mock_companyid.side_effect = ['comp1']
+            # 返回多个时，list中必须要是一个list
+            mock_get_service_companyid.side_effect = [['comp1','comp2','comp3'],['comp1','comp2','comp3'],['comp1','comp3','comp4']]
+
+'''
+
+import unittest
+from app import create_app_swagger, db
+from unittest import mock
+from .init_test_data import init_test_data
+from flask.testing import FlaskClient
+
+class BasicTestCase(unittest.TestCase):
+
+    def setUp(self):
+        self.app = create_app_swagger('testing').app
+        self.app_context = self.app.app_context()
+        self.app_context.push()
+        db.create_all()
+        self.client = self.app.test_client()
+        self.server ='http://localhost:9999/{{swagger.name}}/v1.0'
+        # 超级用户，管理员用户，管理员创建的用户，普通用户
+        self.super_user,self.manage_user,self.user_mng,self.user =init_test_data()
+
+    def get_headers(self,user,ext_header=None):
+        '''
+        :param user:
+        :return:
+        '''
+        from app import rds
+        import json
+        rds.set('session:jwt_test',json.dumps({"uid": "%s"%user.uid, "uname": "%s"%user.uname,
+                                               "systemuser": user.systemuser, "manageuser": user.manageuser, "manageuserid": user.manageuserid}))
+        result ={'X-Consumer-Username': user.uname,
+                 'X-Consumer-Custom-Id': user.uid,
+                 'accept-language':'en-US;q=0.8,en;q=0.7'}
+        if ext_header is not None:
+            result.update(ext_header)
+        return result
+
+    def url_for(self,endpoint):
+        return '%s/%s'%(self.server,endpoint)
+
+    def tearDown(self):
+        db.session.remove()
+        db.drop_all()
+        self.app_context.pop()
+
+
+

gencode/gencode/template/tests/test_classmng.pys

@@ -0,0 +1,58 @@
+from tests.test_base import BasicTestCase
+import json
+# from unittest import mock
+
+class Test{{tag.name}}(BasicTestCase):
+{% for path in tag.paths %}
+    {% for act in path.actions %}
+    def test_{{path.name.replace('{', '').replace('}', '').replace('-', '_') + '_' + act.action}}(self):
+        """
+        # todo add testcacse description
+        """
+        {% if act.action == 'get' and act.default_resp.is_array %}
+        # todo fix url
+        response = self.client.get(self.url_for('{{path.name.replace('_{', '/{')}}{{get_params(act.params)}}'),
+                                   headers=self.get_headers(self.super_user))
+        self.assertEqual(response.status_code,200)
+        # todo add other assert code
+        exp_data = {{get_resp_js(act)}}
+        resp_data = response.json # json.loads(response.data.decode())
+        {% endif %}
+        {% if act.action == 'get' and not act.default_resp.is_array %}
+        # todo fix url
+        response = self.client.get(self.url_for('{{path.name.replace('_{', '/{')}}{{get_params(act.params)}}'),
+                                   headers=self.get_headers(self.manage_user))
+        self.assertEqual(response.status_code,200)
+        # todo add other assert code
+        exp_data = {{get_resp_js(act)}}
+        resp_data = response.json
+        {% endif %}
+        {% if act.action == 'post' %}
+        # todo fix url
+        response = self.client.post(self.url_for('{{path.name.replace('_{', '/{')}}{{get_params(act.params)}}'),
+                                    headers=self.get_headers(self.manage_user,{'Content-Type': 'application/json'}),
+                                    data=json.dumps({{get_resp_js(act)}}))
+        self.assertEqual(response.status_code, 201)
+        # todo add other assert code
+        exp_data = {{get_resp_js(act)}}
+        resp_data = response.json #json.loads(response.data.decode())
+        {% endif %}
+        {% if act.action == 'put' %}
+        # todo fix url
+        response = self.client.post(self.url_for('{{path.name.replace('_{', '/{')}}{{get_params(act.params)}}'),
+                                    headers=self.get_headers(self.manage_user,{'Content-Type': 'application/json'}),
+                                    data=json.dumps({{get_resp_js(act)}}))
+        self.assertEqual(response.status_code, 201)
+        # todo add other assert code
+        exp_data = {{get_resp_js(act)}}
+        resp_data = response.json # json.loads(response.data.decode())
+        {% endif %}
+        {% if act.action == 'delete' %}
+        # todo fix url
+        response = self.client.delete(self.url_for('{{path.name.replace('_{', '/{')}}{{get_params(act.params)}}'),
+                                    headers=self.get_headers(self.manage_user,{'Content-Type': 'application/json'}),
+                                    )
+        self.assertEqual(response.status_code, 204)
+        {% endif %}
+    {% endfor %}
+{% endfor %}

gencode/gencode/template/uwsgi.ini

@@ -0,0 +1,20 @@
+[uwsgi]
+socket = /tmp/uwsgi.sock
+chdir = /var/{{swagger.name}}
+module = uwsgi_run
+callable = app
+#生产环境下不能开启
+;py-autoreload = true
+master = true
+chmod-socket = 666
+plugin = python,http
+processes = 2
+threads = 2
+max-requests = 500
+;worker超过60秒没有反应，会被重启
+harakiri=60
+{% if 'kafka' in plugins %}
+; 在有cassandra，kafka等应用的专案中要开启lazy-apps和touch-chain-reload
+lazy-apps=true
+touch-chain-reload=true
+{% endif %}

gencode/gencode/template/uwsgi_run.pys

@@ -0,0 +1,44 @@
+from app import create_app_swagger
+
+def register_service(config_name,app):
+    from app.config import config
+    # 注册本服务到kong
+    config =  config[config_name]
+    auto_register2consul = config.CONSUL_AUTO_REGISTER
+    auto_register2kong = config.KONG_AUTO_REGISTER
+    web_port = config.PORT
+    # from mwsdk import AgentConf
+    service_host = f"{config.get_host_addr()}:{ web_port}"
+    app.logger.info('auto_register2kong:%s'%auto_register2kong)
+    if auto_register2kong:
+        from mwsdk import Kong
+        kong = Kong()
+        kong.add_upstream_target('{{swagger.name}}-server', service_host, 100, '/{{swagger.name}}/v1.0/health')
+        kong.reg_service('{{swagger.name}}/v1.0', '{{swagger.name}}-server', auth='jwt')
+        #todo add other_api
+        app.logger.info('注册kong成功')
+        #kong.reg_service('xxxx/v1.0/login_jwt', '{{swagger.name}}-server', auth='')
+        #kong.reg_service('xxxx/v1.0/token', '{{swagger.name}}-server', auth='key')
+        #kong.reg_service('xxxx/v1.0/logout_jwt', '{{swagger.name}}-server',auth='jwt')
+        #kong.reg_service('static', '{{swagger.name}}-server', auth='', kong_uris='/auth/static')
+    # 註冊到kong的服務到consul
+    app.logger.info('auto_register2consul:%s'%auto_register2consul)
+    if auto_register2consul:
+        check = {"id": "{{swagger.name}} api",
+                 "name": "{{swagger.name}} on port %s"%web_port,
+                 "http": "http://%s/{{swagger.name}}/v1.0/health"%service_host,
+                 "interval": "20s",
+                 "timeout": "10s",
+                 "DeregisterCriticalServiceAfter": "5m"
+                 }
+        from mwsdk import Kong, reg_service
+        kong = Kong()
+        reg_service('{{swagger.name}}',address=kong.ip, port=kong.port, tags=['kong','jwt'],
+                    check=check)
+        app.logger.info('注册consul成功')
+
+app = create_app_swagger('production')
+register_service('production',app.app)
+application = app.app
+
+