package telegram_test

import (
	"testing"
	"time"

	"github.com/go-faster/errors"
	"github.com/stretchr/testify/assert"

	"go.mau.fi/mautrix-telegram/pkg/gotd/telegram"
	"go.mau.fi/mautrix-telegram/pkg/gotd/tgerr"
)

func TestAsFloodWait(t *testing.T) {
	err := func() error {
		return errors.Wrap(tgerr.New(400, "FLOOD_WAIT_10"), "perform operation")
	}()

	d, ok := telegram.AsFloodWait(err)
	assert.True(t, ok)
	assert.Equal(t, time.Second*10, d)
}