AI generated test for get_chat function

test_get_chat.py

def test_get_chat():
    # Test a Message event
    message = Message(
        id=1,
        from_user=User(id=1, first_name="Alice"),
        chat=Chat(id=1, type="private"),
        date=datetime.now(),
        text="Hello!",
    )
    assert get_chat(message) == message.chat

    # Test a DialogUpdateEvent
    dialog_update_event = DialogUpdateEvent(
        id=1,
        from_user=User(id=1, first_name="Alice"),
        chat=Chat(id=1, type="private"),
        date=datetime.now(),
        message=message,
    )
    assert get_chat(dialog_update_event) == dialog_update_event.chat

    # Test a ChatMemberUpdated event
    chat_member_updated = ChatMemberUpdated(
        id=1,
        from_user=User(id=1, first_name="Alice"),
        chat=Chat(id=1, type="private"),
        date=datetime.now(),
    )
    assert get_chat(chat_member_updated) == chat_member_updated.chat

    # Test a CallbackQuery with a message
    callback_query = CallbackQuery(
        id=1,
        from_user=User(id=1, first_name="Alice"),
        chat_instance="123456",
        data="hello",
        message=message,
    )
    assert get_chat(callback_query) == callback_query.message.chat

    # Test a CallbackQuery without a message
    callback_query = CallbackQuery(
        id=1,
        from_user=User(id=1, first_name="Alice"),
        chat_instance="123456",
        data="hello",
        message=None,
    )
    assert get_chat(callback_query) == Chat(id=1, type="")