pytest-db 0.1.2__tar.gz

pytest_db-0.1.2/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2019 Hanzhichao
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

pytest_db-0.1.2/MANIFEST.in

@@ -0,0 +1,8 @@
+include *.in
+include *.ini
+include *.rst
+include *.txt
+include LICENSE
+
+global-exclude __pycache__ *.py[cod]
+global-exclude *.so *.dylib

pytest_db-0.1.2/PKG-INFO

@@ -0,0 +1,29 @@
+Metadata-Version: 2.4
+Name: pytest-db
+Version: 0.1.2
+Summary: Session scope fixture "db" for mysql query or change
+Home-page: https://github.com/hanzhichao/pytest-db
+Author: Han Zhichao
+Author-email: superhin@126.com
+License: MIT license
+Keywords: pytest,py.test,db,mysql
+Classifier: Framework :: Pytest
+Classifier: Programming Language :: Python
+Classifier: Topic :: Software Development :: Testing
+Classifier: Programming Language :: Python :: 3.7
+License-File: LICENSE
+Requires-Dist: pytest
+Requires-Dist: pytest-runner
+Requires-Dist: pymysql
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: home-page
+Dynamic: keywords
+Dynamic: license
+Dynamic: license-file
+Dynamic: requires-dist
+Dynamic: summary
+
+Session scope fixture "db" for mysql query or change

pytest_db-0.1.2/README.md

