def test_bootstrap(bootstrap): obj = Bootstrap.from_unifi_dict(**deepcopy(bootstrap)) set_no_debug() obj_construct = Bootstrap.from_unifi_dict(**deepcopy(bootstrap)) set_debug() obj_dict = obj.unifi_dict() # TODO: del bootstrap["legacyUFVs"] del bootstrap["displays"] del bootstrap["doorlocks"] del bootstrap["chimes"] del bootstrap["schedules"] del bootstrap["nvr"]["uiVersion"] del bootstrap["nvr"]["errorCode"] del bootstrap["nvr"]["wifiSettings"] del bootstrap["nvr"]["ssoChannel"] del bootstrap["nvr"]["smartDetectAgreement"] for model_type in ModelType.bootstrap_models(): key = model_type + "s" expected_data = bootstrap.pop(key) actual_data = obj_dict.pop(key) assert len(expected_data) == len(actual_data) for index, expected in enumerate(expected_data): actual = actual_data[index] compare_objs(expected["modelKey"], expected, actual) assert bootstrap == obj_dict assert obj == obj_construct
def compare_devices(data): obj = create_from_unifi_dict(deepcopy(data)) obj_dict = obj.unifi_dict() compare_objs(obj.model.value, data, obj_dict) set_no_debug() obj_construct = create_from_unifi_dict(deepcopy(data)) assert obj == obj_construct set_debug()
async def test_get_events(protect_client: ProtectApiClient, raw_events): expected_events = [] for event in raw_events: if event["score"] >= 50 and event["type"] in EventType.device_events(): expected_events.append(event) protect_client._minimum_score = 50 events = await protect_client.get_events() assert len(events) == len(expected_events) for index, event in enumerate(events): compare_objs(event.model.value, expected_events[index], event.unifi_dict()) if event.type.value in EventType.motion_events(): await check_motion_event(event)
def test_bootstrap(bootstrap): obj = Bootstrap.from_unifi_dict(**deepcopy(bootstrap)) set_no_debug() obj_construct = Bootstrap.from_unifi_dict(**deepcopy(bootstrap)) set_debug() obj_dict = obj.unifi_dict() # TODO: del bootstrap["legacyUFVs"] del bootstrap["displays"] del bootstrap["chimes"] del bootstrap["schedules"] if "deviceGroups" in bootstrap: del bootstrap["deviceGroups"] del bootstrap["nvr"]["uiVersion"] del bootstrap["nvr"]["errorCode"] del bootstrap["nvr"]["wifiSettings"] del bootstrap["nvr"]["ssoChannel"] del bootstrap["nvr"]["smartDetectAgreement"] bootstrap["nvr"]["systemInfo"].pop("ustorage", None) for model_type in ModelType.bootstrap_models(): key = model_type + "s" expected_data = bootstrap.pop(key) actual_data = obj_dict.pop(key) assert len(expected_data) == len(actual_data) for index, expected in enumerate(expected_data): actual = actual_data[index] compare_objs(expected["modelKey"], expected, actual) for key in NEW_FIELDS.intersection(obj_dict["nvr"].keys()): if key not in bootstrap["nvr"]: del obj_dict["nvr"][key] bootstrap["nvr"]["ports"]["piongw"] = bootstrap["nvr"]["ports"].get("piongw") assert bootstrap == obj_dict assert obj == obj_construct