selene-backend/api/public/tests/features/steps/wolfram_alpha.py

31 lines
1010 B
Python

from http import HTTPStatus
from behave import when, then
from hamcrest import assert_that
@when('a question is sent')
def send_question(context):
login = context.device_login
access_token = login['accessToken']
context.wolfram_response = context.client.get(
'/v1/wa?input=what+is+the+capital+of+Brazil',
headers=dict(Authorization='Bearer {token}'.format(token=access_token))
)
@when('a question is sent to the wolfram alpha spoken endpoint')
def send_question(context):
login = context.device_login
access_token = login['accessToken']
context.wolfram_response = context.client.get(
'/v1/wolframAlphaSpoken?i=how+tall+was+abraham+lincoln&geolocation=51.50853%2C-0.12574&units=Metric',
headers=dict(Authorization='Bearer {token}'.format(token=access_token))
)
@then('the wolfram alpha endpoint should return 200')
def validate_response(context):
response = context.wolfram_response
assert_that(response.status_code, HTTPStatus.OK)