feat: return business in Update.Msg()

mr-linch · Apr 29, 2024 · ece7c55 · ece7c55
1 parent 3114ee0
commit ece7c55
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/types_gen_ext.go b/types_gen_ext.go
@@ -1250,6 +1250,10 @@ func (update *Update) Msg() *Message {
 		return update.EditedChannelPost
 	case update.CallbackQuery != nil && update.CallbackQuery.Message != nil && update.CallbackQuery.Message.Message != nil:
 		return update.CallbackQuery.Message.Message
+	case update.BusinessMessage != nil:
+		return update.BusinessMessage
+	case update.EditedBusinessMessage != nil:
+		return update.EditedBusinessMessage
 	default:
 		return nil
 	}

diff --git a/types_gen_ext_test.go b/types_gen_ext_test.go
@@ -1124,6 +1124,8 @@ func TestUpdate_Msg(t *testing.T) {
 			Message: msg,
 		}}}, msg},
 		{&Update{CallbackQuery: &CallbackQuery{}}, nil},
+		{&Update{BusinessMessage: msg}, msg},
+		{&Update{EditedBusinessMessage: msg}, msg},
 	} {
 		assert.Equal(t, test.Message, test.Update.Msg())
 	}