camelCaseDave/xrm-mock

export class FormContextMock implements Xrm.FormContext {
  public data: Xrm.Data;
  public ui: Xrm.Ui;

  constructor(data: Xrm.Data, ui: Xrm.Ui) {
    this.data = data;
    this.ui = ui;
  }

  public getAttribute<T extends Xrm.Attributes.Attribute>(attributeNameOrIndex: number | string): T;
  public getAttribute<T extends Xrm.Attributes.Attribute>(delegateFunction?: Xrm.Collection.MatchingDelegate<T>): T[];

  public getAttribute<T extends Xrm.Attributes.Attribute>(param?: number | string |
      Xrm.Collection.MatchingDelegate<T>): T | Xrm.Attributes.Attribute[] {
    if (!arguments.length) {
      return this.data.entity.attributes.get();
    } else if (param && typeof param === "string") {
      return this.data.entity.attributes.get<T>(param);
    } else if (typeof param === "number") {
      return this.data.entity.attributes.get(param) as T;
    } else if (typeof param === "function") {
      return this.data.entity.attributes.get(param as Xrm.Collection.MatchingDelegate<T>) as T[];
    }
    throw new Error(`Collection.Get called with unknown parameter type: ${typeof param}`);
  }

  public getControl<T extends Xrm.Controls.Control>(controlNameOrIndex: string | number): T;
  public getControl<T extends Xrm.Controls.Control>(delegateFunction?: Xrm.Collection.MatchingDelegate<T>): T[];

  public getControl<T extends Xrm.Controls.Control>(param?: number | string |
    Xrm.Collection.MatchingDelegate<T>): T | T[] {
    if (!arguments.length || param === undefined || param === null) {
      return this.ui.controls.get() as T[];
    } else if (typeof param === "string") {
      return this.ui.controls.get<T>(param);
    } else if (typeof param === "number") {
      return this.ui.controls.get(param) as T;
    } else if (typeof param === "function") {
      return this.ui.controls.get(param as Xrm.Collection.MatchingDelegate<T>) as T[];
    }
    throw new Error(`Collection.Get called with unknown parameter type: ${typeof param}`);
  }
}