Aam-Digital/ndb-core

import { ChildrenService } from "./children.service";
import { EntityMapperService } from "../../core/entity/entity-mapper/entity-mapper.service";
import { ChildSchoolRelation } from "./model/childSchoolRelation";
import { TestBed, waitForAsync } from "@angular/core/testing";
import moment from "moment";
import { Database } from "../../core/database/database";
import { Note } from "../notes/model/note";
import { genders } from "./model/genders";
import { DatabaseTestingModule } from "../../utils/database-testing.module";
import { sortByAttribute } from "../../utils/utils";
import { expectEntitiesToMatch } from "../../utils/expect-entity-data.spec";
import { DateWithAge } from "../../core/basic-datatypes/date-with-age/dateWithAge";
import { AttendanceModule } from "../attendance/attendance.module";
import { EntitySchemaService } from "../../core/entity/schema/entity-schema.service";
import { createEntityOfType } from "../../core/demo-data/create-entity-of-type";
import { Entity } from "../../core/entity/model/entity";

describe("ChildrenService", () => {
  let service: ChildrenService;
  let entityMapper: EntityMapperService;

  beforeEach(waitForAsync(() => {
    TestBed.configureTestingModule({
      imports: [DatabaseTestingModule, AttendanceModule],
      providers: [EntitySchemaService],
    });

    entityMapper = TestBed.inject(EntityMapperService);
    generateChildEntities().forEach((c) => entityMapper.save(c));
    generateSchoolEntities().forEach((s) => entityMapper.save(s));
    generateChildSchoolRelationEntities().forEach((cs) =>
      entityMapper.save(cs),
    );

    service = TestBed.inject<ChildrenService>(ChildrenService);
  }));

  afterEach(() => TestBed.inject(Database).destroy());

  it("should be created", () => {
    expect(service).toBeTruthy();
  });

  it("should list newly saved children", async () => {
    const childrenBefore = await service.getChildren();
    const child = createEntityOfType("Child", "10");
    await entityMapper.save(child);
    const childrenAfter = await service.getChildren();

    expect(childrenBefore.some((x) => x.getId() === child.getId())).toBeFalse();
    expect(childrenAfter.some((x) => x.getId() === child.getId())).toBeTrue();
    expect(childrenAfter).toHaveSize(childrenBefore.length + 1);
  });

  it("should find a newly saved child", async () => {
    const child = createEntityOfType("Child", "10");
    try {
      await service.getChild(child.getId());
      fail("Child should not be found");
    } catch (err) {
      expect(err).toBeDefined();
    }

    await entityMapper.save(child);
    const childAfter = await service.getChild(child.getId());
    expect(childAfter.getId()).toEqual(child.getId());
  });

  // TODO: test getAttendances

  it("calculates days since last note for children", async () => {
    const allChildren = await entityMapper.loadType("Child");

    const c0 = allChildren[0].getId();
    await entityMapper.save(
      Note.create(moment().subtract(5, "days").toDate(), "n0-1", [c0]),
    );
    await entityMapper.save(
      Note.create(moment().subtract(8, "days").toDate(), "n0-2", [c0]),
    );

    const c1 = allChildren[1].getId();
    // no notes

    const recentNotesMap = await service.getDaysSinceLastNoteOfEachEntity();

    expect(recentNotesMap).toHaveSize(allChildren.length);
    expect(recentNotesMap.get(c0)).toBe(5);
    expect(recentNotesMap.get(c1)).toBePositiveInfinity();
  });

  it("calculates days since last note as infinity if above cut-off period for better performance", async () => {
    const allChildren = await entityMapper.loadType("Child");

    const c0 = allChildren[0].getId();
    await entityMapper.save(
      Note.create(moment().subtract(50, "days").toDate(), "n0-1", [c0]),
    );

    const recentNotesMap = await service.getDaysSinceLastNoteOfEachEntity(
      "Child",
      49,
    );

    expect(recentNotesMap.get(c0)).toBePositiveInfinity();
  });

  it("should calculate days since last note for other entity types", async () => {
    const schools = await entityMapper.loadType("School");
    const s1 = schools[0];
    const s2 = schools[1];
    const n1 = new Note();
    n1.date = moment().subtract(10, "days").toDate();
    n1.schools.push(s1.getId());
    n1.schools.push(s2.getId());
    const n2 = new Note();
    n2.date = moment().subtract(2, "days").toDate();
    n2.schools.push(s1.getId());
    await entityMapper.saveAll([n1, n2]);

    const recentNotesMap =
      await service.getDaysSinceLastNoteOfEachEntity("School");

    expect(recentNotesMap.get(s1.getId())).toBe(2);
    expect(recentNotesMap.get(s2.getId())).toBe(10);
  });

  it("should load a single child and add school info", async () => {
    // no active relation
    const child2 = await service.getChild("Child:2");
    expect(child2["schoolClass"]).toBeUndefined();
    expect(child2["schoolId"]).toBeEmpty();

    // one active relation
    let child1 = await service.getChild("Child:1");
    expect(child1["schoolClass"]).toBe("2");
    expect(child1["schoolId"]).toEqual(["School:1"]);

    // multiple active relations
    const newRelation = new ChildSchoolRelation();
    newRelation.childId = child1.getId();
    newRelation.start = new Date();
    newRelation["schoolId"] = "School:2";
    newRelation["schoolClass"] = "3";
    await entityMapper.save(newRelation);
    child1 = await service.getChild(child1.getId());
    expect(child1["schoolClass"]).toBe("3");
    expect(child1["schoolId"]).toEqual(["School:2", "School:1"]);

    // multiple active, no start date on one
    const noStartDate = new ChildSchoolRelation();
    noStartDate.childId = child1.getId();
    noStartDate["schoolId"] = "School:2";
    noStartDate["schoolClass"] = "4";
    await entityMapper.save(noStartDate);
    child1 = await service.getChild(child1.getId());
    expect(child1["schoolClass"]).toBe("4");
    expect(child1["schoolId"]).toEqual(["School:2", "School:2", "School:1"]);
  });

  it("should load all children with school info", async () => {
    const children = await service.getChildren();
    const child1 = children.find((child) => child.getId() === "Child:1");
    expect(child1["schoolClass"]).toBe("2");
    expect(child1["schoolId"]).toEqual(["School:1"]);
    const child2 = children.find((child) => child.getId() === "Child:2");
    expect(child2["schoolClass"]).toBeUndefined();
    expect(child2["schoolId"]).toBeEmpty();
    const child3 = children.find((child) => child.getId() === "Child:3");
    expect(child3["schoolClass"]).toBe("2");
    expect(child3["schoolId"]).toEqual(["School:1"]);
  });

  it("should get the relations for a child in sorted order", async () => {
    const relations = await service.queryRelations("Child:3");

    expect(relations).toHaveSize(2);
    expect(relations[0].start.getTime()).toBeGreaterThanOrEqual(
      relations[1].start.getTime(),
    );
  });

  it("should get all relations for a school", async () => {
    const relations = await service.queryRelations("School:1");

    expect(relations).toHaveSize(2);
    const relation1 = relations.find(
      (relation) => relation.getId() === "ChildSchoolRelation:1",
    );
    expect(relation1.childId).toBe("Child:1");
    const relation2 = relations.find(
      (relation) => relation.getId() === "ChildSchoolRelation:4",
    );
    expect(relation2.childId).toBe("Child:3");
  });

  it("should get a active relation which starts today", async () => {
    const todayRelation = new ChildSchoolRelation("today");
    todayRelation["schoolId"] = "School:3";
    todayRelation.start = new Date();
    await entityMapper.save(todayRelation);
    const relations = await service.queryActiveRelationsOf("School:3");
    expectEntitiesToMatch(relations, [todayRelation]);
  });

  it("should on default only return active relations", async () => {
    const allRelations = await entityMapper.loadType(ChildSchoolRelation);
    const activeRelations = allRelations
      .filter((rel) => rel.isActive && rel.childId === "Child:3")
      .sort(sortByAttribute("start", "desc"));

    const result = await service.queryActiveRelationsOf("Child:3");
    expect(result).toEqual(activeRelations);
  });

  it("should return active relations for a given date", async () => {
    let relations = await service.queryActiveRelationsOf(
      "School:1",
      new Date("2010-01-01"),
    );
    expect(relations).toHaveSize(1);

    relations = await service.queryActiveRelationsOf(
      "School:1",
      new Date("2016-10-01"),
    );
    expect(relations).toHaveSize(2);
  });

  it("should return related notes", async () => {
    const c1 = createEntityOfType("Child", "c1");
    const c2 = createEntityOfType("Child", "c2");
    const s1 = createEntityOfType("School", "s1");
    const s2 = createEntityOfType("School", "s2");
    const n1 = new Note("n1");
    n1.addChild(c1);
    n1.addChild(c2);
    n1.addSchool(s1);
    const n2 = new Note("n2");
    n2.addChild(c1);
    const n3 = new Note("n3");
    n3.addSchool(s2);
    await entityMapper.saveAll([n1, n2, n3]);

    let res = await service.getNotesRelatedTo(c1.getId());
    expect(res).toEqual([n1, n2]);

    res = await service.getNotesRelatedTo(s1.getId());
    expect(res).toEqual([n1]);

    res = await service.getNotesRelatedTo(s2.getId());
    expect(res).toEqual([n3]);
  });

  it("should include related notes through children and schools links (legacy)", async () => {
    const c1 = createEntityOfType("Child", "c1");
    const s1 = createEntityOfType("School", "s1");
    const n1 = new Note("n1");
    n1.children.push(c1.getId());
    n1.relatedEntities.push(c1.getId());
    n1.schools.push(s1.getId());
    await entityMapper.saveAll([n1]);

    let res = await service.getNotesRelatedTo(c1.getId());
    expect(res).toEqual([n1]);

    res = await service.getNotesRelatedTo(s1.getId());
    expect(res).toEqual([n1]);
  });

  it("should return the correct notes in a timespan", async () => {
    const n1 = Note.create(moment("2023-01-01").toDate());
    const n2 = Note.create(moment("2023-01-02").toDate());
    const n3 = Note.create(moment("2023-01-03").toDate());
    const n4 = Note.create(moment("2023-01-03").toDate());
    const n5 = Note.create(moment("2023-01-04").toDate());
    await entityMapper.saveAll([n1, n2, n3, n4, n5]);

    const res = await service.getNotesInTimespan(
      moment("2023-01-02"),
      moment("2023-01-03"),
    );

    expect(res).toEqual(jasmine.arrayWithExactContents([n2, n3, n4]));
  });
});

