import { SelectorType } from './options';

export function takeScreenshot(t: TestController, element: SelectorType, filePath: string): TestControllerPromise<any> {
  return element
    ? t.takeElementScreenshot(element, filePath)
    : t.takeScreenshot(filePath);
}