FitTrackee/fittrackee/tests/workouts/utils.py

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']