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")
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")
def test_start_success_under_api_18(self):
server = AutomatorServer()
server.push = MagicMock()
server.push.return_value = ["bundle.jar", "uiautomator-stub.jar"]
server.ping = MagicMock()
server.ping.return_value = "pong"
server.sdk_version = MagicMock()
server.sdk_version.return_value = 17
server.adb = MagicMock()
server.start()
server.adb.cmd.assert_called_with('shell', 'uiautomator', 'runtest',
'bundle.jar', 'uiautomator-stub.jar',
'-c',
'com.github.uiautomatorstub.Stub')
def test_start_success_over_api_18(self):
server = AutomatorServer()
server.push = MagicMock()
server.push.return_value = ["bundle.jar", "uiautomator-stub.jar"]
server.ping = MagicMock()
server.ping.return_value = "pong"
server.sdk_version = MagicMock()
server.sdk_version.return_value = 20
server.adb = MagicMock()
server.start()
server.adb.cmd.assert_called_with(
"shell", "am", "instrument", "-w",
"com.github.uiautomator.test/android.support.test.runner.AndroidJUnitRunner"
)
