mirror of
https://github.com/lukaszraczylo/go-telegram.git
synced 2026-07-01 14:45:13 +00:00
Initial release of go-telegram
A fully-generated, strongly-typed Go client for the Telegram Bot API. * 176 methods + 301 types generated from Bot API v10.0 * 1408 auto-generated tests (8 scenarios per method) * Typed unions throughout — no 'any' in the public surface * Pluggable HTTP transport and JSON codec (default goccy/go-json) * Built-in retry middleware honouring Telegram's retry_after * Generic dispatcher with filters and conversation handlers * Self-verifying codegen pipeline (regen → audit → emit → run tests) * 14 example bots covering common patterns
This commit is contained in:
@@ -0,0 +1,144 @@
|
||||
package client
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"errors"
|
||||
"io"
|
||||
"net/http"
|
||||
"strings"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/stretchr/testify/mock"
|
||||
"github.com/stretchr/testify/require"
|
||||
)
|
||||
|
||||
type retryMockDoer struct{ mock.Mock }
|
||||
|
||||
func (m *retryMockDoer) Do(r *http.Request) (*http.Response, error) {
|
||||
args := m.Called(r)
|
||||
if v := args.Get(0); v != nil {
|
||||
return v.(*http.Response), args.Error(1)
|
||||
}
|
||||
return nil, args.Error(1)
|
||||
}
|
||||
|
||||
func okResp(body string) *http.Response {
|
||||
return &http.Response{
|
||||
StatusCode: 200,
|
||||
Body: io.NopCloser(bytes.NewBufferString(body)),
|
||||
Header: http.Header{"Content-Type": []string{"application/json"}},
|
||||
}
|
||||
}
|
||||
|
||||
func TestRetryDoer_HappyPath(t *testing.T) {
|
||||
m := &retryMockDoer{}
|
||||
m.On("Do", mock.Anything).Return(okResp(`{"ok":true,"result":"hi"}`), nil).Once()
|
||||
|
||||
d := NewRetryDoer(m)
|
||||
req, _ := http.NewRequest("POST", "http://x", strings.NewReader(`{}`))
|
||||
resp, err := d.Do(req)
|
||||
require.NoError(t, err)
|
||||
require.Equal(t, 200, resp.StatusCode)
|
||||
m.AssertExpectations(t)
|
||||
}
|
||||
|
||||
func TestRetryDoer_RetriesOnNetworkError(t *testing.T) {
|
||||
m := &retryMockDoer{}
|
||||
m.On("Do", mock.Anything).Return(nil, errors.New("dial timeout")).Once()
|
||||
m.On("Do", mock.Anything).Return(okResp(`{"ok":true,"result":"hi"}`), nil).Once()
|
||||
|
||||
d := NewRetryDoer(m, WithBaseBackoff(time.Millisecond))
|
||||
req, _ := http.NewRequest("POST", "http://x", strings.NewReader(`{}`))
|
||||
resp, err := d.Do(req)
|
||||
require.NoError(t, err)
|
||||
require.Equal(t, 200, resp.StatusCode)
|
||||
m.AssertExpectations(t)
|
||||
}
|
||||
|
||||
func TestRetryDoer_HonoursRetryAfter(t *testing.T) {
|
||||
m := &retryMockDoer{}
|
||||
m.On("Do", mock.Anything).Return(
|
||||
okResp(`{"ok":false,"error_code":429,"description":"Too Many","parameters":{"retry_after":1}}`), nil).Once()
|
||||
m.On("Do", mock.Anything).Return(okResp(`{"ok":true,"result":1}`), nil).Once()
|
||||
|
||||
// base is 10s — retry_after=1s should override it (much shorter wait).
|
||||
d := NewRetryDoer(m, WithBaseBackoff(10*time.Second))
|
||||
req, _ := http.NewRequest("POST", "http://x", strings.NewReader(`{}`))
|
||||
start := time.Now()
|
||||
resp, err := d.Do(req)
|
||||
elapsed := time.Since(start)
|
||||
require.NoError(t, err)
|
||||
require.Equal(t, 200, resp.StatusCode)
|
||||
require.GreaterOrEqual(t, elapsed, 900*time.Millisecond, "should honour retry_after=1s")
|
||||
require.Less(t, elapsed, 3*time.Second, "should NOT use base backoff (10s)")
|
||||
m.AssertExpectations(t)
|
||||
}
|
||||
|
||||
func TestRetryDoer_Retries5xx(t *testing.T) {
|
||||
m := &retryMockDoer{}
|
||||
m.On("Do", mock.Anything).Return(
|
||||
okResp(`{"ok":false,"error_code":500,"description":"Internal Server Error"}`), nil).Once()
|
||||
m.On("Do", mock.Anything).Return(okResp(`{"ok":true,"result":1}`), nil).Once()
|
||||
|
||||
d := NewRetryDoer(m, WithBaseBackoff(time.Millisecond))
|
||||
req, _ := http.NewRequest("POST", "http://x", strings.NewReader(`{}`))
|
||||
resp, err := d.Do(req)
|
||||
require.NoError(t, err)
|
||||
require.Equal(t, 200, resp.StatusCode)
|
||||
m.AssertExpectations(t)
|
||||
}
|
||||
|
||||
func TestRetryDoer_AllAttemptsFail(t *testing.T) {
|
||||
m := &retryMockDoer{}
|
||||
m.On("Do", mock.Anything).Return(nil, errors.New("dial timeout"))
|
||||
|
||||
d := NewRetryDoer(m, WithMaxAttempts(3), WithBaseBackoff(time.Millisecond))
|
||||
req, _ := http.NewRequest("POST", "http://x", strings.NewReader(`{}`))
|
||||
_, err := d.Do(req)
|
||||
require.Error(t, err)
|
||||
require.Contains(t, err.Error(), "dial timeout")
|
||||
require.Equal(t, 3, len(m.Calls))
|
||||
}
|
||||
|
||||
func TestRetryDoer_ContextCancellationAborts(t *testing.T) {
|
||||
m := &retryMockDoer{}
|
||||
m.On("Do", mock.Anything).Return(
|
||||
okResp(`{"ok":false,"error_code":500,"description":"server error"}`), nil).Maybe()
|
||||
|
||||
d := NewRetryDoer(m, WithBaseBackoff(100*time.Millisecond))
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 50*time.Millisecond)
|
||||
defer cancel()
|
||||
req, _ := http.NewRequestWithContext(ctx, "POST", "http://x", strings.NewReader(`{}`))
|
||||
_, err := d.Do(req)
|
||||
require.Error(t, err)
|
||||
require.True(t, errors.Is(err, context.DeadlineExceeded))
|
||||
}
|
||||
|
||||
func TestRetryDoer_ReplaysBody(t *testing.T) {
|
||||
m := &retryMockDoer{}
|
||||
var seen []string
|
||||
|
||||
// First call: capture body, return 500 to trigger retry.
|
||||
m.On("Do", mock.Anything).Return(okResp(`{"ok":false,"error_code":500}`), nil).Once().Run(func(args mock.Arguments) {
|
||||
r := args.Get(0).(*http.Request)
|
||||
body, _ := io.ReadAll(r.Body)
|
||||
seen = append(seen, string(body))
|
||||
})
|
||||
// Second call: capture body, return success.
|
||||
m.On("Do", mock.Anything).Return(okResp(`{"ok":true}`), nil).Once().Run(func(args mock.Arguments) {
|
||||
r := args.Get(0).(*http.Request)
|
||||
body, _ := io.ReadAll(r.Body)
|
||||
seen = append(seen, string(body))
|
||||
})
|
||||
|
||||
d := NewRetryDoer(m, WithBaseBackoff(time.Millisecond))
|
||||
req, _ := http.NewRequest("POST", "http://x", strings.NewReader(`{"chat_id":42}`))
|
||||
_, err := d.Do(req)
|
||||
require.NoError(t, err)
|
||||
require.Len(t, seen, 2)
|
||||
require.Equal(t, seen[0], seen[1])
|
||||
require.Equal(t, `{"chat_id":42}`, seen[0])
|
||||
m.AssertExpectations(t)
|
||||
}
|
||||
Reference in New Issue
Block a user