mirror of
https://github.com/veekun/pokedex.git
synced 2024-08-20 18:16:34 +00:00
Fix the i18n test; now uses multilang's session subclasses.
This commit is contained in:
parent
50b3adf949
commit
d90db39d43
1 changed files with 5 additions and 9 deletions
|
@ -7,7 +7,8 @@ from sqlalchemy.orm.session import Session
|
||||||
from sqlalchemy.ext.declarative import declarative_base
|
from sqlalchemy.ext.declarative import declarative_base
|
||||||
|
|
||||||
from pokedex.db import tables, markdown
|
from pokedex.db import tables, markdown
|
||||||
from pokedex.db.multilang import create_translation_table
|
from pokedex.db.multilang import MultilangScopedSession, MultilangSession, \
|
||||||
|
create_translation_table
|
||||||
|
|
||||||
def test_variable_names():
|
def test_variable_names():
|
||||||
"""We want pokedex.db.tables to export tables using the class name"""
|
"""We want pokedex.db.tables to export tables using the class name"""
|
||||||
|
@ -63,16 +64,10 @@ def test_i18n_table_creation():
|
||||||
name = Column(String(100)),
|
name = Column(String(100)),
|
||||||
)
|
)
|
||||||
|
|
||||||
class FauxSession(Session):
|
|
||||||
def execute(self, clause, params=None, *args, **kwargs):
|
|
||||||
if not params:
|
|
||||||
params = {}
|
|
||||||
params.setdefault('_default_language', 'en')
|
|
||||||
return super(FauxSession, self).execute(clause, params, *args, **kwargs)
|
|
||||||
|
|
||||||
# OK, create all the tables and gimme a session
|
# OK, create all the tables and gimme a session
|
||||||
Base.metadata.create_all()
|
Base.metadata.create_all()
|
||||||
sess = sessionmaker(engine, class_=FauxSession)()
|
sm = sessionmaker(class_=MultilangSession)
|
||||||
|
sess = MultilangScopedSession(sm)
|
||||||
|
|
||||||
# Create some languages and foos to bind together
|
# Create some languages and foos to bind together
|
||||||
lang_en = Language(identifier='en')
|
lang_en = Language(identifier='en')
|
||||||
|
@ -87,6 +82,7 @@ def test_i18n_table_creation():
|
||||||
|
|
||||||
# Commit so the above get primary keys filled in
|
# Commit so the above get primary keys filled in
|
||||||
sess.commit()
|
sess.commit()
|
||||||
|
sess.default_language = lang_en.id
|
||||||
|
|
||||||
# Give our foo some names, as directly as possible
|
# Give our foo some names, as directly as possible
|
||||||
foo_text = FooText()
|
foo_text = FooText()
|
||||||
|
|
Loading…
Reference in a new issue