mirror of https://github.com/mvt-project/mvt.git
35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
# Mobile Verification Toolkit (MVT)
|
|
# Copyright (c) 2021-2022 Claudio Guarnieri.
|
|
# Use of this software is governed by the MVT License 1.1 that can be found at
|
|
# https://license.mvt.re/1.1/
|
|
|
|
import logging
|
|
|
|
from pymobiledevice3.lockdown import LockdownClient
|
|
|
|
from mvt.common.module import run_module
|
|
from mvt.ios.modules.usb.device_info import DeviceInfo
|
|
|
|
|
|
class TestUSBDeviceInfo:
|
|
def test_run(self, mocker):
|
|
mocker.patch("pymobiledevice3.usbmux.select_device")
|
|
mocker.patch("pymobiledevice3.service_connection.ServiceConnection.create")
|
|
mocker.patch(
|
|
"pymobiledevice3.lockdown.LockdownClient.query_type",
|
|
return_value="com.apple.mobile.lockdown")
|
|
mocker.patch(
|
|
"pymobiledevice3.lockdown.LockdownClient.validate_pairing",
|
|
return_value=True)
|
|
mocker.patch(
|
|
"pymobiledevice3.lockdown.LockdownClient.get_value",
|
|
return_value={'DeviceClass': 'iPhone', 'ProductVersion': '14.3'}
|
|
)
|
|
|
|
lockdown = LockdownClient()
|
|
|
|
m = DeviceInfo(log=logging)
|
|
m.lockdown = lockdown
|
|
run_module(m)
|
|
assert len(m.results) == 2
|