[Behat] Change 'iShouldSeeActionBarButton' exception for assertion

ezsystems · Apr 20, 2016 · 572f6ae · 572f6ae
1 parent e79ba26
commit 572f6ae
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/Features/Context/SubContext/CommonActions.php b/Features/Context/SubContext/CommonActions.php
@@ -321,9 +321,7 @@ private function goToContentWithPath($path)
     public function iShouldSeeActionBarButton($buttonText)
     {
         $element = $this->getElementByText($buttonText, '.ez-actionbar-container .ez-action', '.action-label');
-        if (!$element) {
-            throw new \Exception("Action bar button '$buttonText' not found");
-        }
+        EzAssertion::assertSingleElement($buttonText, $element, null, 'Action bar button');
     }
 
     /**
@@ -334,9 +332,7 @@ public function iShouldSeeActionBarButton($buttonText)
     public function iShouldNotSeeActionBarButton($buttonText)
     {
         $element = $this->getElementByText($buttonText, '.ez-actionbar-container .ez-action', '.action-label');
-        if ($element) {
-            throw new \Exception("Action bar button '$buttonText' was found");
-        }
+        EzAssertion::assertElementNotFound($buttonText, $element, null, 'Action bar button');
     }
 
     /**