FitTrackee/fittrackee/tests/workouts/utils.py
2022-06-11 11:33:08 +02:00

41 lines
1.1 KiB
Python

import json
from io import BytesIO
from typing import Optional, Tuple
from uuid import uuid4
from flask import Flask
from fittrackee.workouts.utils.short_id import encode_uuid
def get_random_short_id() -> str:
return encode_uuid(uuid4())
def post_a_workout(
app: Flask, gpx_file: str, notes: Optional[str] = None
) -> Tuple[str, str]:
client = app.test_client()
resp_login = client.post(
'/api/auth/login',
data=json.dumps(dict(email='test@test.com', password='12345678')),
content_type='application/json',
)
token = json.loads(resp_login.data.decode())['auth_token']
workout_data = '{"sport_id": 1'
if notes is not None:
workout_data += f', "notes": "{notes}"'
workout_data += '}'
response = client.post(
'/api/workouts',
data=dict(
file=(BytesIO(str.encode(gpx_file)), 'example.gpx'),
data=workout_data,
),
headers=dict(
content_type='multipart/form-data', Authorization=f'Bearer {token}'
),
)
data = json.loads(response.data.decode())
return token, data['data']['workouts'][0]['id']