add size check in page source for easy regression

Author: KazuCocoa

test/functional/android/webdriver/device_test.rb

@@ -39,9 +39,36 @@ def test_set_immediate_value
       end
 
       def test_page_source
+        require 'rexml/document'
+
+        expected = %w(
+          hierarchy
+          android.widget.FrameLayout
+          android.view.ViewGroup
+          android.widget.FrameLayout
+          android.view.ViewGroup
+          android.widget.TextView
+          android.widget.FrameLayout
+          android.widget.ListView
+          android.widget.TextView
+          android.widget.TextView
+          android.widget.TextView
+          android.widget.TextView
+          android.widget.TextView
+          android.widget.TextView
+          android.widget.TextView
+          android.widget.TextView
+          android.widget.TextView
+          android.widget.TextView
+          android.widget.TextView
+          android.widget.TextView
+        )
+
         s_source = @driver.page_source
+        xml = REXML::Document.new s_source
 
-        assert s_source.include?('o.appium.android.apis')
+        assert s_source.include?('io.appium.android.apis')
+        assert_equal expected, xml[1].elements.each('//*'){ |v| v }.map{ |v| v.name }
       end
 
       # def test_location

test/functional/ios/webdriver/device_test.rb

@@ -48,16 +48,22 @@ def by_predicate(value)
 
       def test_page_source
         require 'json'
+        require 'rexml/document'
+
         response = Net::HTTP.get(URI('http://localhost:8100/source'))
         source = JSON.parse(response)['value']
+        xml = REXML::Document.new source
 
         assert !source.include?('AppiumAUT')
         assert source.include?('XCUIElementTypeApplication type')
+        assert xml[2].elements.each('//*'){ |v| v }.map{ |v| v.name } == 77
 
         s_source = @@driver.page_source
+        s_xml = REXML::Document.new s_source
 
         assert s_source.include?('AppiumAUT')
         assert s_source.include?('XCUIElementTypeApplication type')
+        assert s_xml[2].elements.each('//*'){ |v| v }.map{ |v| v.name } == 81
       end
 
       def test_location