function generateChildEntities(): Entity[] {
  const data = [];

  const a1 = createEntityOfType("Child", "1");
  a1.name = "Arjun A.";
  a1.projectNumber = "1";
  a1["religion"] = "Hindu";
  a1.gender = genders[1];
  a1.dateOfBirth = new DateWithAge("2000-03-13");
  a1["motherTongue"] = "Hindi";
  a1.center = { id: "delhi", label: "Delhi" };
  data.push(a1);

  const a2 = createEntityOfType("Child", "2");
  a2.name = "Bandana B.";
  a2.projectNumber = "2";
  a2["religion"] = "Hindu";
  a2.gender = genders[2];
  a2.dateOfBirth = new DateWithAge("2001-01-01");
  a2["motherTongue"] = "Bengali";
  a2.center = { id: "kolkata", label: "Kolkata" };
  data.push(a2);

  const a3 = createEntityOfType("Child", "3");
  a3.name = "Chandan C.";
  a3.projectNumber = "3";
  a3["religion"] = "Hindu";
  a3.gender = genders[1];
  a3.dateOfBirth = new DateWithAge("2002-07-29");
  a3["motherTongue"] = "Hindi";
  a3.center = { id: "kolkata", label: "Kolkata" };
  data.push(a3);

  return data;
}

