@@ -43,24 +43,24 @@ public void testByImageLocator() {
4343 Assert .assertTrue (docLabels .size () > 1 , "List of elements should be possible to find by image" );
4444 Assert .assertEquals (docLabels .size (), childLabels .size (), "Should be possible to find child elements by image with the same count" );
4545
46- ILabel documentByTag = AqualityServices .getElementFactory ().getLabel (By .tagName ("body " ), "document by tag" );
46+ ILabel imgByTag = AqualityServices .getElementFactory ().getLabel (By .tagName ("img " ), "img by tag" );
4747 float fullThreshold = 1 ;
4848 AtomicReference <ILabel > documentByImageContainer = new AtomicReference <>();
4949 boolean isFound = AqualityServices .getConditionalWait ().waitFor (browser -> {
50- ILabel documentByImage = AqualityServices .getElementFactory ().getLabel (new ByImage (documentByTag .getElement ().getScreenshotAs (OutputType .BYTES )).setThreshold (fullThreshold ),
51- "body screen" );
52- documentByImageContainer .set (documentByImage );
53- boolean isDisplayed = documentByImage .state ().isDisplayed ();
50+ ILabel image = AqualityServices .getElementFactory ().getLabel (new ByImage (imgByTag .getElement ().getScreenshotAs (OutputType .BYTES )).setThreshold (fullThreshold ),
51+ "img screen" );
52+ documentByImageContainer .set (image );
53+ boolean isDisplayed = image .state ().isDisplayed ();
5454 if (!isDisplayed ) {
5555 browser .navigate ().refresh ();
5656 }
5757 return isDisplayed ;
5858 });
59- ILabel documentByImage = documentByImageContainer .get ();
59+ ILabel elementByImage = documentByImageContainer .get ();
6060
61- Assert .assertTrue (isFound , "Should be possible to find element by document screenshot" );
62- Assert .assertEquals (((ByImage ) documentByImage .getLocator ()).getThreshold (), fullThreshold , "Should be possible to get ByImage threshold" );
63- Assert .assertEquals (documentByImage .getElement ().getTagName (), "body " , "Correct element must be found" );
61+ Assert .assertTrue (isFound , "Should be possible to find element by element screenshot" );
62+ Assert .assertEquals (((ByImage ) elementByImage .getLocator ()).getThreshold (), fullThreshold , "Should be possible to get ByImage threshold" );
63+ Assert .assertEquals (elementByImage .getElement ().getTagName (), "img " , "Correct element must be found" );
6464 }
6565
6666 @ Test
0 commit comments