mirror of
https://github.com/HugeFrog24/go-telegram-bot.git
synced 2026-03-02 00:14:34 +00:00
c8af457af1
md formatting doesnt work yet Started implementing owner feature Add .gitattributes to enforce LF line endings Temporary commit before merge Updated owner management Updated json and gitignore Proceed with role management Again, CI Fix some lint errors Implemented screening Per-bot API keys implemented Use getRoleByName func Fix unused imports Upgrade actions rm unused function Upgrade action Fix unaddressed errors
36 lines
1.0 KiB
Go
36 lines
1.0 KiB
Go
// telegram_client_mock.go
|
|
package main
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/go-telegram/bot"
|
|
"github.com/go-telegram/bot/models"
|
|
)
|
|
|
|
// MockTelegramClient is a mock implementation of TelegramClient for testing.
|
|
type MockTelegramClient struct {
|
|
// You can add fields to keep track of calls if needed.
|
|
SendMessageFunc func(ctx context.Context, params *bot.SendMessageParams) (*models.Message, error)
|
|
StartFunc func(ctx context.Context) // Optional: track Start calls
|
|
}
|
|
|
|
// SendMessage mocks sending a message.
|
|
func (m *MockTelegramClient) SendMessage(ctx context.Context, params *bot.SendMessageParams) (*models.Message, error) {
|
|
if m.SendMessageFunc != nil {
|
|
return m.SendMessageFunc(ctx, params)
|
|
}
|
|
// Default behavior: return an empty message without error.
|
|
return &models.Message{}, nil
|
|
}
|
|
|
|
// Start mocks starting the Telegram client.
|
|
func (m *MockTelegramClient) Start(ctx context.Context) {
|
|
if m.StartFunc != nil {
|
|
m.StartFunc(ctx)
|
|
}
|
|
// Default behavior: do nothing.
|
|
}
|
|
|
|
// Add other mocked methods if your Bot uses more TelegramClient methods.
|