import { EntityManager } from '@mikro-orm/core';
import { Group } from '../../../src/entities/assignments/group.entity';
import { Assignment } from '../../../src/entities/assignments/assignment.entity';
import { Student } from '../../../src/entities/users/student.entity';
import { getConditionalPathAssignment } from './assignments.testdata';
import { getTestleerling1 } from '../users/students.testdata';

export function makeTestGroups(em: EntityManager, students: Student[], assignments: Assignment[]): Group[] {
    /*
     * Group #1 for Assignment #1 in class 'id01'
     * => Assigned to do learning path 'id02'
     */
    group01 = em.create(Group, {
        assignment: assignments[0],
        groupNumber: 21001,
        members: students.slice(0, 2),
    });

    /*
     * Group #2 for Assignment #1 in class 'id01'
     * => Assigned to do learning path 'id02'
     */
    group02 = em.create(Group, {
        assignment: assignments[0],
        groupNumber: 21002,
        members: students.slice(2, 4),
    });

    /*
     * Group #3 for Assignment #1 in class 'id01'
     * => Assigned to do learning path 'id02'
     */
    group03 = em.create(Group, {
        assignment: assignments[0],
        groupNumber: 21003,
        members: students.slice(4, 6),
    });

    /*
     * Group #4 for Assignment #2 in class 'id02'
     * => Assigned to do learning path 'id01'
     */
    group04 = em.create(Group, {
        assignment: assignments[1],
        groupNumber: 21004,
        members: students.slice(3, 4),
    });

    /*
     * Group #5 for Assignment #4 in class 'id01'
     * => Assigned to do learning path 'id01'
     */
    group05 = em.create(Group, {
        assignment: assignments[3],
        groupNumber: 21001,
        members: students.slice(0, 2),
    });

    /**
     * Group 1 for the assignment of the testing learning path with conditions.
     */
    group1ConditionalLearningPath = em.create(Group, {
        assignment: getConditionalPathAssignment(),
        groupNumber: 1,
        members: [getTestleerling1()],
    });

    return [group01, group02, group03, group04, group05, group1ConditionalLearningPath];
}

let group01: Group;
let group02: Group;
let group03: Group;
let group04: Group;
let group05: Group;
let group1ConditionalLearningPath: Group;

export function getTestGroup01(): Group {
    return group01;
}

export function getTestGroup02(): Group {
    return group02;
}

export function getTestGroup03(): Group {
    return group03;
}

export function getTestGroup04(): Group {
    return group04;
}

export function getTestGroup05(): Group {
    return group05;
}

export function getGroup1ConditionalLearningPath(): Group {
    return group1ConditionalLearningPath;
}