Personal/realms-wiki
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

Make temp dir handling in tests a bit more pythonic

Browse source 
(and let them work on Windows)
This commit is contained in:
Chase Sterling 2016-07-05 20:29:36 -04:00
parent 2e6db650f2
commit fefbfcd503
1 changed files with 7 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
realms/lib/test.py
View file

@ -1,17 +1,20 @@
import os
import shutil
import tempfile
from flask.ext.testing import TestCase from flask.ext.testing import TestCase
from realms.lib.util import random_string from realms.lib.util import random_string
from realms import create_app from realms import create_app
from subprocess import call
class BaseTest(TestCase): class BaseTest(TestCase):
def create_app(self): def create_app(self):
self.tempdir = tempfile.mkdtemp()
app = create_app() app = create_app()
app.config['TESTING'] = True app.config['TESTING'] = True
app.config['PRESERVE_CONTEXT_ON_EXCEPTION'] = False app.config['PRESERVE_CONTEXT_ON_EXCEPTION'] = False
app.config['WIKI_PATH'] = '/tmp/%s' % random_string(12) app.config['WIKI_PATH'] = os.path.join(self.tempdir, random_string(12))
app.config['DB_URI'] = 'sqlite:////tmp/%s.db' % random_string(12) app.config['DB_URI'] = 'sqlite:///%s/%s.db' % (self.tempdir, random_string(12))
app.config.update(self.configure()) app.config.update(self.configure())
return app return app
@ -19,5 +22,4 @@ class BaseTest(TestCase):
return {} return {}
def tearDown(self): def tearDown(self):
call(['rm', '-rf', self.app.config['WIKI_PATH']]) shutil.rmtree(self.tempdir)
call(['rm', '-f', self.app.config['DB_URI'][10:]])