mirror of
https://github.com/tucnak/telebot
synced 2024-11-11 01:10:39 +00:00
75 lines
1.6 KiB
Go
75 lines
1.6 KiB
Go
package telebot
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"io"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
// testPayload implements json.Marshaler
|
|
// to test json encoding error behaviour.
|
|
type testPayload struct{}
|
|
|
|
func (testPayload) MarshalJSON() ([]byte, error) {
|
|
return nil, errors.New("test error")
|
|
}
|
|
|
|
func testRawServer(w http.ResponseWriter, r *http.Request) {
|
|
switch {
|
|
// causes EOF error on ioutil.ReadAll
|
|
case strings.HasSuffix(r.URL.Path, "/testReadError"):
|
|
// tells the body is 1 byte length but actually it's 0
|
|
w.Header().Set("Content-Length", "1")
|
|
|
|
// returns unknown telegram error
|
|
case strings.HasSuffix(r.URL.Path, "/testUnknownError"):
|
|
data, _ := json.Marshal(struct {
|
|
Ok bool `json:"ok"`
|
|
Code int `json:"error_code"`
|
|
Description string `json:"description"`
|
|
}{
|
|
Ok: false,
|
|
Code: 400,
|
|
Description: "unknown error",
|
|
})
|
|
|
|
w.WriteHeader(400)
|
|
w.Write(data)
|
|
}
|
|
}
|
|
|
|
func TestRaw(t *testing.T) {
|
|
if token == "" {
|
|
t.Skip("TELEBOT_SECRET is required")
|
|
}
|
|
|
|
b, err := newTestBot()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
_, err = b.Raw("BAD METHOD", nil)
|
|
assert.EqualError(t, err, ErrNotFound.Error())
|
|
|
|
_, err = b.Raw("", &testPayload{})
|
|
assert.Error(t, err)
|
|
|
|
srv := httptest.NewServer(http.HandlerFunc(testRawServer))
|
|
defer srv.Close()
|
|
|
|
b.URL = srv.URL
|
|
b.client = srv.Client()
|
|
|
|
_, err = b.Raw("testReadError", nil)
|
|
assert.EqualError(t, err, "telebot: "+io.ErrUnexpectedEOF.Error())
|
|
|
|
_, err = b.Raw("testUnknownError", nil)
|
|
assert.EqualError(t, err, "telegram unknown: unknown error (400)")
|
|
}
|