Python AutomatorServer.screenshot Examples

Example #1

File: test_server.py Project: anuraagbaishya/uiautomator

    def test_screenshot(self):
        server = AutomatorServer()
        server.sdk_version = MagicMock()
        server.sdk_version.return_value = 17
        self.assertEqual(server.screenshot(), None)

        server.sdk_version.return_value = 18
        self.urlopen.return_value.read = MagicMock()
        self.urlopen.return_value.read.return_value = b"123456"
        self.assertEqual(server.screenshot(), b"123456")
        self.assertEqual(server.screenshot("/tmp/test.txt"), "/tmp/test.txt")

Example #2

File: test_server.py Project: Andy-hpliu/uiautomator

    def test_screenshot(self):
        server = AutomatorServer()
        server.sdk_version = MagicMock()
        server.sdk_version.return_value = 17
        self.assertEqual(server.screenshot(), None)

        server.sdk_version.return_value = 18
        self.urlopen.return_value.read = MagicMock()
        self.urlopen.return_value.read.return_value = b"123456"
        self.assertEqual(server.screenshot(), b"123456")
        self.assertEqual(server.screenshot("/tmp/test.txt"), "/tmp/test.txt")