Improve engine identification and id generation.

2023-10-08 14:53:05 -07:00
parent fb5fbe4ce1
commit 7a33f75889
12 changed files with 222 additions and 142 deletions
@@ -1,7 +1,10 @@
 import { HomeAssistant } from 'custom-card-helpers';
 import { describe, expect, it } from 'vitest';
-import { hasHAConnectionStateChanged } from '../../../src/utils/ha/index.js';
-import { createHASS } from '../../test-utils.js';
+import {
+  getEntityIcon,
+  hasHAConnectionStateChanged,
+} from '../../../src/utils/ha/index.js';
+import { createHASS, createStateEntity } from '../../test-utils.js';

 const createConnected = (connected: boolean): HomeAssistant => {
  const hass = createHASS();
@@ -43,3 +46,24 @@ describe('hasHAConnectionStateChanged', () => {
    expect(hasHAConnectionStateChanged(null, null)).toBeFalsy();
  });
 });
+
+describe('getEntityIcon', () => {
+  it('should get icon from attributes', () => {
+    expect(
+      getEntityIcon(
+        createHASS({
+          'camera.test': createStateEntity({
+            attributes: {
+              icon: 'mdi:cow',
+            },
+          }),
+        }),
+        'camera.test',
+      ),
+    ).toBe('mdi:cow');
+  });
+
+  it('should get icon from domain', () => {
+    expect(getEntityIcon(createHASS(), 'camera.test')).toBe('mdi:video');
+  });
+});