BrowserExtension.php 4.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160
  1. <?php
  2. namespace Tests;
  3. use Facebook\WebDriver\Exception\TimeoutException;
  4. use Illuminate\Support\Arr;
  5. use Illuminate\Support\Str;
  6. use Laravel\Dusk\Browser;
  7. use PHPUnit\Framework\Assert as PHPUnit;
  8. trait BrowserExtension
  9. {
  10. public function extendBrowser()
  11. {
  12. $this->extendBrowserWhenTextAvailable();
  13. $this->extendBrowserWhenElementAvailable();
  14. $this->extendBrowserWait();
  15. $this->extendBrowserAssertHasInput();
  16. $this->extendBrowserAssertHidden();
  17. }
  18. private function extendBrowserAssertHidden()
  19. {
  20. Browser::macro('assertHidden', function ($selector) {
  21. $fullSelector = $this->resolver->format($selector);
  22. PHPUnit::assertTrue(
  23. $this->resolver->findOrFail($selector)->isDisplayed(),
  24. "Element [{$fullSelector}] is visible."
  25. );
  26. return $this;
  27. });
  28. }
  29. private function extendBrowserWait()
  30. {
  31. $self = $this;
  32. Browser::macro('wait', function ($seconds, \Closure $callback = null) use ($self) {
  33. $delayBrowser = $self->makeDelayBrowser($this);
  34. try {
  35. $this->waitUsing($seconds, 200, function () {});
  36. } catch (TimeoutException $e) {
  37. $callback && $callback();
  38. $delayBrowser();
  39. }
  40. return $delayBrowser;
  41. });
  42. }
  43. private function extendBrowserAssertHasInput()
  44. {
  45. Browser::macro('hasInput', function ($field) {
  46. /* @var \Facebook\WebDriver\Remote\RemoteWebElement $element */
  47. $this->resolver->resolveForTyping($field);
  48. return $this;
  49. });
  50. }
  51. private function extendBrowserWhenElementAvailable()
  52. {
  53. $self = $this;
  54. Browser::macro('whenElementAvailable', function ($selector, $callbackOrSeconds = null, $seconds = null) use ($self) {
  55. /* @var Browser $this */
  56. $callback = null;
  57. if (is_callable($callbackOrSeconds)) {
  58. $callback = $callbackOrSeconds;
  59. } elseif (is_int($callbackOrSeconds)) {
  60. $seconds = $callbackOrSeconds;
  61. }
  62. $delayBrowser = $self->makeDelayBrowser($this);
  63. $this->waitFor($selector, $seconds)->with($selector, function ($value) use ($callback, $delayBrowser) {
  64. $callback && $callback($value);
  65. return $delayBrowser();
  66. });
  67. return $delayBrowser;
  68. });
  69. }
  70. private function extendBrowserWhenTextAvailable()
  71. {
  72. $self = $this;
  73. Browser::macro('whenTextAvailable', function ($text, $callbackOrSeconds = null, $seconds = null) use ($self) {
  74. $callback = null;
  75. if (is_callable($callbackOrSeconds)) {
  76. $callback = $callbackOrSeconds;
  77. } elseif (is_int($callbackOrSeconds)) {
  78. $seconds = $callbackOrSeconds;
  79. }
  80. $delayBrowser = $self->makeDelayBrowser($this);
  81. $text = Arr::wrap($text);
  82. $message = $this->formatTimeOutMessage('Waited %s seconds for text', implode("', '", $text));
  83. $this->waitUsing($seconds, 100, function () use ($text, $callback, $delayBrowser) {
  84. $results = Str::contains($this->resolver->findOrFail('')->getText(), $text);
  85. if ($results) {
  86. $callback && $callback($this);
  87. $delayBrowser();
  88. }
  89. return $results;
  90. }, $message);
  91. return $delayBrowser;
  92. });
  93. }
  94. public function makeDelayBrowser($browser)
  95. {
  96. return new class($browser) {
  97. protected $browser;
  98. protected $callbacks = [];
  99. public function __construct(Browser $browser)
  100. {
  101. $this->browser = $browser;
  102. }
  103. public function __call($method, $arguments = [])
  104. {
  105. $this->callbacks[] = [
  106. 'method' => $method,
  107. 'arguments' => $arguments,
  108. ];
  109. return $this;
  110. }
  111. public function __invoke()
  112. {
  113. $browser = $this->browser;
  114. foreach ($this->callbacks as $value) {
  115. $method = $value['method'];
  116. $browser = $browser->{$method}(...$value['arguments']);
  117. }
  118. return $browser;
  119. }
  120. };
  121. }
  122. }