From 60dd1f4f3ceaab05dcb6c9f5506c5064d8052481 Mon Sep 17 00:00:00 2001
From: Jesse Hills <3060199+jesserockz@users.noreply.github.com>
Date: Mon, 22 Apr 2024 04:35:48 +0000
Subject: [PATCH] Add model tests

---
 tests/test_model.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/tests/test_model.py b/tests/test_model.py
index 33f2624..9a0454c 100644
--- a/tests/test_model.py
+++ b/tests/test_model.py
@@ -16,6 +16,7 @@ from aioesphomeapi.api_pb2 import (
     DateStateResponse,
     DateTimeStateResponse,
     DeviceInfoResponse,
+    EventResponse,
     FanStateResponse,
     HomeassistantServiceMap,
     HomeassistantServiceResponse,
@@ -27,6 +28,7 @@ from aioesphomeapi.api_pb2 import (
     ListEntitiesCoverResponse,
     ListEntitiesDateResponse,
     ListEntitiesDateTimeResponse,
+    ListEntitiesEventResponse,
     ListEntitiesFanResponse,
     ListEntitiesLightResponse,
     ListEntitiesLockResponse,
@@ -82,6 +84,8 @@ from aioesphomeapi.model import (
     DateTimeInfo,
     DateTimeState,
     DeviceInfo,
+    EventInfo,
+    Event,
     FanInfo,
     FanState,
     HomeassistantServiceCall,
@@ -280,6 +284,8 @@ def test_api_version_ord():
         (TimeState, TimeStateResponse),
         (DateTimeInfo, ListEntitiesDateTimeResponse),
         (DateTimeState, DateTimeStateResponse),
+        (EventInfo, ListEntitiesEventResponse),
+        (Event, EventResponse),
     ],
 )
 def test_basic_pb_conversions(model, pb):