mocks.ts

import type {ExecutionContext} from '../api_types';
import type {FetchResponse} from '../api_types';
import type {Sync} from '../api_types';
import sinon from 'sinon';
import {v4} from 'uuid';

export interface MockExecutionContext extends ExecutionContext {
  fetcher: {
    fetch: sinon.SinonStub;
  };
  temporaryBlobStorage: {
    storeUrl: sinon.SinonStub;
    storeBlob: sinon.SinonStub;
  };
}

export interface MockSyncExecutionContext extends MockExecutionContext {
  readonly sync: Sync;
}

export function newMockSyncExecutionContext(overrides?: Partial<MockSyncExecutionContext>): MockSyncExecutionContext {
  return {...newMockExecutionContext(), sync: {}, ...overrides};
}

export function newMockExecutionContext(overrides?: Partial<MockExecutionContext>): MockExecutionContext {
  return {
    invocationLocation: {
      protocolAndHost: 'https://coda.io',
    },
    timezone: 'America/Los_Angeles',
    invocationToken: v4(),
    fetcher: {
      fetch: sinon.stub(),
    },
    temporaryBlobStorage: {
      storeUrl: sinon.stub(),
      storeBlob: sinon.stub(),
    },
    ...overrides,
  };
}

export function newJsonFetchResponse<T>(
  body: T,
  status: number = 200,
  headers?: {[header: string]: string | string[] | undefined},
): FetchResponse<T> {
  const allHeaders = {'Content-Type': 'application/json', ...headers};
  return {status, body, headers: allHeaders};
}