FILE / ScuroNeko/Laniakea
tgapi/api_test.go
Исходный файл и его история в репозитории.
91 lines
2.2 KiB
Go
91 lines
2.2 KiB
Go
package tgapi
|
|
|
|
import (
|
|
"io"
|
|
"net/http"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
type roundTripFunc func(*http.Request) (*http.Response, error)
|
|
|
|
func (fn roundTripFunc) RoundTrip(req *http.Request) (*http.Response, error) {
|
|
return fn(req)
|
|
}
|
|
|
|
type closingTransport struct {
|
|
roundTripFunc
|
|
closed bool
|
|
}
|
|
|
|
func (t *closingTransport) CloseIdleConnections() {
|
|
t.closed = true
|
|
}
|
|
|
|
func TestAPILeavesAcceptEncodingToHTTPTransport(t *testing.T) {
|
|
var gotPath string
|
|
var gotAcceptEncoding string
|
|
|
|
client := &http.Client{
|
|
Transport: roundTripFunc(func(req *http.Request) (*http.Response, error) {
|
|
gotPath = req.URL.Path
|
|
gotAcceptEncoding = req.Header.Get("Accept-Encoding")
|
|
return &http.Response{
|
|
StatusCode: http.StatusOK,
|
|
Header: http.Header{"Content-Type": []string{"application/json"}},
|
|
Body: io.NopCloser(strings.NewReader(`{"ok":true,"result":{"id":1,"is_bot":true,"first_name":"Test"}}`)),
|
|
}, nil
|
|
}),
|
|
}
|
|
|
|
api := NewAPI(
|
|
NewAPIOpts("token").
|
|
SetAPIUrl("https://example.test").
|
|
SetHTTPClient(client),
|
|
)
|
|
defer func() {
|
|
if err := api.Close(); err != nil {
|
|
t.Fatalf("Close returned error: %v", err)
|
|
}
|
|
}()
|
|
|
|
user, err := api.GetMe()
|
|
if err != nil {
|
|
t.Fatalf("GetMe returned error: %v", err)
|
|
}
|
|
if user.FirstName != "Test" {
|
|
t.Fatalf("unexpected first name: %q", user.FirstName)
|
|
}
|
|
if gotPath != "/bottoken/getMe" {
|
|
t.Fatalf("unexpected request path: %s", gotPath)
|
|
}
|
|
if gotAcceptEncoding != "" {
|
|
t.Fatalf("expected empty Accept-Encoding header, got %q", gotAcceptEncoding)
|
|
}
|
|
}
|
|
|
|
func TestAPICloseClosesIdleConnections(t *testing.T) {
|
|
transport := &closingTransport{
|
|
roundTripFunc: func(req *http.Request) (*http.Response, error) {
|
|
return &http.Response{
|
|
StatusCode: http.StatusOK,
|
|
Header: http.Header{"Content-Type": []string{"application/json"}},
|
|
Body: io.NopCloser(strings.NewReader(`{"ok":true,"result":{"id":1,"is_bot":true,"first_name":"Test"}}`)),
|
|
}, nil
|
|
},
|
|
}
|
|
|
|
api := NewAPI(
|
|
NewAPIOpts("token").
|
|
SetAPIUrl("https://example.test").
|
|
SetHTTPClient(&http.Client{Transport: transport}),
|
|
)
|
|
|
|
if err := api.Close(); err != nil {
|
|
t.Fatalf("Close returned error: %v", err)
|
|
}
|
|
if !transport.closed {
|
|
t.Fatal("expected Close to close idle HTTP connections")
|
|
}
|
|
}
|