import { UniDriver } from '@wix/wix-ui-test-utils/unidriver';
import {
  EmptyStateUniDriver,
  TextButtonUniDriver,
} from '@wix/design-system/dist/testkit/unidriver';
import { baseUniDriverFactory } from '../../unidriver';

export default (base: UniDriver, body: UniDriver) => {
  return {
    ...baseUniDriverFactory(base),
    ...EmptyStateUniDriver(base.$('[data-hook="page-error-state"'), body),

    getRetryButton: () =>
      TextButtonUniDriver(base.$('[data-hook="retry-button"]'), body),
  };
};