Files
frontend/test/common/entity/context/context-mock.ts
2025-09-15 14:18:52 +00:00

114 lines
2.3 KiB
TypeScript

import type { HassEntity } from "home-assistant-js-websocket";
import type { AreaRegistryEntry } from "../../../../src/data/area_registry";
import type { DeviceRegistryEntry } from "../../../../src/data/device_registry";
import type {
EntityRegistryDisplayEntry,
EntityRegistryEntry,
} from "../../../../src/data/entity_registry";
import type { FloorRegistryEntry } from "../../../../src/data/floor_registry";
export const mockStateObj = (partial: Partial<HassEntity>): HassEntity => ({
entity_id: "",
attributes: {},
state: "on",
last_changed: "",
last_updated: "",
context: {
id: "",
user_id: null,
parent_id: null,
},
...partial,
});
export const mockEntity = (
partial: Partial<EntityRegistryDisplayEntry>
): EntityRegistryDisplayEntry => ({
entity_id: "",
labels: [],
...partial,
});
export const mockEntityEntry = (
partial: Partial<EntityRegistryEntry>
): EntityRegistryEntry => ({
entity_id: "",
name: null,
icon: null,
platform: "",
config_entry_id: null,
config_subentry_id: null,
device_id: null,
area_id: null,
labels: [],
disabled_by: null,
hidden_by: null,
entity_category: null,
has_entity_name: false,
unique_id: "",
id: "",
options: null,
categories: {},
created_at: 0,
modified_at: 0,
...partial,
});
export const mockDevice = (
partial: Partial<DeviceRegistryEntry>
): DeviceRegistryEntry => ({
id: "",
config_entries: [],
config_entries_subentries: {},
connections: [],
identifiers: [],
manufacturer: null,
model: null,
model_id: null,
name: null,
labels: [],
sw_version: null,
hw_version: null,
serial_number: null,
via_device_id: null,
area_id: null,
name_by_user: null,
entry_type: null,
disabled_by: null,
configuration_url: null,
primary_config_entry: null,
created_at: 0,
modified_at: 0,
...partial,
});
export const mockArea = (
partial: Partial<AreaRegistryEntry>
): AreaRegistryEntry => ({
aliases: [],
area_id: "",
name: "",
floor_id: null,
created_at: 0,
modified_at: 0,
humidity_entity_id: null,
temperature_entity_id: null,
icon: null,
labels: [],
picture: null,
...partial,
});
export const mockFloor = (
partial: Partial<FloorRegistryEntry>
): FloorRegistryEntry => ({
aliases: [],
floor_id: "",
name: "",
created_at: 0,
modified_at: 0,
icon: null,
level: 0,
...partial,
});