@@ -0,0 +1,81 @@
+# pytest-db
+
+![Languate - Python](https://img.shields.io/badge/language-python-blue.svg)
+![PyPI - License](https://img.shields.io/pypi/l/pytest-db)
+![PyPI](https://img.shields.io/pypi/v/pytest-db)
+![PyPI - Downloads](https://img.shields.io/pypi/dm/pytest-db)
+
+Pytest操作数据库
+
+---
+
+### 如何使用
+
+1. 安装 `pytest-db`
+
+使用pip从github安装
+
+```
+pip install pytest-db
+```
+
+2. 使用方法
+   在环境变量中添加
+
+```sh
+export DB_URI=mysql://root:password@localhost:3306/test
+```
+
+或在pytest.ini中配置
+
+```ini
+[pytest]
+db_uri = mysql://root:password@localhost:3306/test
+```
+
+或命令行传入
+
+```sh
+$ pytest --db-uri=mysql://root:password@localhost:3306/test
+```
+
+或使用Fixture配置
+
+```python
+import pytest
+
+
+@pytest.fixture
+def db_uri():
+    return 'mysql://root:password@localhost:3306/test'
+```
+
+3. 使用fixture函数: db
+
+```python
+def test_a(db):
+    print(db.query('select id from user limit 1;')
+```
+
+使用pytest -s 运行，查看结果
+
+```
+...
+[{'id': 123321336}]
+...
+```
+
+> 游标使用pymysql.cursors.DictCursor，结果使用fetchone,返回字典格式的结果
+
+4. db对象支持的方法
+
+- db.query(sql): 执行查询sql，返回一条结果
+- db.query_all(sql): 执行查询sql，返回所有结果
+- db.execute(sql): 执行修改sql
+
+---
+
+- Email: <a href="mailto:superhin@126.com?Subject=Pytest%20Email" target="_blank">`superhin@126.com`</a>
+- Blog: <a href="https://www.cnblogs.com/superhin/" target="_blank">`博客园 韩志超`</a>
+- 简书: <a href="https://www.jianshu.com/u/0115903ded22" target="_blank">`简书 韩志超`</a>
+

pytest_db-0.1.2/pytest_db/mysql_client.py

@@ -0,0 +1,65 @@
+import logging
+from typing import List
+from urllib.parse import urlparse, parse_qs
+
+import pymysql
+
+DEFAULT_MYSQL_PORT = 3306
+DEFAULT_MYSQL_CHARSET = 'utf8'
+
+LOGGER = logging.getLogger('pytest-db')
+
+def parse_db_uri(db_uri):
+    parsed_uri = urlparse(db_uri)
+    assert parsed_uri.scheme == 'mysql', 'Only support mysql'
+
+    user = parsed_uri.username
+    password = parsed_uri.password
+    host = parsed_uri.hostname
+    port = parsed_uri.port or DEFAULT_MYSQL_PORT
+    db = parsed_uri.path.lstrip('/')
+    # db = parsed_uri.fragment
+
+    charset = DEFAULT_MYSQL_CHARSET
+    if parsed_uri.query is not None:
+        charset = parse_qs(parsed_uri.query).get('charset', [DEFAULT_MYSQL_CHARSET])[0]
+
+    db_conf = dict(host=host, port=int(port), user=user, password=password, db=db, charset=charset)
+    return db_conf
+
+
+class MySQLClient(object):
+    def __init__(self, db_uri: str):
+        db_conf = parse_db_uri(db_uri)
+        db_conf.setdefault('charset', 'utf8')
+        LOGGER.debug(f'Connect db: {db_conf["host"]}:{db_conf["port"]}')
+        self.conn = pymysql.connect(**db_conf, autocommit=True)
+        self.cursor = self.conn.cursor(pymysql.cursors.DictCursor)
+
+    def query(self, sql: str) -> dict:
+        LOGGER.debug(f'Query sql: {sql}')
+        self.cursor.execute(sql)
+        result = self.cursor.fetchone()
+        LOGGER.debug(f"Query result: {result}")
+        return result
+
+    def query_all(self, sql: str) -> List[dict]:
+        LOGGER.debug(f'Query sql: {sql}')
+        self.cursor.execute(sql)
+        result = self.cursor.fetchall()
+        LOGGER.debug(f"Query result num: {len(result)}")
+        return result
+
+    def execute(self, sql):
+        LOGGER.debug(f'Execute sql: {sql}')
+        try:
+            self.cursor.execute(sql)
+            self.conn.commit()
+        except Exception as e:
+            LOGGER.exception(e)
+            self.conn.rollback()
+
+    def close(self):
+        LOGGER.debug(f'Close db cursor and connection')
+        self.cursor.close()
+        self.conn.close()

pytest_db-0.1.2/pytest_db/plugin.py

@@ -0,0 +1,31 @@
+import os
+import re
+import pytest
+
+from pytest_db.mysql_client import MySQLClient, LOGGER
+
+
+def pytest_addoption(parser):
+    parser.addoption('--db-uri', help='DB URI such like mysql://root:password@localhost:3306/test')
+    parser.addini('db_uri', help='DB URI such like mysql://root:password@localhost:3306/test')
+
+
+@pytest.fixture(scope='session')
+def db_uri(request):
+    """Get or overwrite the db uri config"""
+    return request.config.getoption('--db-uri') or request.config.getini('db_uri') or os.getenv('DB_URI')
+
+
+@pytest.fixture(scope='session')
+def db(db_uri):
+    """MySQL db client"""
+    if not db_uri:
+        pytest.skip('Not set db_uri')
+    try:
+        db = MySQLClient(db_uri)
+    except Exception as ex:
+        LOGGER.exception(ex)
+        pytest.skip(str(ex))
+    else:
+        yield db
+        db.close()

pytest_db-0.1.2/pytest_db.egg-info/PKG-INFO

@@ -0,0 +1,29 @@
+Metadata-Version: 2.4
+Name: pytest-db
+Version: 0.1.2
+Summary: Session scope fixture "db" for mysql query or change
+Home-page: https://github.com/hanzhichao/pytest-db
+Author: Han Zhichao
+Author-email: superhin@126.com
+License: MIT license
+Keywords: pytest,py.test,db,mysql
+Classifier: Framework :: Pytest
+Classifier: Programming Language :: Python
+Classifier: Topic :: Software Development :: Testing
+Classifier: Programming Language :: Python :: 3.7
+License-File: LICENSE
+Requires-Dist: pytest
+Requires-Dist: pytest-runner
+Requires-Dist: pymysql
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: home-page
+Dynamic: keywords
+Dynamic: license
+Dynamic: license-file
+Dynamic: requires-dist
+Dynamic: summary
+
+Session scope fixture "db" for mysql query or change

pytest_db-0.1.2/pytest_db.egg-info/SOURCES.txt

@@ -0,0 +1,16 @@
+LICENSE
+MANIFEST.in
+README.md
+requirements.txt
+setup.py
+pytest_db/__init__.py
+pytest_db/mysql_client.py
+pytest_db/plugin.py
+pytest_db.egg-info/PKG-INFO
+pytest_db.egg-info/SOURCES.txt
+pytest_db.egg-info/dependency_links.txt
+pytest_db.egg-info/entry_points.txt
+pytest_db.egg-info/requires.txt
+pytest_db.egg-info/top_level.txt
+pytest_db.egg-info/zip-safe
+tests/test_pytest_db.py

pytest_db-0.1.2/pytest_db.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

pytest_db-0.1.2/pytest_db.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[pytest11]
+pytest-db = pytest_db.plugin

pytest_db-0.1.2/pytest_db.egg-info/requires.txt

@@ -0,0 +1,3 @@
+pytest
+pytest-runner
+pymysql

pytest_db-0.1.2/pytest_db.egg-info/top_level.txt

@@ -0,0 +1 @@
+pytest_db

pytest_db-0.1.2/pytest_db.egg-info/zip-safe

@@ -0,0 +1 @@
+

pytest_db-0.1.2/requirements.txt

@@ -0,0 +1,2 @@
+pytest
+pymysql

pytest_db-0.1.2/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

pytest_db-0.1.2/setup.py

@@ -0,0 +1,42 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+from setuptools import setup, find_packages
+
+version = '0.1.2'
+
+setup_requirements = ['pytest-runner',]
+
+setup(
+    author="Han Zhichao",
+    author_email='superhin@126.com',
+    description='Session scope fixture "db" for mysql query or change',
+    long_description='Session scope fixture "db" for mysql query or change',
+    classifiers=[
+        'Framework :: Pytest',
+        'Programming Language :: Python',
+        'Topic :: Software Development :: Testing',
+        'Programming Language :: Python :: 3.7',
+    ],
+    license="MIT license",
+    include_package_data=True,
+    keywords=[
+        'pytest', 'py.test', 'db', 'mysql'
+    ],
+    name='pytest-db',
+    packages=find_packages(include=['pytest_db']),
+    setup_requires=setup_requirements,
+    url='https://github.com/hanzhichao/pytest-db',
+    version=version,
+    zip_safe=True,
+    install_requires=[
+        'pytest',
+        'pytest-runner',
+        'pymysql'
+    ],
+    entry_points={
+        'pytest11': [
+            'pytest-db = pytest_db.plugin',
+        ]
+    }
+)

pytest_db-0.1.2/tests/test_pytest_db.py

@@ -0,0 +1,17 @@
+import pytest
+
+from pytest_db.mysql_client import parse_db_uri
+
+@pytest.fixture(scope='session')
+def db_uri():
+    return 'mysql://root:passw0rd@localhost:3306/test?charset=utf8'
+
+def test_parse_db_uri(db_uri):
+    db_conf = parse_db_uri(db_uri)
+    assert db_conf == {'host': 'localhost',
+                       'port': 3306,
+                       'user': 'root',
+                       'password': 'passw0rd',
+                       'db': 'test',
+                       'charset': 'utf8'}
+