111 lines
3.8 KiB
TypeScript
111 lines
3.8 KiB
TypeScript
import { afterEach, describe, expect, it, vi } from 'vitest';
|
|
import { homeAssistantWSRequest } from '../../../../../src/utils/ha';
|
|
import {
|
|
createEntityRegistryCache,
|
|
EntityRegistryManager,
|
|
} from '../../../../../src/utils/ha/registry/entity';
|
|
import { createHASS, createRegistryEntity } from '../../../../test-utils.js';
|
|
|
|
vi.mock('../../../../../src/utils/ha');
|
|
|
|
describe('EntityRegistryManager', () => {
|
|
afterEach(() => {
|
|
vi.clearAllMocks();
|
|
});
|
|
|
|
describe('getEntity', () => {
|
|
it('should not fetch when cached', async () => {
|
|
const cache = createEntityRegistryCache();
|
|
const testEntity = createRegistryEntity({ entity_id: 'test' });
|
|
|
|
cache.add(testEntity);
|
|
|
|
const manager = new EntityRegistryManager(cache);
|
|
expect(await manager.getEntity(createHASS(), 'test')).toEqual(testEntity);
|
|
|
|
expect(homeAssistantWSRequest).not.toHaveBeenCalled();
|
|
});
|
|
|
|
it('should fetch and cache when not cached', async () => {
|
|
const testEntity = createRegistryEntity({ entity_id: 'test' });
|
|
|
|
const manager = new EntityRegistryManager(createEntityRegistryCache());
|
|
vi.mocked(homeAssistantWSRequest).mockResolvedValueOnce(testEntity);
|
|
|
|
expect(await manager.getEntity(createHASS(), 'test')).toEqual(testEntity);
|
|
expect(homeAssistantWSRequest).toBeCalledTimes(1);
|
|
|
|
expect(await manager.getEntity(createHASS(), 'test')).toEqual(testEntity);
|
|
expect(homeAssistantWSRequest).toBeCalledTimes(1);
|
|
});
|
|
|
|
it('should return null when entity does not exist', async () => {
|
|
vi.mocked(homeAssistantWSRequest).mockRejectedValueOnce(new Error('Not found'));
|
|
|
|
const manager = new EntityRegistryManager(createEntityRegistryCache());
|
|
expect(await manager.getEntity(createHASS(), 'missing')).toBeNull();
|
|
});
|
|
});
|
|
|
|
it('getEntities', () => {
|
|
const cachedEntity = createRegistryEntity({ entity_id: 'cached' });
|
|
const notCachedEntity = createRegistryEntity({ entity_id: 'not-cached' });
|
|
|
|
const cache = createEntityRegistryCache();
|
|
cache.add(cachedEntity);
|
|
|
|
const manager = new EntityRegistryManager(cache);
|
|
vi.mocked(homeAssistantWSRequest).mockResolvedValueOnce(notCachedEntity);
|
|
vi.mocked(homeAssistantWSRequest).mockRejectedValueOnce(new Error('Not found'));
|
|
|
|
expect(
|
|
manager.getEntities(createHASS(), ['cached', 'not-cached', 'missing']),
|
|
).resolves.toEqual(
|
|
new Map([
|
|
['cached', cachedEntity],
|
|
['not-cached', notCachedEntity],
|
|
]),
|
|
);
|
|
});
|
|
|
|
it('fetchEntityList', async () => {
|
|
const hass = createHASS();
|
|
const entity = createRegistryEntity({ entity_id: 'cached' });
|
|
vi.mocked(homeAssistantWSRequest).mockResolvedValueOnce([entity]);
|
|
|
|
const manager = new EntityRegistryManager(createEntityRegistryCache());
|
|
|
|
await manager.fetchEntityList(hass);
|
|
|
|
expect(homeAssistantWSRequest).toBeCalledTimes(1);
|
|
expect(homeAssistantWSRequest).toBeCalledWith(expect.anything(), expect.anything(), {
|
|
type: 'config/entity_registry/list',
|
|
});
|
|
|
|
expect(await manager.getEntity(hass, 'cached')).toEqual(entity);
|
|
expect(homeAssistantWSRequest).toBeCalledTimes(1);
|
|
|
|
await manager.fetchEntityList(hass);
|
|
expect(homeAssistantWSRequest).toBeCalledTimes(1);
|
|
});
|
|
|
|
it('getMatchingEntities', async () => {
|
|
const matchingEntity = createRegistryEntity({ entity_id: 'matching' });
|
|
const notMatchingEntity = createRegistryEntity({ entity_id: 'not-matching' });
|
|
const hass = createHASS();
|
|
|
|
vi.mocked(homeAssistantWSRequest).mockResolvedValueOnce([
|
|
matchingEntity,
|
|
notMatchingEntity,
|
|
]);
|
|
|
|
const manager = new EntityRegistryManager(createEntityRegistryCache());
|
|
expect(
|
|
await manager.getMatchingEntities(
|
|
hass,
|
|
(entity) => entity.entity_id == 'matching',
|
|
),
|
|
).toEqual([matchingEntity]);
|
|
});
|
|
});
|