function expectCustomButton() {
  mockCustomButtonHandler.handleEvent(
      new ArgumentMatcher((arg) => arg.type == CUSTOM_EVENT));
}
Sets up the mock event handler to expect a custom button event.
@suppress {missingProperties} suppression added to enable type checking