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" )