implement screenshot feature

[platform/core/uifw/aurum.git] / protocol / examples / python / testFeatures.py

diff --git a/protocol/examples/python/testFeatures.py b/protocol/examples/python/testFeatures.py
index 39aeba3..a4d95e6 100644 (file)
--- a/protocol/examples/python/testFeatures.py
+++ b/protocol/examples/python/testFeatures.py
@@ -269,6 +269,14 @@ def getLocationTest(stub):
     if response.lat < 0: return False
     return True
 
+def takeScreenshotTest(stub):
+    responses = stub.takeScreenshot(ReqTakeScreenshot())
+    image = open("screenshot.png", "wb")
+    for response in responses:
+        image.write(response.image)
+    image.close()
+    return True;
+
 def defaultSetup(stub):
     if stub.getAppInfo(ReqGetAppInfo(packageName='com.samsung.ui-widget-sample')).isRunning:
         stub.closeApp(ReqCloseApp(packageName='com.samsung.ui-widget-sample'))
@@ -310,6 +318,7 @@ def run():
         runTest(stub, touchTest)
         runTest(stub, sendKeyTest)
         runTest(stub, setValueClearTest)
+        runTest(stub, takeScreenshotTest)
         runTestWithoutSetupAndTearDown(stub, installAppTest)
         runTestWithoutSetupAndTearDown(stub, launchAppTest)
         runTestWithoutSetupAndTearDown(stub, getAppInfoTest)