54 lines
1.2 KiB
TypeScript
54 lines
1.2 KiB
TypeScript
import { describe, it, expect } from "vitest";
|
|
import { getAreaContext } from "../../../../src/common/entity/context/get_area_context";
|
|
import type { HomeAssistant } from "../../../../src/types";
|
|
import { mockArea, mockFloor } from "./context-mock";
|
|
|
|
describe("getAreaContext", () => {
|
|
it("should return the correct context when the area exists without a floor", () => {
|
|
const area = mockArea({
|
|
area_id: "area_1",
|
|
});
|
|
|
|
const hass = {
|
|
areas: {
|
|
area_1: area,
|
|
},
|
|
floors: {},
|
|
} as unknown as HomeAssistant;
|
|
|
|
const result = getAreaContext(area, hass);
|
|
|
|
expect(result).toEqual({
|
|
area,
|
|
floor: null,
|
|
});
|
|
});
|
|
|
|
it("should return the correct context when the area exists with a floor", () => {
|
|
const area = mockArea({
|
|
area_id: "area_2",
|
|
floor_id: "floor_1",
|
|
});
|
|
|
|
const floor = mockFloor({
|
|
floor_id: "floor_1",
|
|
});
|
|
|
|
const hass = {
|
|
areas: {
|
|
area_2: area,
|
|
},
|
|
floors: {
|
|
floor_1: floor,
|
|
},
|
|
} as unknown as HomeAssistant;
|
|
|
|
const result = getAreaContext(area, hass);
|
|
|
|
expect(result).toEqual({
|
|
area,
|
|
floor,
|
|
});
|
|
});
|
|
});
|