pyicloud/tests/__init__.py

"""Library tests."""

from pyicloud import base
from pyicloud.exceptions import PyiCloudFailedLoginException
from pyicloud.services.findmyiphone import FindMyiPhoneServiceManager, AppleDevice


AUTHENTICATED_USER = "authenticated_user"
REQUIRES_2SA_USER = "requires_2sa_user"
VALID_USERS = [AUTHENTICATED_USER, REQUIRES_2SA_USER]


class PyiCloudServiceMock(base.PyiCloudService):
    """Mocked PyiCloudService."""
    def __init__(
            self,
            apple_id,
            password=None,
            cookie_directory=None,
            verify=True,
            client_id=None,
            with_family=True
        ):
        base.PyiCloudService.__init__(self, apple_id, password, cookie_directory, verify, client_id, with_family)
        base.FindMyiPhoneServiceManager = FindMyiPhoneServiceManagerMock

    def authenticate(self):
        if not self.user.get("apple_id") or self.user.get("apple_id") not in VALID_USERS:
            raise PyiCloudFailedLoginException("Invalid email/password combination.", None)
        if not self.user.get("password") or self.user.get("password") != "valid_pass":
            raise PyiCloudFailedLoginException("Invalid email/password combination.", None)
        
        self.params.update({'dsid': 'ID'})
        self._webservices = {
            'account': {
                'url': 'account_url',
            },
            'findme': {
                'url': 'findme_url',
            },
            'calendar': {
                'url': 'calendar_url',
            },
            'contacts': {
                'url': 'contacts_url',
            },
            'reminders': {
                'url': 'reminders_url',
            }
        }

    @property
    def requires_2sa(self):
        return self.user["apple_id"] is REQUIRES_2SA_USER

    @property
    def trusted_devices(self):
        return  [
            {"deviceType": "SMS", "areaCode": "", "phoneNumber": "*******58", "deviceId": "1"}
        ]
    
    def send_verification_code(self, device):
        return device
    
    def validate_verification_code(self, device, code):
        if not device or code != 0:
            self.user["apple_id"] = AUTHENTICATED_USER
        self.authenticate()
        return not self.requires_2sa


IPHONE_DEVICE_ID = "X1x/X&x="
IPHONE_DEVICE = AppleDevice(
    {
        "msg": {
            "strobe": False,
            "userText": False,
            "playSound": True,
            "vibrate": True,
            "createTimestamp": 1568031021347,
            "statusCode": "200"
        },
        "canWipeAfterLock": True,
        "baUUID": "",
        "wipeInProgress": False,
        "lostModeEnabled": False,
        "activationLocked": True,
        "passcodeLength": 6,
        "deviceStatus": "200",
        "deviceColor": "1-6-0",
        "features": {
            "MSG": True,
            "LOC": True,
            "LLC": False,
            "CLK": False,
            "TEU": True,
            "LMG": False,
            "SND": True,
            "CLT": False,
            "LKL": False,
            "SVP": False,
            "LST": True,
            "LKM": False,
            "WMG": True,
            "SPN": False,
            "XRM": False,
            "PIN": False,
            "LCK": True,
            "REM": False,
            "MCS": False,
            "CWP": False,
            "KEY": False,
            "KPD": False,
            "WIP": True
        },
        "lowPowerMode": True,
        "rawDeviceModel": "iPhone11,8",
        "id": IPHONE_DEVICE_ID,
        "remoteLock": None,
        "isLocating": True,
        "modelDisplayName": "iPhone",
        "lostTimestamp": "",
        "batteryLevel": 0.47999998927116394,
        "mesg": None,
        "locationEnabled": True,
        "lockedTimestamp": None,
        "locFoundEnabled": False,
        "snd": {
            "createTimestamp": 1568031021347,
            "statusCode": "200"
        },
        "fmlyShare": False,
        "lostDevice": {
            "stopLostMode": False,
            "emailUpdates": False,
            "userText": True,
            "sound": False,
            "ownerNbr": "",
            "text": "",
            "createTimestamp": 1558383841233,
            "statusCode": "2204"
        },
        "lostModeCapable": True,
        "wipedTimestamp": None,
        "deviceDisplayName": "iPhone XR",
        "prsId": None,
        "audioChannels": [],
        "locationCapable": True,
        "batteryStatus": "NotCharging",
        "trackingInfo": None,
        "name": "Quentin's iPhone",
        "isMac": False,
        "thisDevice": False,
        "deviceClass": "iPhone",
        "location": {
            "isOld": False,
            "isInaccurate": False,
            "altitude": 0.0,
            "positionType": "GPS",
            "latitude": 46.012345678,
            "floorLevel": 0,
            "horizontalAccuracy": 12.012345678,
            "locationType": "",
            "timeStamp": 1568827039692,
            "locationFinished": False,
            "verticalAccuracy": 0.0,
            "longitude": 5.012345678
        },
        "deviceModel": "iphoneXR-1-6-0",
        "maxMsgChar": 160,
        "darkWake": False,
        "remoteWipe": None
    },
    None,
    None,
    None
)

DEVICES = {
    IPHONE_DEVICE_ID: IPHONE_DEVICE,
}


class FindMyiPhoneServiceManagerMock(FindMyiPhoneServiceManager):
    """Mocked FindMyiPhoneServiceManager."""
    def __init__(self, service_root, session, params, with_family=False):
        FindMyiPhoneServiceManager.__init__(self, service_root, session, params, with_family)

    def refresh_client(self):
        self._devices = DEVICES