Aam-Digital/ndb-core

import { fakeAsync, TestBed, tick, waitForAsync } from "@angular/core/testing";

import { AbilityService } from "./ability.service";
import { BehaviorSubject, Subject } from "rxjs";
import { Note } from "../../../child-dev-project/notes/model/note";
import { EntityMapperService } from "../../entity/entity-mapper/entity-mapper.service";
import { PermissionEnforcerService } from "../permission-enforcer/permission-enforcer.service";
import { defaultInteractionTypes } from "../../config/default-config/default-interaction-types";
import { EntityAbility } from "./entity-ability";
import { DatabaseRule, DatabaseRules } from "../permission-types";
import { Config } from "../../config/config";
import { Logging } from "../../logging/logging.service";
import { UpdatedEntity } from "../../entity/model/entity-update";
import { mockEntityMapper } from "../../entity/entity-mapper/mock-entity-mapper-service";
import { CoreTestingModule } from "../../../utils/core-testing.module";
import { DefaultDatatype } from "../../entity/default-datatype/default.datatype";
import { EventAttendanceMapDatatype } from "../../../child-dev-project/attendance/model/event-attendance.datatype";
import { SessionSubject } from "../../session/auth/session-info";
import { TEST_USER } from "../../user/demo-user-generator.service";
import { TestEntity } from "../../../utils/test-utils/TestEntity";
import { CurrentUserSubject } from "../../session/current-user-subject";

