From 7c96bde026c74b541ce6fc60dc2683c654fc18bd Mon Sep 17 00:00:00 2001 From: Chris Veilleux Date: Wed, 22 May 2019 13:22:45 -0500 Subject: [PATCH] fixed tests still using the old connection pool --- api/public/tests/features/steps/device_skills.py | 6 +++--- .../tests/features/steps/get_device_subscription.py | 8 ++++---- 2 files changed, 7 insertions(+), 7 deletions(-) diff --git a/api/public/tests/features/steps/device_skills.py b/api/public/tests/features/steps/device_skills.py index 6a0c013e..63846c87 100644 --- a/api/public/tests/features/steps/device_skills.py +++ b/api/public/tests/features/steps/device_skills.py @@ -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') diff --git a/api/public/tests/features/steps/get_device_subscription.py b/api/public/tests/features/steps/get_device_subscription.py index 18c36f42..6543976f 100644 --- a/api/public/tests/features/steps/get_device_subscription.py +++ b/api/public/tests/features/steps/get_device_subscription.py @@ -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