bug fix

airtai · Mar 4, 2024 · 1af3945 · 1af3945
1 parent eab5fa8
commit 1af3945
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/asyncapi/base/arguments.py b/tests/asyncapi/base/arguments.py
@@ -434,7 +434,7 @@ async def handle(user: descriminator): ...
 
         schema = get_app_schema(self.build_app(broker)).to_jsonable()
 
-        key = next(schema["components"]["messages"].keys())
+        key = next(iter(schema["components"]["messages"].keys()))
         assert key == IsStr(regex=r"test[\w:]*:Handle:Message")
         assert schema["components"] == {
             "messages": {