Add screenrecord unittest

appium/webdriver/extensions/screen_record.py

@@ -73,9 +73,9 @@ def start_recording_screen(self, **options):
                 but doing so results in larger movie files.
 
         Returns:
-            bytes: Base-64 encoded content of the recorded media file or an empty string
-                if the file has been successfully uploaded to a remote location
-                (depends on the actual `remotePath` value).
+            bytes: Base-64 encoded content of the recorded media
+                if `stop_recording_screen` isn't called after previous `start_recording_screen`.
+                Otherwise returns an empty string.
         """
         if 'password' in options:
             options['pass'] = options['password']
@@ -101,7 +101,7 @@ def stop_recording_screen(self, **options):
                 Only has an effect if `remotePath` is set.
 
         Returns:
-            bytes:Base-64 encoded content of the recorded media file or an empty string
+            bytes: Base-64 encoded content of the recorded media file or an empty string
                 if the file has been successfully uploaded to a remote location
                 (depends on the actual `remotePath` value).
         """

test/unit/webdriver/device/app_test.py → test/unit/webdriver/app_test.py

@@ -22,7 +22,7 @@
 )
 
 
-class TestApp(object):
+class TestWebDriverApp(object):
 
     @httpretty.activate
     def test_reset(self):

test/unit/webdriver/device/context_test.py → test/unit/webdriver/context_test.py

@@ -17,7 +17,7 @@
 from test.unit.helper.test_helper import android_w3c_driver, appium_command
 
 
-class TestWebDriverDeviceContext(object):
+class TestWebDriverContext(object):
 
     @httpretty.activate
     def test_get_contexts(self):

test/unit/webdriver/device/activities_test.py

@@ -21,7 +21,7 @@
 )
 
 
-class TestWebDriverDeviceActivities(object):
+class TestWebDriverActivities(object):
 
     @httpretty.activate
     def test_start_activity(self):

test/unit/webdriver/device/clipboard_test.py

@@ -24,7 +24,7 @@
 )
 
 
-class TestWebDriverDeviceClipboard(object):
+class TestWebDriverClipboard(object):
 
     @httpretty.activate
     def test_set_clipboard_with_url(self):

test/unit/webdriver/device/fingerprint_test.py

@@ -21,7 +21,7 @@
 )
 
 
-class TestWebDriverDeviceFingerprint(object):
+class TestWebDriverFingerprint(object):
 
     @httpretty.activate
     def test_finger_print(self):

test/unit/webdriver/device/lock_test.py

@@ -21,7 +21,7 @@
 )
 
 
-class TestWebDriverDeviceLock(object):
+class TestWebDriverLock(object):
 
     @httpretty.activate
     def test_lock(self):

test/unit/webdriver/device/shake_test.py

@@ -19,7 +19,7 @@
 from test.unit.helper.test_helper import android_w3c_driver, appium_command
 
 
-class TestWebDriverDeviceShake(object):
+class TestWebDriverShake(object):
 
     @httpretty.activate
     def test_shake(self):

test/unit/webdriver/device/execute_driver_test.py → test/unit/webdriver/execute_driver_test.py

@@ -23,7 +23,7 @@
 )
 
 
-class TestWebDriverDeviceActivities(object):
+class TestWebDriverExecuteDriver(object):
 
     @httpretty.activate
     def test_batch(self):

test/unit/webdriver/screen_record_test.py

@@ -0,0 +1,48 @@
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import httpretty
+
+from test.unit.helper.test_helper import (
+    android_w3c_driver,
+    appium_command,
+    get_httpretty_request_body
+)
+
+
+class TestWebDriverScreenRecord(object):
+
+    @httpretty.activate
+    def test_start_recording_screen(self):
+        driver = android_w3c_driver()
+        httpretty.register_uri(
+            httpretty.POST,
+            appium_command('/session/1234567890/appium/start_recording_screen'),
+        )
+        assert driver.start_recording_screen(user='userA', password='12345') is None
+
+        d = get_httpretty_request_body(httpretty.last_request())
+        assert d['options']['user'] == 'userA'
+        assert d['options']['pass'] == '12345'
+        assert 'password' not in d['options'].keys()
+
+    @httpretty.activate
+    def test_stop_recording_screen(self):
+        driver = android_w3c_driver()
+        httpretty.register_uri(
+            httpretty.POST,
+            appium_command('/session/1234567890/appium/stop_recording_screen'),
+            body='{"value": "b64_video_data"}'
+        )
+        assert driver.stop_recording_screen(user='userA', password='12345') == 'b64_video_data'
+
+        d = get_httpretty_request_body(httpretty.last_request())
+        assert d['options']['user'] == 'userA'
+        assert d['options']['pass'] == '12345'
+        assert 'password' not in d['options'].keys()