privacy.sexy/tests/unit/shared/Stubs/CategoryStub.ts

import { BaseEntity } from '@/infrastructure/Entity/BaseEntity';
import type { Category } from '@/domain/Executables/Category/Category';
import { RecommendationLevel } from '@/domain/Executables/Script/RecommendationLevel';
import type { Script } from '@/domain/Executables/Script/Script';
import { ScriptStub } from './ScriptStub';

export class CategoryStub extends BaseEntity<number> implements Category {
  public name = `category-with-id-${this.id}`;

  public readonly subCategories = new Array<Category>();

  public readonly scripts = new Array<Script>();

  public docs: readonly string[] = new Array<string>();

  private allScriptsRecursively: (readonly Script[]) | undefined;

  public constructor(id: number) {
    super(id);
  }

  public includes(script: Script): boolean {
    return this.getAllScriptsRecursively().some((s) => s.id === script.id);
  }

  public getAllScriptsRecursively(): readonly Script[] {
    if (this.allScriptsRecursively === undefined) {
      return [
        ...this.scripts,
        ...this.subCategories.flatMap((c) => c.getAllScriptsRecursively()),
      ];
    }
    return this.allScriptsRecursively;
  }

  public withScriptIds(...scriptIds: readonly string[]): this {
    return this.withScripts(
      ...scriptIds.map((id) => new ScriptStub(id)),
    );
  }

  public withScripts(...scripts: Script[]): this {
    for (const script of scripts) {
      this.withScript(script);
    }
    return this;
  }

  public withAllScriptIdsRecursively(...scriptIds: readonly string[]): this {
    return this.withAllScriptsRecursively(...scriptIds.map((id) => new ScriptStub(id)));
  }

  public withAllScriptsRecursively(...scripts: Script[]): this {
    this.allScriptsRecursively = [...scripts];
    return this;
  }

  public withMandatoryScripts(): this {
    return this
      .withScript(new ScriptStub(`[${CategoryStub.name}] script-1`).withLevel(RecommendationLevel.Standard))
      .withScript(new ScriptStub(`[${CategoryStub.name}] script-2`).withLevel(RecommendationLevel.Strict))
      .withScript(new ScriptStub(`[${CategoryStub.name}] script-3`).withLevel(undefined));
  }

  public withCategories(...categories: Category[]): this {
    for (const category of categories) {
      this.withCategory(category);
    }
    return this;
  }

  public withCategory(category: Category): this {
    this.subCategories.push(category);
    return this;
  }

  public withScript(script: Script): this {
    this.scripts.push(script);
    return this;
  }

  public withName(categoryName: string): this {
    this.name = categoryName;
    return this;
  }

  public withDocs(docs: readonly string[]): this {
    this.docs = docs;
    return this;
  }
}