MycroftAI
/
selene-backend
mirror of https://github.com/MycroftAI/selene-backend.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fixed tests still using the old connection pool

pull/148/head
Chris Veilleux 2019-05-22 13:22:45 -05:00
parent 00be679ad4
commit 7c96bde026
2 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
api/public/tests/features/steps/device_skills.py
View File

@ -6,7 +6,7 @@ from hamcrest import assert_that, equal_to, not_none, is_not
from selene.api.etag import ETagManager, device_skill_etag_key
from selene.data.skill import SkillSettingRepository
from selene.util.db import get_db_connection
from selene.util.db import connect_to_db
skill = {
'skill_gid': 'wolfram-alpha|19.02',
@ -106,8 +106,8 @@ def update_skill(context):
}]
response = json.loads(context.upload_device_response.data)
skill_id = response['uuid']
with get_db_connection(context.client_config['DB_CONNECTION_POOL']) as db:
SkillSettingRepository(db).update_device_skill_settings(skill_id, update_settings)
db = connect_to_db(context.client_config['DB_CONNECTION_CONFIG'])
SkillSettingRepository(db).update_device_skill_settings(skill_id, update_settings)
@when('the skill settings is fetched')

8
api/public/tests/features/steps/get_device_subscription.py
View File

@ -7,7 +7,7 @@ from behave import when, then
from hamcrest import assert_that, has_entry, equal_to
from selene.data.account import AccountRepository, AccountMembership
from selene.util.db import get_db_connection
from selene.util.db import connect_to_db
@when('the subscription endpoint is called')
@ -42,9 +42,9 @@ def get_device_subscription(context):
login = context.device_login
device_id = login['uuid']
access_token = login['accessToken']
headers=dict(Authorization='Bearer {token}'.format(token=access_token))
with get_db_connection(context.client_config['DB_CONNECTION_POOL']) as db:
AccountRepository(db).add_membership(context.account.id, membership)
headers = dict(Authorization='Bearer {token}'.format(token=access_token))
db = connect_to_db(context.client_config['DB_CONNECTION_CONFIG'])
AccountRepository(db).add_membership(context.account.id, membership)
context.subscription_response = context.client.get(
'/v1/device/{uuid}/subscription'.format(uuid=device_id),
headers=headers