mock_data.py

"""Mock data for Roborock tests."""

import hashlib
import json

# All data is based on a U.S. customer with a Roborock S7 MaxV Ultra
USER_EMAIL = "user@domain.com"

BASE_URL = "https://usiot.roborock.com"

USER_ID = "user123"
K_VALUE = "domain123"
USER_DATA = {
    "uid": 123456,
    "tokentype": "token_type",
    "token": "abc123",
    "rruid": "abc123",
    "region": "us",
    "countrycode": "1",
    "country": "US",
    "nickname": "user_nickname",
    "rriot": {
        "u": USER_ID,
        "s": "pass123",
        "h": "unknown123",
        "k": K_VALUE,
        "r": {
            "r": "US",
            "a": "https://api-us.roborock.com",
            "m": "tcp://mqtt-us.roborock.com:8883",  # Skip SSL code in MQTT client library
            "l": "https://wood-us.roborock.com",
        },
    },
    "tuyaDeviceState": 2,
    "avatarurl": "https://files.roborock.com/iottest/default_avatar.png",
}
LOCAL_KEY = "key123key123key1"  # 16 bytes / 128 bits
PRODUCT_ID = "product-id-123"
HOME_DATA_SCENES_RAW = [
    {
        "id": 1234567,
        "name": "My plan",
        "param": json.dumps(
            {
                "triggers": [],
                "action": {
                    "type": "S",
                    "items": [
                        {
                            "id": 5,
                            "type": "CMD",
                            "name": "",
                            "entityId": "EEEEEEEEEEEEEE",
                            "param": json.dumps(
                                {
                                    "id": 5,
                                    "method": "do_scenes_app_start",
                                    "params": [
                                        {
                                            "fan_power": 104,
                                            "water_box_mode": 200,
                                            "mop_mode": 300,
                                            "mop_template_id": 300,
                                            "repeat": 1,
                                            "auto_dustCollection": 1,
                                            "source": 101,
                                        }
                                    ],
                                }
                            ),
                            "finishDpIds": [130],
                        },
                        {
                            "id": 4,
                            "type": "CMD",
                            "name": "",
                            "entityId": "EEEEEEEEEEEEEE",
                            "param": json.dumps(
                                {
                                    "id": 4,
                                    "method": "do_scenes_segments",
                                    "params": {
                                        "data": [
                                            {
                                                "tid": "111111111111111111",
                                                "segs": [
                                                    {"sid": 19},
                                                    {"sid": 18},
                                                    {"sid": 22},
                                                    {"sid": 21},
                                                    {"sid": 16},
                                                ],
                                                "map_flag": 0,
                                                "fan_power": 105,
                                                "water_box_mode": 201,
                                                "mop_mode": 300,
                                                "mop_template_id": 300,
                                                "repeat": 1,
                                                "clean_order_mode": 1,
                                                "auto_dry": 1,
                                                "auto_dustCollection": 1,
                                                "region_num": 0,
                                            }
                                        ],
                                        "source": 101,
                                    },
                                }
                            ),
                            "finishDpIds": [130],
                        },
                    ],
                },
                "matchType": "NONE",
                "tagId": "4444",
            }
        ),
        "enabled": True,
        "extra": None,
        "type": "WORKFLOW",
    }
]
HOME_DATA_RAW = {
    "id": 123456,
    "name": "My Home",
    "lon": None,
    "lat": None,
    "geoName": None,
    "products": [
        {
            "id": PRODUCT_ID,
            "name": "Roborock S7 MaxV",
            "code": "a27",
            "model": "roborock.vacuum.a27",
            "iconUrl": None,
            "attribute": None,
            "capability": 0,
            "category": "robot.vacuum.cleaner",
            "schema": [
                {
                    "id": "101",
                    "name": "rpc_request",
                    "code": "rpc_request_code",
                    "mode": "rw",
                    "type": "RAW",
                    "property": None,
                    "desc": None,
                },
                {
                    "id": "102",
                    "name": "rpc_response",
                    "code": "rpc_response",
                    "mode": "rw",
                    "type": "RAW",
                    "property": None,
                    "desc": None,
                },
                {
                    "id": "120",
                    "name": "错误代码",
                    "code": "error_code",
                    "mode": "ro",
                    "type": "ENUM",
                    "property": '{"range": []}',
                    "desc": None,
                },
                {
                    "id": "121",
                    "name": "设备状态",
                    "code": "state",
                    "mode": "ro",
                    "type": "ENUM",
                    "property": '{"range": []}',
                    "desc": None,
                },
                {
                    "id": "122",
                    "name": "设备电量",
                    "code": "battery",
                    "mode": "ro",
                    "type": "ENUM",
                    "property": '{"range": []}',
                    "desc": None,
                },
                {
                    "id": "123",
                    "name": "清扫模式",
                    "code": "fan_power",
                    "mode": "rw",
                    "type": "ENUM",
                    "property": '{"range": []}',
                    "desc": None,
                },
                {
                    "id": "124",
                    "name": "拖地模式",
                    "code": "water_box_mode",
                    "mode": "rw",
                    "type": "ENUM",
                    "property": '{"range": []}',
                    "desc": None,
                },
                {
                    "id": "125",
                    "name": "主刷寿命",
                    "code": "main_brush_life",
                    "mode": "rw",
                    "type": "VALUE",
                    "property": '{"max": 100, "min": 0, "step": 1, "unit": null, "scale": 1}',
                    "desc": None,
                },
                {
                    "id": "126",
                    "name": "边刷寿命",
                    "code": "side_brush_life",
                    "mode": "rw",
                    "type": "VALUE",
                    "property": '{"max": 100, "min": 0, "step": 1, "unit": null, "scale": 1}',
                    "desc": None,
                },
                {
                    "id": "127",
                    "name": "滤网寿命",
                    "code": "filter_life",
                    "mode": "rw",
                    "type": "VALUE",
                    "property": '{"max": 100, "min": 0, "step": 1, "unit": null, "scale": 1}',
                    "desc": None,
                },
                {
                    "id": "128",
                    "name": "额外状态",
                    "code": "additional_props",
                    "mode": "ro",
                    "type": "RAW",
                    "property": None,
                    "desc": None,
                },
                {
                    "id": "130",
                    "name": "完成事件",
                    "code": "task_complete",
                    "mode": "ro",
                    "type": "RAW",
                    "property": None,
                    "desc": None,
                },
                {
                    "id": "131",
                    "name": "电量不足任务取消",
                    "code": "task_cancel_low_power",
                    "mode": "ro",
                    "type": "RAW",
                    "property": None,
                    "desc": None,
                },
                {
                    "id": "132",
                    "name": "运动中任务取消",
                    "code": "task_cancel_in_motion",
                    "mode": "ro",
                    "type": "RAW",
                    "property": None,
                    "desc": None,
                },
                {
                    "id": "133",
                    "name": "充电状态",
                    "code": "charge_status",
                    "mode": "ro",
                    "type": "RAW",
                    "property": None,
                    "desc": None,
                },
                {
                    "id": "134",
                    "name": "烘干状态",
                    "code": "drying_status",
                    "mode": "ro",
                    "type": "RAW",
                    "property": None,
                    "desc": None,
                },
            ],
        }
    ],
    "devices": [
        {
            "duid": "abc123",
            "name": "Roborock S7 MaxV",
            "attribute": None,
            "activeTime": 1672364449,
            "localKey": LOCAL_KEY,
            "runtimeEnv": None,
            "timeZoneId": "America/Los_Angeles",
            "iconUrl": "no_url",
            "productId": PRODUCT_ID,
            "lon": None,
            "lat": None,
            "share": False,
            "shareTime": None,
            "online": True,
            "fv": "02.56.02",
            "pv": "1.0",
            "roomId": 2362003,
            "tuyaUuid": None,
            "tuyaMigrated": False,
            "extra": '{"RRPhotoPrivacyVersion": "1"}',
            "sn": "abc123",
            "featureSet": "2234201184108543",
            "newFeatureSet": "0000000000002041",
            "deviceStatus": {
                "121": 8,
                "122": 100,
                "123": 102,
                "124": 203,
                "125": 94,
                "126": 90,
                "127": 87,
                "128": 0,
                "133": 1,
                "120": 0,
            },
            "silentOtaSwitch": True,
        }
    ],
    "receivedDevices": [],
    "rooms": [
        {"id": 2362048, "name": "Example room 1"},
        {"id": 2362044, "name": "Example room 2"},
        {"id": 2362041, "name": "Example room 3"},
    ],
}
WASHER_PRODUCT = {
    "id": PRODUCT_ID,
    "name": "Zeo One",
    "model": "roborock.wm.a102",
    "category": "roborock.wm",
    "capability": 2,
    "schema": [
        {
            "id": "134",
            "name": "烘干状态",
            "code": "drying_status",
            "mode": "ro",
            "type": "RAW",
        },
        {
            "id": "200",
            "name": "启动",
            "code": "start",
            "mode": "rw",
            "type": "BOOL",
        },
        {
            "id": "201",
            "name": "暂停",
            "code": "pause",
            "mode": "rw",
            "type": "BOOL",
        },
        {
            "id": "202",
            "name": "关机",
            "code": "shutdown",
            "mode": "rw",
            "type": "BOOL",
        },
        {
            "id": "203",
            "name": "状态",
            "code": "status",
            "mode": "ro",
            "type": "VALUE",
        },
        {
            "id": "204",
            "name": "模式",
            "code": "mode",
            "mode": "rw",
            "type": "VALUE",
        },
        {
            "id": "205",
            "name": "程序",
            "code": "program",
            "mode": "rw",
            "type": "VALUE",
        },
        {
            "id": "206",
            "name": "童锁",
            "code": "child_lock",
            "mode": "rw",
            "type": "BOOL",
        },
        {
            "id": "207",
            "name": "洗涤温度",
            "code": "temp",
            "mode": "rw",
            "type": "VALUE",
        },
        {
            "id": "208",
            "name": "漂洗次数",
            "code": "rinse_times",
            "mode": "rw",
            "type": "VALUE",
        },
        {
            "id": "209",
            "name": "滚筒转速",
            "code": "spin_level",
            "mode": "rw",
            "type": "VALUE",
        },
        {
            "id": "210",
            "name": "干燥度",
            "code": "drying_mode",
            "mode": "rw",
            "type": "VALUE",
        },
        {
            "id": "211",
            "name": "自动投放-洗衣液",
            "code": "detergent_set",
            "mode": "rw",
            "type": "BOOL",
        },
        {
            "id": "212",
            "name": "自动投放-柔顺剂",
            "code": "softener_set",
            "mode": "rw",
            "type": "BOOL",
        },
        {
            "id": "213",
            "name": "洗衣液投放量",
            "code": "detergent_type",
            "mode": "rw",
            "type": "VALUE",
        },
        {
            "id": "214",
            "name": "柔顺剂投放量",
            "code": "softener_type",
            "mode": "rw",
            "type": "VALUE",
        },
        {
            "id": "217",
            "name": "预约时间",
            "code": "countdown",
            "mode": "rw",
            "type": "VALUE",
        },
        {
            "id": "218",
            "name": "洗衣剩余时间",
            "code": "washing_left",
            "mode": "ro",
            "type": "VALUE",
        },
        {
            "id": "219",
            "name": "门锁状态",
            "code": "doorlock_state",
            "mode": "ro",
            "type": "BOOL",
        },
        {
            "id": "220",
            "name": "故障",
            "code": "error",
            "mode": "ro",
            "type": "VALUE",
        },
        {
            "id": "221",
            "name": "云程序设置",
            "code": "custom_param_save",
            "mode": "rw",
            "type": "VALUE",
        },
        {
            "id": "222",
            "name": "云程序读取",
            "code": "custom_param_get",
            "mode": "ro",
            "type": "VALUE",
        },
        {
            "id": "223",
            "name": "提示音",
            "code": "sound_set",
            "mode": "rw",
            "type": "BOOL",
        },
        {
            "id": "224",
            "name": "距离上次筒自洁次数",
            "code": "times_after_clean",
            "mode": "ro",
            "type": "VALUE",
        },
        {
            "id": "225",
            "name": "记忆洗衣偏好开关",
            "code": "default_setting",
            "mode": "rw",
            "type": "BOOL",
        },
        {
            "id": "226",
            "name": "洗衣液用尽",
            "code": "detergent_empty",
            "mode": "ro",
            "type": "BOOL",
        },
        {
            "id": "227",
            "name": "柔顺剂用尽",
            "code": "softener_empty",
            "mode": "ro",
            "type": "BOOL",
        },
        {
            "id": "229",
            "name": "筒灯设定",
            "code": "light_setting",
            "mode": "rw",
            "type": "BOOL",
        },
        {
            "id": "230",
            "name": "洗衣液投放量（单次）",
            "code": "detergent_volume",
            "mode": "rw",
            "type": "VALUE",
        },
        {
            "id": "231",
            "name": "柔顺剂投放量（单次）",
            "code": "softener_volume",
            "mode": "rw",
            "type": "VALUE",
        },
        {
            "id": "232",
            "name": "远程控制授权",
            "code": "app_authorization",
            "mode": "rw",
            "type": "VALUE",
        },
        {
            "id": "10000",
            "name": "ID点查询",
            "code": "id_query",
            "mode": "rw",
            "type": "STRING",
        },
        {
            "id": "10001",
            "name": "防串货",
            "code": "f_c",
            "mode": "ro",
            "type": "STRING",
        },
        {
            "id": "10004",
            "name": "语音包/OBA信息",
            "code": "snd_state",
            "mode": "rw",
            "type": "STRING",
        },
        {
            "id": "10005",
            "name": "产品信息",
            "code": "product_info",
            "mode": "ro",
            "type": "STRING",
        },
        {
            "id": "10006",
            "name": "隐私协议",
            "code": "privacy_info",
            "mode": "rw",
            "type": "STRING",
        },
        {
            "id": "10007",
            "name": "OTA info",
            "code": "ota_nfo",
            "mode": "rw",
            "type": "STRING",
        },
        {
            "id": "10008",
            "name": "洗衣记录",
            "code": "washing_log",
            "mode": "ro",
            "type": "BOOL",
        },
        {
            "id": "10101",
            "name": "rpc req",
            "code": "rpc_req",
            "mode": "wo",
            "type": "STRING",
        },
        {
            "id": "10102",
            "name": "rpc resp",
            "code": "rpc_resp",
            "mode": "ro",
            "type": "STRING",
        },
    ],
}
ZEO_ONE_DEVICE = {
    "duid": "zeo_duid",
    "name": "Zeo One",
    "localKey": LOCAL_KEY,
    "fv": "01.00.94",
    "productId": PRODUCT_ID,
    "activeTime": 1699964128,
    "timeZoneId": "Europe/Berlin",
    "iconUrl": "",
    "share": True,
    "shareTime": 1712763572,
    "online": True,
    "pv": "A01",
    "tuyaMigrated": False,
    "sn": "zeo_sn",
    "featureSet": "0",
    "newFeatureSet": "40",
    "deviceStatus": {
        "208": 2,
        "205": 33,
        "221": 0,
        "226": 0,
        "10001": '{"f":"t"}',
        "214": 2,
        "225": 0,
        "232": 0,
        "222": 347414,
        "206": 0,
        "200": 1,
        "219": 0,
        "223": 0,
        "220": 0,
        "201": 0,
        "202": 1,
        "10005": '{"sn":"zeo_sn","ssid":"internet","timezone":"Europe/Berlin","posix_timezone":"CET-1CEST,M3.5.0,M10.5.0/3","ip":"192.111.11.11","mac":"b0:4a:00:00:00:00","rssi":-57,"oba":{"language":"en","name":"A.03.0403_CE","bom":"A.03.0403","location":"de","wifiplan":"EU","timezone":"CET-1CEST,M3.5.0,M10.5.0/3;Europe/Berlin","logserver":"awsde0","loglevel":"4","featureset":"0"}}',  # noqa: E501
        "211": 1,
        "210": 1,
        "217": 0,
        "203": 7,
        "213": 2,
        "209": 7,
        "224": 21,
        "218": 227,
        "212": 1,
        "207": 4,
        "204": 1,
        "10007": '{"mqttOtaData":{"mqttOtaStatus":{"status":"IDLE"}}}',
        "227": 1,
    },
    "silentOtaSwitch": False,
    "f": False,
}

