pytest autouse fixtures

Author: DovSnier

case/argparse/conftest.py

@@ -24,19 +24,3 @@ def _temp_file():
     print('the current temp path is {}'.format(temp))
     yield temp
     os.remove(temp)
-
-
-@pytest.fixture()
-def cleandir():
-    ''' cleandir '''
-    newpath = tempfile.mkdtemp()
-    os.chdir(newpath)
-    print('\nthe current temp dir is {}'.format(newpath))
-
-
-# @pytest.mark.xfail(raises=ZeroDivisionError)
-# def test_zero_division():
-#     ''' test zero division'''
-#     1/0
-# with pytest.raises(ZeroDivisionError):
-#     1 / 0

case/pytest/__init__.py

@@ -0,0 +1 @@
+# -*- coding:utf-8 -*-

case/pytest/conftest.py

@@ -0,0 +1,42 @@
+# -*- coding:utf-8 -*-
+
+import os
+import shutil
+import tempfile
+import pytest
+
+
+@pytest.fixture(scope="session")
+def _temp_dir():
+    ''' _temp_dir '''
+    temp = tempfile.mkdtemp(prefix='argprase-')
+    print('\n')
+    print('the current temp dir is {}'.format(temp))
+    yield temp
+    shutil.rmtree(temp)
+
+
+@pytest.fixture(scope="session")
+def _temp_file():
+    ''' _temp_file '''
+    fd, temp = tempfile.mkstemp(prefix='argprase-', suffix='.tmp')
+    # print('\n')
+    print('the current temp path is {}'.format(temp))
+    yield temp
+    os.remove(temp)
+
+
+@pytest.fixture()
+def cleandir():
+    ''' cleandir '''
+    newpath = tempfile.mkdtemp()
+    os.chdir(newpath)
+    print('\nthe current temp dir is {}'.format(newpath))
+
+
+# @pytest.mark.xfail(raises=ZeroDivisionError)
+# def test_zero_division():
+#     ''' test zero division'''
+#     1/0
+# with pytest.raises(ZeroDivisionError):
+#     1 / 0

case/pytest/test_db_transact.py

@@ -0,0 +1,54 @@
+# -*- coding:utf-8 -*-
+
+import os
+import pytest
+
+
+class DB(object):
+    ''' db '''
+    def __init__(self):
+        self.intransaction = []
+
+    def begin(self, name):
+        ''' begin '''
+        print('\nDB::begin: {}'.format(name))
+        self.intransaction.append(name)
+
+    def rollback(self):
+        ''' rollback '''
+        item = self.intransaction.pop()
+        print('\nDB::rollback: {}'.format(item))
+
+
+@pytest.fixture(scope="module")
+def db():
+    ''' db '''
+    return DB()
+
+
+class TestClass(object):
+    ''' TestClass '''
+    @pytest.fixture(autouse=True)
+    def transact(self, request, db):
+        ''' transact '''
+        db.begin(request.function.__name__)
+        yield
+        db.rollback()
+
+    def test_method1(self, db):
+        ''' test_method1 '''
+        print('\nTestClass::test_method1: {}'.format(db))
+        assert db.intransaction == ["test_method1"]
+
+    def test_method2(self, db):
+        ''' test_method2 '''
+        print('\nTestClass::test_method2: {}'.format(db))
+        assert db.intransaction == ["test_method2"]
+
+
+if __name__ == "__main__":
+    ''' the main point '''
+    retcode = pytest.main(args=['-v', '-s', os.path.abspath(__file__)])
+    # retcode = pytest.main(args=['--fixtures', os.path.abspath(__file__)])
+    # retcode = pytest.main(args=['--collect-only', os.path.abspath(__file__)])
+    print(retcode)