function generateSchoolEntities(): Entity[] {
  const data = [];

  const s1 = createEntityOfType("School", "1");
  s1.name = "People's Primary";
  data.push(s1);

  const s2 = createEntityOfType("School", "2");
  s2.name = "Hope High School";
  data.push(s2);

  return data;
}

function generateChildSchoolRelationEntities(): ChildSchoolRelation[] {
  const data: ChildSchoolRelation[] = [];
  const rel1: ChildSchoolRelation = new ChildSchoolRelation("1");
  rel1.childId = "Child:1";
  rel1["schoolId"] = "School:1";
  rel1.start = new Date("2016-10-01");
  rel1["schoolClass"] = "2";
  data.push(rel1);

  const rel4: ChildSchoolRelation = new ChildSchoolRelation("2");
  rel4.childId = "Child:3";
  rel4["schoolId"] = "School:2";
  rel4.start = new Date("2001-01-01");
  rel4.end = new Date("2002-01-01");
  rel4["schoolClass"] = "1";
  data.push(rel4);

  const rel2: ChildSchoolRelation = new ChildSchoolRelation("3");
  rel2.childId = "Child:2";
  rel2["schoolId"] = "School:2";
  rel2.start = new Date("2018-05-07");
  rel2.end = new Date("2018-05-09");
  rel2["schoolClass"] = "3";
  data.push(rel2);

  const rel3: ChildSchoolRelation = new ChildSchoolRelation("4");
  rel3.childId = "Child:3";
  rel3["schoolId"] = "School:1";
  rel3.start = new Date("2010-01-01");
  rel3["schoolClass"] = "2";
  data.push(rel3);

  return data;
}