forked from 3wordchant/capsul-flask
@ -4,6 +4,7 @@ from nanoid import generate
|
||||
from flask_testing import TestCase
|
||||
|
||||
from capsulflask import create_app
|
||||
from capsulflask.db import get_model
|
||||
|
||||
class BaseTestCase(TestCase):
|
||||
def create_app(self):
|
||||
@ -11,6 +12,7 @@ class BaseTestCase(TestCase):
|
||||
os.environ['POSTGRES_CONNECTION_PARAMETERS'] = "host=localhost port=5432 user=postgres password=dev dbname=capsulflask_test"
|
||||
os.environ['TESTING'] = '1'
|
||||
os.environ['SPOKE_MODEL'] = 'mock'
|
||||
os.environ['HUB_MODEL'] = 'mock'
|
||||
return create_app()
|
||||
|
||||
def setUp(self):
|
||||
@ -20,6 +22,7 @@ class BaseTestCase(TestCase):
|
||||
pass
|
||||
|
||||
def _login(self, user_email):
|
||||
get_model().login(user_email)
|
||||
with self.client.session_transaction() as session:
|
||||
session['account'] = user_email
|
||||
session['csrf-token'] = generate()
|
||||
|
Reference in New Issue
Block a user