MockActionSheet.swift 1.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172
  1. import Sheeeeeeeeet
  2. class MockActionSheet: ActionSheet {
  3. var dismissInvokeCount = 0
  4. var handleTapInvokeCount = 0
  5. var handleTapInvokeItems = [ActionSheetItem]()
  6. var prepareForPresentationInvokeCount = 0
  7. var refreshInvokeCount = 0
  8. var refreshButtonsInvokeCount = 0
  9. var refreshItemsInvokeCount = 0
  10. var refreshHeaderInvokeCount = 0
  11. var reloadDataInvokeCount = 0
  12. var setupInvokeCount = 0
  13. var setupItemsInvokeCount = 0
  14. var setupItemsInvokeItems = [[ActionSheetItem]]()
  15. private var _presentingViewController: UIViewController?
  16. override var presentingViewController: UIViewController? {
  17. get { return _presentingViewController }
  18. set { _presentingViewController = newValue }
  19. }
  20. override func dismiss(completion: @escaping () -> ()) {
  21. super.dismiss { completion() }
  22. dismissInvokeCount += 1
  23. }
  24. override func handleTap(on item: ActionSheetItem) {
  25. super.handleTap(on: item)
  26. handleTapInvokeCount += 1
  27. handleTapInvokeItems.append(item)
  28. }
  29. override func refresh() {
  30. super.refresh()
  31. refreshInvokeCount += 1
  32. }
  33. override func refreshButtons() {
  34. super.refreshButtons()
  35. refreshButtonsInvokeCount += 1
  36. }
  37. override func refreshItems() {
  38. super.refreshItems()
  39. refreshItemsInvokeCount += 1
  40. }
  41. override func refreshHeader() {
  42. super.refreshHeader()
  43. refreshHeaderInvokeCount += 1
  44. }
  45. override func reloadData() {
  46. super.reloadData()
  47. reloadDataInvokeCount += 1
  48. }
  49. override func setup() {
  50. super.setup()
  51. setupInvokeCount += 1
  52. }
  53. override func setup(items: [ActionSheetItem]) {
  54. super.setup(items: items)
  55. setupItemsInvokeCount += 1
  56. setupItemsInvokeItems.append(items)
  57. }
  58. }