describe("AbilityService", () => {
  let service: AbilityService;
  let ability: EntityAbility;
  let entityUpdates: Subject<UpdatedEntity<Config<DatabaseRules>>>;
  let entityMapper: jasmine.SpyObj<EntityMapperService>;
  const rules: DatabaseRules = {
    user_app: [
      { subject: TestEntity.ENTITY_TYPE, action: "read" },
      { subject: Note.ENTITY_TYPE, action: "manage", inverted: true },
    ],
    admin_app: [{ subject: "all", action: "manage" }],
  };

  beforeEach(waitForAsync(() => {
    entityUpdates = new Subject();
    entityMapper = mockEntityMapper() as any;
    spyOn(entityMapper, "receiveUpdates").and.returnValue(entityUpdates);
    spyOn(entityMapper, "load").and.callThrough();

    TestBed.configureTestingModule({
      imports: [CoreTestingModule],
      providers: [
        AbilityService,
        EntityAbility,
        {
          provide: SessionSubject,
          useValue: new BehaviorSubject({
            name: TEST_USER,
            roles: ["user_app"],
            entityId: TestEntity.createPrefixedId(
              TestEntity.ENTITY_TYPE,
              TEST_USER,
            ),
          }),
        },
        {
          provide: CurrentUserSubject,
          useValue: new BehaviorSubject(new TestEntity(TEST_USER)),
        },
        {
          provide: DefaultDatatype,
          useClass: EventAttendanceMapDatatype,
          multi: true,
        },
        {
          provide: PermissionEnforcerService,
          useValue: jasmine.createSpyObj(["enforcePermissionsOnLocalData"]),
        },
        { provide: EntityMapperService, useValue: entityMapper },
      ],
    });
    service = TestBed.inject(AbilityService);
    ability = TestBed.inject(EntityAbility);
  }));

  afterEach(() => {
    entityUpdates.complete();
  });

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

  it("should fetch the rules object from the database", () => {
    service.initializeRules();
    expect(entityMapper.load).toHaveBeenCalledWith(
      Config,
      Config.PERMISSION_KEY,
    );
  });

  it("should give all permissions if no rules object can be fetched but not until checking for rules object", fakeAsync(() => {
    entityMapper.load.and.rejectWith("no initial config");

    // no permissions until rules object has been attempted to fetch initially
    expect(ability.rules).toEqual([]);

    service.initializeRules();
    tick(); // initial attempt to load rules resolved now

    // Request failed, sync not started - offline without cached rules object
    expect(ability.rules).toEqual([{ action: "manage", subject: "all" }]);
  }));

  it("should update the rules when a change is published", fakeAsync(() => {
    entityMapper.load.and.rejectWith("no initial config");
    service.initializeRules();
    tick();

    expect(entityMapper.load).toHaveBeenCalled();
    // Default rule
    expect(ability.rules).toEqual([{ action: "manage", subject: "all" }]);

    entityUpdates.next({
      entity: new Config(Config.PERMISSION_KEY, rules),
      type: "update",
    });
    tick();

    expect(ability.rules).toEqual(rules["user_app"]);
  }));

  it("should update the ability with the received rules for the logged in user", fakeAsync(() => {
    service.initializeRules();
    tick();

    spyOn(ability, "update");
    entityUpdates.next({
      entity: new Config(Config.PERMISSION_KEY, rules),
      type: "update",
    });
    tick();

    expect(ability.update).toHaveBeenCalledWith(rules.user_app);
  }));

  it("should update the ability with rules for all roles the logged in user has", fakeAsync(() => {
    service.initializeRules();
    tick();

    spyOn(ability, "update");
    TestBed.inject(SessionSubject).next({
      name: "testAdmin",
      id: "1",
      roles: ["user_app", "admin_app"],
    });

    entityUpdates.next({
      entity: new Config(Config.PERMISSION_KEY, rules),
      type: "update",
    });
    tick();

    expect(ability.update).toHaveBeenCalledWith(
      rules.user_app.concat(rules.admin_app),
    );
  }));

  it("should create an ability that correctly uses the defined rules", fakeAsync(() => {
    service.initializeRules();
    tick();

    entityUpdates.next({
      entity: new Config(Config.PERMISSION_KEY, rules),
      type: "update",
    });
    tick();

    expect(ability.can("read", TestEntity)).toBeTrue();
    expect(ability.can("create", TestEntity)).toBeFalse();
    expect(ability.can("manage", TestEntity)).toBeFalse();
    expect(ability.can("read", new TestEntity())).toBeTrue();
    expect(ability.can("create", new TestEntity())).toBeFalse();
    expect(ability.can("manage", Note)).toBeFalse();
    expect(ability.can("manage", new Note())).toBeFalse();
    expect(ability.can("create", new Note())).toBeFalse();

    TestBed.inject(SessionSubject).next({
      name: "testAdmin",
      id: "1",
      roles: ["user_app", "admin_app"],
    });

    const updatedConfig = new Config(Config.PERMISSION_KEY, rules);
    updatedConfig._rev = "update";
    entityUpdates.next({ entity: updatedConfig, type: "update" });
    tick();

    expect(ability.can("manage", TestEntity)).toBeTrue();
    expect(ability.can("manage", new TestEntity())).toBeTrue();
    expect(ability.can("manage", Note)).toBeTrue();
    expect(ability.can("manage", new Note())).toBeTrue();
  }));

  it("should throw an error when checking permissions on a object that is not a Entity", () => {
    entityUpdates.next({
      entity: new Config(Config.PERMISSION_KEY, rules),
      type: "update",
    });

    class TestClass {}

    expect(() => ability.can("read", new TestClass() as any)).toThrowError();
  });

  it("should call the ability enforcer after updating the rules", fakeAsync(() => {
    service.initializeRules();
    tick();

    entityUpdates.next({
      entity: new Config(Config.PERMISSION_KEY, rules),
      type: "update",
    });
    tick();

    expect(
      TestBed.inject(PermissionEnforcerService).enforcePermissionsOnLocalData,
    ).toHaveBeenCalled();
  }));

  it("should allow to access user properties in the rules", fakeAsync(() => {
    service.initializeRules();
    tick();

    function testPlaceholderCondition(placeholder: string) {
      const config = new Config<DatabaseRules>(Config.PERMISSION_KEY, {
        user_app: [
          {
            subject: "TestEntity",
            action: "manage",
            conditions: { _id: placeholder },
          },
        ],
      });
      entityUpdates.next({ entity: config, type: "update" });
      tick();

      const userEntity = new TestEntity(TEST_USER);
      expect(ability.can("manage", userEntity)).toBeTrue();
      const anotherUser = new TestEntity();
      expect(ability.cannot("manage", anotherUser)).toBeTrue();
    }

    testPlaceholderCondition("${user.name}");
    testPlaceholderCondition("${user.entityId}");
  }));

  it("should allow to check conditions with complex data types", fakeAsync(() => {
    service.initializeRules();
    tick();

    const classInteraction = defaultInteractionTypes.find(
      (type) => type.id === "SCHOOL_CLASS",
    );
    const config = new Config<DatabaseRules>(Config.PERMISSION_KEY, {
      user_app: [
        {
          subject: "Note",
          action: "read",
          conditions: { category: classInteraction.id },
        },
      ],
    });
    entityUpdates.next({ entity: config, type: "update" });
    tick();

    const note = new Note();
    expect(ability.can("read", note)).toBeFalse();
    note.category = classInteraction;
    tick();
    expect(ability.can("read", note)).toBeTrue();
  }));

  it("should log a warning if no rules are found for a user", fakeAsync(() => {
    service.initializeRules();
    tick();

    TestBed.inject(SessionSubject).next({
      name: "new-user",
      id: "1",
      roles: ["invalid_role"],
    });
    const warnSpy = spyOn(Logging, "warn");
    entityUpdates.next({
      entity: new Config(Config.PERMISSION_KEY, rules),
      type: "update",
    });
    tick();

    expect(warnSpy).toHaveBeenCalled();
  }));

  it("should prepend default rules to all users", fakeAsync(() => {
    service.initializeRules();
    tick();
    const defaultRules: DatabaseRule[] = [
      { subject: "Config", action: "read" },
      { subject: "ProgressDashboardConfig", action: "manage" },
    ];
    const config = new Config<DatabaseRules>(
      Config.PERMISSION_KEY,
      Object.assign({ default: defaultRules } as DatabaseRules, rules),
    );

    entityUpdates.next({ entity: config, type: "update" });
    tick();

    expect(ability.rules).toEqual(defaultRules.concat(...rules.user_app));

    TestBed.inject(SessionSubject).next({
      name: "admin",
      id: "1",
      roles: ["user_app", "admin_app"],
    });

    config._rev = "update";
    entityUpdates.next({ entity: config, type: "update" });
    tick();
    expect(ability.rules).toEqual(
      defaultRules.concat(...rules.user_app, ...rules.admin_app),
    );
  }));

  it("should allow everything if permission doc has been deleted", fakeAsync(() => {
    service.initializeRules();
    tick();
    entityUpdates.next({
      type: "new",
      entity: new Config<DatabaseRules>(Config.PERMISSION_KEY, rules),
    });
    tick();

    expect(ability.rules).toEqual(rules["user_app"]);

    entityUpdates.next({
      type: "remove",
      entity: new Config<DatabaseRules>(Config.PERMISSION_KEY),
    });
    tick();

    expect(ability.rules).toEqual([{ subject: "all", action: "manage" }]);
  }));
});