add context related tests

Author: KazuCocoa

test/functional/android/android/device_test.rb

@@ -79,7 +79,7 @@ def test_context_related
         @@driver.set_context webview_context
         assert @@driver.current_context.start_with? 'WEBVIEW'
 
-        @@driver.set_context 'NATIVE_APP'
+        @@driver.switch_to_default_context
         assert_equal 'NATIVE_APP', @@driver.current_context
       end
 
@@ -266,6 +266,14 @@ def test_ime_available_engins
         assert_equal available_ime, @@driver.ime_active_engine
       end
 
+      def test_within_context
+        result = @@driver.within_context('NATIVE_APP') do
+          @@core.wait { @@driver.find_element :accessibility_id, 'App' }.click
+        end
+
+        assert_equal 'App', result.text
+      end
+
       private
 
       def scroll_to(text)

test/functional/ios/ios/device_test.rb

@@ -74,7 +74,7 @@ def test_context_related
         @@driver.set_context contexts.last
         assert_match 'NATIVE_APP', @@driver.current_context
 
-        @@driver.set_context 'NATIVE_APP'
+        @@driver.switch_to_default_context
         assert_equal 'NATIVE_APP', @@driver.current_context
 
         @@driver.back # go to top
@@ -183,6 +183,14 @@ def test_hidekeyboard
 
         @@driver.back
       end
+
+      def test_within_context
+        result = @@driver.within_context('NATIVE_APP') do
+          @@core.wait { @@driver.find_element :accessibility_id, 'Buttons' }.click
+        end
+
+        assert_equal 'Buttons', result.name
+      end
     end
   end
 end