Merge pull request #327 from TikhomirovSergey/#320-fix

@SButterfly

You can:
- switch to your java_client fork using git

- > git checkout -b your_new_branch_name master

- > git pull https://github.com/User_Name/java-client.git target_branch

But now you can clone master

Author: TikhomirovSergey

src/main/java/io/appium/java_client/pagefactory/Widget.java

@@ -61,4 +61,8 @@ public WebDriver getWrappedDriver() {
     public WebElement getWrappedElement() {
         return (WebElement) element;
     }
+
+    public Widget getSelfReference() {
+        return this;
+    }
 }

src/test/java/io/appium/java_client/pagefactory_tests/widgets/AndroidOverrideWidgetTest.java

@@ -3,6 +3,8 @@
 import io.appium.java_client.android.AndroidDriver;
 import io.appium.java_client.pagefactory.AppiumFieldDecorator;
 import io.appium.java_client.pagefactory.TimeOutDuration;
+import io.appium.java_client.pagefactory_tests.widgets.android.annotated.AnnotatedAndroidMovie;
+import io.appium.java_client.pagefactory_tests.widgets.android.simple.AndroidMovie;
 import io.appium.java_client.remote.MobileCapabilityType;
 import io.appium.java_client.service.local.AppiumDriverLocalService;
 import org.apache.commons.lang3.StringUtils;
@@ -68,6 +70,7 @@ public void checkACommonWidget() {
         movie.goToReview();
 
         rottenTomatoes.checkSimpleReview();
+        assertTrue(movie.getSelfReference().getClass().equals(AndroidMovie.class));
     }
 
     @Override
@@ -81,6 +84,7 @@ public void checkAnAnnotatedWidget() {
         movie.goToReview();
 
         rottenTomatoes.checkAnnotatedReview();
+        assertTrue(movie.getSelfReference().getClass().equals(AnnotatedAndroidMovie.class));
     }
 
 
@@ -95,6 +99,7 @@ public void checkAnExtendedWidget() {
         movie.goToReview();
 
         rottenTomatoes.checkExtendedReview();
+        assertTrue(movie.getSelfReference().getClass().equals(AnnotatedAndroidMovie.class));
     }
 
     @Override

src/test/java/io/appium/java_client/pagefactory_tests/widgets/HtmlOverrideWidgetTest.java

@@ -2,6 +2,8 @@
 
 import io.appium.java_client.pagefactory.AppiumFieldDecorator;
 import io.appium.java_client.pagefactory.TimeOutDuration;
+import io.appium.java_client.pagefactory_tests.widgets.html.annotated.AnnotatedHtmlMovie;
+import io.appium.java_client.pagefactory_tests.widgets.html.simple.HtmlMovie;
 import org.apache.commons.lang3.StringUtils;
 import org.junit.AfterClass;
 import org.junit.Before;
