kraken/backend/tests/utils/train.py

"""testing subjects utils"""
from sqlalchemy.orm import Session

from db.repository import trains as db_rep_train
from schemas.trains import TrainCreate
from utils.utils import fake
from utils.utils import get_random_position


def generate_fake_train_data():
    return {
        "position": "左手",
        "rank": fake.train_rank(),
        "trial_num": fake.random_digit_not_null(),
        "start_time": "2022-11-03 00:00",
        "end_time": "2022-11-04 00:00"
    }


def create_test_train2db(db: Session,
                      subject,
                      position=None,
                      rank=fake.train_rank(),
                      trial_num=fake.random_digit_not_null(),
                      start_time="2022-11-03 00:00",
                      end_time="2022-11-04 00:00",
                      device_parm=None) -> TrainCreate:
    if position is None:
        position = get_random_position(subject)
    train = TrainCreate(position=position,
                        rank=rank,
                        trial_num=trial_num,
                        start_time=start_time,
                        end_time=end_time,
                        device_parm=device_parm,
                        owner_id=subject.id)
    train = db_rep_train.create_train(train, db)
    return train