@cloud-carbon-footprint/gcp/src/__tests__/helpers.ts

The core logic to get cloud usage data and estimate energy and carbon emissions from Google Cloud Platform.

/*
 * © 2021 Thoughtworks, Inc.
 */
export function setupSpy(
  spyMethod: any,
  targetFunction: string,
  returnValue: any,
): void {
  const targetFunctionSpy = jest.spyOn(spyMethod, targetFunction)
  ;(targetFunctionSpy as jest.Mock).mockResolvedValue(returnValue)
}

export function setupSpyWithMultipleValues(
  spyMethod: any,
  targetFunction: string,
  defaultReturnValue: any,
  secondaryReturnValue: any,
): void {
  const targetFunctionSpy = jest.spyOn(spyMethod, targetFunction)
  ;(targetFunctionSpy as jest.Mock).mockResolvedValueOnce(defaultReturnValue)
  ;(targetFunctionSpy as jest.Mock).mockResolvedValueOnce(secondaryReturnValue)
}

export function setupSpyWithRejectedValue(
  spyMethod: any,
  targetFunction: string,
  defaultRejectedValue: any,
): void {
  const targetFunctionSpy = jest.spyOn(spyMethod, targetFunction)
  ;(targetFunctionSpy as jest.Mock).mockRejectedValueOnce(defaultRejectedValue)
}