@@ -45,7 +47,7 @@ public static void beforeClass() throws Exception {
     @Before
     public void setUp() throws Exception {
         if (driver != null)
-            driver.get("file:///" + new File("src/test/java/io/appium/java_client/RottenTomatoesSnapshot.html").getAbsolutePath());
+            driver.get( new File("src/test/java/io/appium/java_client/RottenTomatoesSnapshot.html").toURI().toString());
     }
 
     @AfterClass
@@ -65,6 +67,7 @@ public void checkACommonWidget() {
         movie.goToReview();
 
         rottenTomatoes.checkSimpleReview();
+        assertTrue(movie.getSelfReference().getClass().equals(HtmlMovie.class));
     }
 
     @Override
@@ -78,6 +81,7 @@ public void checkAnAnnotatedWidget() {
         movie.goToReview();
 
         rottenTomatoes.checkAnnotatedReview();
+        assertTrue(movie.getSelfReference().getClass().equals(AnnotatedHtmlMovie.class));
     }
 
 
@@ -92,6 +96,7 @@ public void checkAnExtendedWidget() {
         movie.goToReview();
 
         rottenTomatoes.checkExtendedReview();
+        assertTrue(movie.getSelfReference().getClass().equals(AnnotatedHtmlMovie.class));
     }
 
     @Override

src/test/java/io/appium/java_client/pagefactory_tests/widgets/IOSOverrideWidgetTest.java

@@ -3,6 +3,8 @@
 import io.appium.java_client.ios.IOSDriver;
 import io.appium.java_client.pagefactory.AppiumFieldDecorator;
 import io.appium.java_client.pagefactory.TimeOutDuration;
+import io.appium.java_client.pagefactory_tests.widgets.ios.annotated.AnnotatedIOSMovie;
+import io.appium.java_client.pagefactory_tests.widgets.ios.simple.IOSMovie;
 import io.appium.java_client.remote.MobileCapabilityType;
 import io.appium.java_client.service.local.AppiumDriverLocalService;
 import org.apache.commons.lang3.StringUtils;
@@ -68,6 +70,7 @@ public void checkACommonWidget() {
         movie.goToReview();
 
         rottenTomatoes.checkSimpleReview();
+        assertTrue(movie.getSelfReference().getClass().equals(IOSMovie.class));
     }
 
     @Override
@@ -81,6 +84,7 @@ public void checkAnAnnotatedWidget() {
         movie.goToReview();
 
         rottenTomatoes.checkAnnotatedReview();
+        assertTrue(movie.getSelfReference().getClass().equals(AnnotatedIOSMovie.class));
     }
 
 
@@ -95,6 +99,7 @@ public void checkAnExtendedWidget() {
         movie.goToReview();
 
         rottenTomatoes.checkExtendedReview();
+        assertTrue(movie.getSelfReference().getClass().equals(AnnotatedIOSMovie.class));
     }
 
     @Override

src/test/java/io/appium/java_client/pagefactory_tests/widgets/Movie.java

@@ -15,4 +15,9 @@ protected Movie(WebElement element) {
     public abstract Object getPoster();
 
     public abstract void goToReview();
+
+    @Override
+    public Movie getSelfReference() {
+        return (Movie) super.getSelfReference();
+    }
 }

src/test/java/io/appium/java_client/pagefactory_tests/widgets/Movies.java

@@ -12,4 +12,9 @@ protected Movies(WebElement element) {
     public abstract int getMovieCount();
 
     public abstract Movie getMovie(int index);
+
+    @Override
+    public Movies getSelfReference() {
+        return (Movies) super.getSelfReference();
+    }
 }

src/test/java/io/appium/java_client/pagefactory_tests/widgets/Review.java

@@ -16,4 +16,9 @@ protected Review(WebElement element) {
     public abstract String info();
 
     public abstract Object getPoster();
+
+    @Override
+    public Review getSelfReference() {
+        return (Review) super.getSelfReference();
+    }
 }

src/test/java/io/appium/java_client/pagefactory_tests/widgets/SelendroidOverrideWidgetTest.java

@@ -3,6 +3,8 @@
 import io.appium.java_client.android.AndroidDriver;
 import io.appium.java_client.pagefactory.AppiumFieldDecorator;
 import io.appium.java_client.pagefactory.TimeOutDuration;
+import io.appium.java_client.pagefactory_tests.widgets.selendroid.annotated.AnnotatedSelendroidMovie;
+import io.appium.java_client.pagefactory_tests.widgets.selendroid.simple.SelendroidMovie;
 import io.appium.java_client.remote.AutomationName;
 import io.appium.java_client.remote.MobileCapabilityType;
 import io.appium.java_client.service.local.AppiumDriverLocalService;
@@ -72,6 +74,7 @@ public void checkACommonWidget() {
         movie.goToReview();
         driver.getPageSource();  //forcing the refreshing hierarchy
         rottenTomatoes.checkSimpleReview();
+        assertTrue(movie.getSelfReference().getClass().equals(SelendroidMovie.class));
     }
 
     @Override
@@ -85,6 +88,7 @@ public void checkAnAnnotatedWidget() {
         movie.goToReview();
         driver.getPageSource();  //forcing the refreshing hierarchy
         rottenTomatoes.checkAnnotatedReview();
+        assertTrue(movie.getSelfReference().getClass().equals(AnnotatedSelendroidMovie.class));
     }
 
 
@@ -99,6 +103,7 @@ public void checkAnExtendedWidget() {
         movie.goToReview();
         driver.getPageSource();  //forcing the refreshing hierarchy
         rottenTomatoes.checkExtendedReview();
+        assertTrue(movie.getSelfReference().getClass().equals(AnnotatedSelendroidMovie.class));
     }
 
     @Override