CLEAN_RECORD = {
    "begin": 1672543330,
    "end": 1672544638,
    "duration": 1176,
    "area": 20965000,
    "error": 0,
    "complete": 1,
    "start_type": 2,
    "clean_type": 3,
    "finish_reason": 56,
    "dust_collection_status": 1,
    "avoid_count": 19,
    "wash_count": 2,
    "map_flag": 0,
}

CLEAN_SUMMARY = {
    "clean_time": 74382,
    "clean_area": 1159182500,
    "clean_count": 31,
    "dust_collection_count": 25,
    "records": [
        1672543330,
        1672458041,
    ],
}

CONSUMABLE = {
    "main_brush_work_time": 74382,
    "side_brush_work_time": 74383,
    "filter_work_time": 74384,
    "filter_element_work_time": 0,
    "sensor_dirty_time": 74385,
    "strainer_work_times": 65,
    "dust_collection_work_times": 25,
    "cleaning_brush_work_times": 66,
}

DND_TIMER = {
    "start_hour": 22,
    "start_minute": 0,
    "end_hour": 7,
    "end_minute": 0,
    "enabled": 1,
}

STATUS = {
    "msg_ver": 2,
    "msg_seq": 458,
    "state": 8,
    "battery": 100,
    "clean_time": 1176,
    "clean_area": 20965000,
    "error_code": 0,
    "map_present": 1,
    "in_cleaning": 0,
    "in_returning": 0,
    "in_fresh_state": 1,
    "lab_status": 1,
    "water_box_status": 1,
    "back_type": -1,
    "wash_phase": 0,
    "wash_ready": 0,
    "fan_power": 102,
    "dnd_enabled": 0,
    "map_status": 3,
    "is_locating": 0,
    "lock_status": 0,
    "water_box_mode": 203,
    "water_box_carriage_status": 1,
    "mop_forbidden_enable": 1,
    "camera_status": 3457,
    "is_exploring": 0,
    "home_sec_status": 0,
    "home_sec_enable_password": 0,
    "adbumper_status": [0, 0, 0],
    "water_shortage_status": 0,
    "dock_type": 3,
    "dust_collection_status": 0,
    "auto_dust_collection": 1,
    "avoid_count": 19,
    "mop_mode": 300,
    "debug_mode": 0,
    "collision_avoid_status": 1,
    "switch_map_mode": 0,
    "dock_error_status": 0,
    "charge_status": 1,
    "unsave_map_reason": 0,
    "unsave_map_flag": 0,
}

BASE_URL_REQUEST = {
    "code": 200,
    "msg": "success",
    "data": {"url": "https://sample.com"},
}

GET_CODE_RESPONSE = {"code": 200, "msg": "success", "data": None}
HASHED_USER = hashlib.md5((USER_ID + ":" + K_VALUE).encode()).hexdigest()[2:10]
MQTT_PUBLISH_TOPIC = f"rr/m/o/{USER_ID}/{HASHED_USER}/{PRODUCT_ID}"
TEST_LOCAL_API_HOST = "1.1.1.1"